[`AWQ` ] Addresses TODO for awq tests (#27467)

addresses todo for awq tests

[`AWQ` ] Addresses TODO for awq tests (#27467)
addresses todo for awq tests
7b139023 · Younes Belkada · GitHub · 04af4b90 · 7b139023
Unverified Commit 7b139023 authored Nov 13, 2023 by Younes Belkada Committed by GitHub Nov 13, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

tests/quantization/autoawq/test_awq.py tests/quantization/autoawq/test_awq.py +1 -2

No files found.
--- a/tests/quantization/autoawq/test_awq.py
+++ b/tests/quantization/autoawq/test_awq.py
@@ -85,8 +85,7 @@ class AwqConfigTest(unittest.TestCase):
 @require_auto_awq
 @require_accelerate
 class AwqTest(unittest.TestCase):
-    # TODO: @younesbelkada change it to `TheBloke/Mistral-7B-v0.1-AWQ` in the future
-    model_name = "ybelkada/test-mistral-7b-v0.1-awq"
+    model_name = "TheBloke/Mistral-7B-v0.1-AWQ"
    dummy_transformers_model_name = "bigscience/bloom-560m"

    input_text = "Hello my name is"