Add a test for mixed precision (#9806)
This commit is contained in:
@@ -330,12 +330,8 @@ class TFFlaubertModelTest(TFModelTesterMixin, unittest.TestCase):
|
||||
model = TFFlaubertModel.from_pretrained(model_name)
|
||||
self.assertIsNotNone(model)
|
||||
|
||||
def test_saved_model_with_hidden_states_output(self):
|
||||
# Should be uncommented during patrick TF refactor
|
||||
pass
|
||||
|
||||
def test_saved_model_with_attentions_output(self):
|
||||
# Should be uncommented during patrick TF refactor
|
||||
def test_mixed_precision(self):
|
||||
# TODO JP: Make Flaubert float16 compliant
|
||||
pass
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user