Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3312e96bfbebbad67ff29539d2df9211923ddd70"
Unverified Commit 9ca3aa01 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `all_model_classes` in `FlaxBloomGenerationTest` (#25211) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 59dcea3f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/bloom/test_modeling_flax_bloom.py
View file @ 9ca3aa01
...@@ -198,7 +198,7 @@ class FlaxBloomModelTest(FlaxModelTesterMixin, unittest.TestCase, FlaxGeneration ...@@ -198,7 +198,7 @@ class FlaxBloomModelTest(FlaxModelTesterMixin, unittest.TestCase, FlaxGeneration
@slow @slow
@require_flax @require_flax
class FlaxBloomGenerationTest(unittest.TestCase): class FlaxBloomGenerationTest(unittest.TestCase):
all_model_classes = (FlaxBloomForCausalLM) if is_flax_available() else () all_model_classes = (FlaxBloomForCausalLM,) if is_flax_available() else ()
all_generative_model_classes = () if is_flax_available() else () all_generative_model_classes = () if is_flax_available() else ()
def setUp(self): def setUp(self):
......
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