From 7b139023c358928d72d944e8b5053739c1696a68 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Mon, 13 Nov 2023 18:18:41 +0100
Subject: [PATCH] [`AWQ` ] Addresses TODO for awq tests (#27467)

addresses todo for awq tests
---
 tests/quantization/autoawq/test_awq.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tests/quantization/autoawq/test_awq.py b/tests/quantization/autoawq/test_awq.py
index 1f1b51b778..f0854e4255 100644
--- 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"