"examples/vscode:/vscode.git/clone" did not exist on "b8697bc62216b9e2ca60811626c6a6ca992b0d34"
Unverified Commit 7b139023 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

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

addresses todo for awq tests
parent 04af4b90
...@@ -85,8 +85,7 @@ class AwqConfigTest(unittest.TestCase): ...@@ -85,8 +85,7 @@ class AwqConfigTest(unittest.TestCase):
@require_auto_awq @require_auto_awq
@require_accelerate @require_accelerate
class AwqTest(unittest.TestCase): class AwqTest(unittest.TestCase):
# TODO: @younesbelkada change it to `TheBloke/Mistral-7B-v0.1-AWQ` in the future model_name = "TheBloke/Mistral-7B-v0.1-AWQ"
model_name = "ybelkada/test-mistral-7b-v0.1-awq"
dummy_transformers_model_name = "bigscience/bloom-560m" dummy_transformers_model_name = "bigscience/bloom-560m"
input_text = "Hello my name is" input_text = "Hello my name is"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment