fix t5gemma tests (#39052)
* fix * fix * fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -595,6 +595,11 @@ class T5GemmaModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMi
|
||||
|
||||
# used in `test_torch_compile_for_training`
|
||||
_torch_compile_train_cls = T5GemmaForConditionalGeneration if is_torch_available() else None
|
||||
# `t5gemma` will give warning or raise error if it is not `eager` during training.
|
||||
_torch_compile_train_attn_implementation = "eager"
|
||||
|
||||
# won't fix
|
||||
test_torchscript = False
|
||||
|
||||
def setUp(self):
|
||||
self.model_tester = T5GemmaModelTester(self)
|
||||
@@ -1584,6 +1589,9 @@ class T5GemmaEncoderOnlyModelTest(ModelTesterMixin, unittest.TestCase):
|
||||
is_encoder_decoder = False
|
||||
model_split_percents = [0.4, 0.5]
|
||||
|
||||
# won't fix
|
||||
test_torchscript = False
|
||||
|
||||
def setUp(self):
|
||||
self.model_tester = T5GemmaEncoderOnlyModelTester(self)
|
||||
self.config_tester = ConfigTester(
|
||||
|
||||
Reference in New Issue
Block a user