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
c876d121
Unverified
Commit
c876d121
authored
May 21, 2024
by
Younes Belkada
Committed by
GitHub
May 21, 2024
Browse files
FIX / TST: Fix expected results on Mistral slow test (A10) (#30909)
Update test_modeling_mistral.py
parent
0df888ff
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/models/mistral/test_modeling_mistral.py
tests/models/mistral/test_modeling_mistral.py
+1
-1
No files found.
tests/models/mistral/test_modeling_mistral.py
View file @
c876d121
...
...
@@ -627,7 +627,7 @@ class MistralIntegrationTest(unittest.TestCase):
def
test_speculative_generation
(
self
):
EXPECTED_TEXT_COMPLETION
=
{
7
:
"My favourite condiment is 100% Sriracha. I love the heat, the tang and the fact costs"
,
8
:
"My favourite condiment is 100%
Sriracha
. I love
the heat, the sweetness, the tan
g"
,
8
:
"My favourite condiment is 100%
ketchup
. I love
it on everything. I’m not a bi
g"
,
}
prompt
=
"My favourite condiment is "
tokenizer
=
AutoTokenizer
.
from_pretrained
(
"mistralai/Mistral-7B-v0.1"
,
use_fast
=
False
)
...
...
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