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
......@@ -198,7 +198,7 @@ class FlaxBloomModelTest(FlaxModelTesterMixin, unittest.TestCase, FlaxGeneration
@slow
@require_flax
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 ()
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