We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
all_model_classes
FlaxBloomGenerationTest
1 parent 59dcea3 commit 9ca3aa0Copy full SHA for 9ca3aa0
tests/models/bloom/test_modeling_flax_bloom.py
@@ -198,7 +198,7 @@ def test_model_from_pretrained(self):
198
@slow
199
@require_flax
200
class FlaxBloomGenerationTest(unittest.TestCase):
201
- all_model_classes = (FlaxBloomForCausalLM) if is_flax_available() else ()
+ all_model_classes = (FlaxBloomForCausalLM,) if is_flax_available() else ()
202
all_generative_model_classes = () if is_flax_available() else ()
203
204
def setUp(self):
0 commit comments