Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
ae87f979
Unverified
Commit
ae87f979
authored
May 24, 2024
by
Marc Sun
Committed by
GitHub
May 24, 2024
Browse files
FIX / TST: Fix expected results on Mistral AWQ test (#30971)
fix awq mistral test
parent
04c7c176
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/quantization/autoawq/test_awq.py
tests/quantization/autoawq/test_awq.py
+1
-1
No files found.
tests/quantization/autoawq/test_awq.py
View file @
ae87f979
...
@@ -288,7 +288,7 @@ class AwqFusedTest(unittest.TestCase):
...
@@ -288,7 +288,7 @@ class AwqFusedTest(unittest.TestCase):
"You end up exactly where you started. Where are you?"
"You end up exactly where you started. Where are you?"
)
)
EXPECTED_GENERATION
=
prompt
+
"
\n\n
You are at the starting point.
\n\n
If
"
EXPECTED_GENERATION
=
prompt
+
"
\n\n
This is a classic puzzle that has been around for
"
EXPECTED_GENERATION_CUSTOM_MODEL
=
"Hello,
\n\n
I have a problem with my 20"
EXPECTED_GENERATION_CUSTOM_MODEL
=
"Hello,
\n\n
I have a problem with my 20"
EXPECTED_GENERATION_MIXTRAL
=
prompt
+
" You're on the North Pole.
\n\n
The"
EXPECTED_GENERATION_MIXTRAL
=
prompt
+
" You're on the North Pole.
\n\n
The"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment