Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
e107ae36
Unverified
Commit
e107ae36
authored
Nov 14, 2023
by
Arthur
Committed by
GitHub
Nov 14, 2023
Browse files
[`CI-test_torch`] skip `test_tf_from_pt_safetensors` for 4 models (#27481)
* skip 4 tests * nits * style * wow it's not my day
parent
d71fa9f6
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
16 additions
and
0 deletions
+16
-0
tests/models/mobilebert/test_modeling_mobilebert.py
tests/models/mobilebert/test_modeling_mobilebert.py
+4
-0
tests/models/speech_to_text_2/test_modeling_speech_to_text_2.py
...models/speech_to_text_2/test_modeling_speech_to_text_2.py
+4
-0
tests/models/transfo_xl/test_modeling_transfo_xl.py
tests/models/transfo_xl/test_modeling_transfo_xl.py
+4
-0
tests/models/xglm/test_modeling_xglm.py
tests/models/xglm/test_modeling_xglm.py
+4
-0
No files found.
tests/models/mobilebert/test_modeling_mobilebert.py
View file @
e107ae36
...
...
@@ -302,6 +302,10 @@ class MobileBertModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
def
test_resize_tokens_embeddings
(
self
):
super
().
test_resize_tokens_embeddings
()
@
unittest
.
skip
(
"This test is currently broken because of safetensors."
)
def
test_tf_from_pt_safetensors
(
self
):
pass
def
setUp
(
self
):
self
.
model_tester
=
MobileBertModelTester
(
self
)
self
.
config_tester
=
ConfigTester
(
self
,
config_class
=
MobileBertConfig
,
hidden_size
=
37
)
...
...
tests/models/speech_to_text_2/test_modeling_speech_to_text_2.py
View file @
e107ae36
...
...
@@ -196,6 +196,10 @@ class Speech2Text2StandaloneDecoderModelTest(
def
test_inputs_embeds
(
self
):
pass
@
unittest
.
skip
(
"This test is currently broken because of safetensors."
)
def
test_tf_from_pt_safetensors
(
self
):
pass
# speech2text2 has no base model
def
test_save_load_fast_init_from_base
(
self
):
pass
...
...
tests/models/transfo_xl/test_modeling_transfo_xl.py
View file @
e107ae36
...
...
@@ -490,6 +490,10 @@ class TransfoXLModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTester
def
test_left_padding_compatibility
(
self
):
pass
@
unittest
.
skip
(
"This test is currently broken because of safetensors."
)
def
test_tf_from_pt_safetensors
(
self
):
pass
@
require_torch
class
TransfoXLModelLanguageGenerationTest
(
unittest
.
TestCase
):
...
...
tests/models/xglm/test_modeling_xglm.py
View file @
e107ae36
...
...
@@ -357,6 +357,10 @@ class XGLMModelTest(ModelTesterMixin, GenerationTesterMixin, PipelineTesterMixin
def
test_model_parallelism
(
self
):
super
().
test_model_parallelism
()
@
unittest
.
skip
(
"This test is currently broken because of safetensors."
)
def
test_tf_from_pt_safetensors
(
self
):
pass
@
require_torch
class
XGLMModelLanguageGenerationTest
(
unittest
.
TestCase
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment