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
OpenDAS
ColossalAI
Commits
c3ca53cf
Commit
c3ca53cf
authored
Aug 02, 2023
by
FoolPlayer
Committed by
Hongxin Liu
Aug 15, 2023
Browse files
[test] skip some not compatible models
parent
726541af
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
3 deletions
+7
-3
tests/test_booster/test_plugin/test_gemini_plugin.py
tests/test_booster/test_plugin/test_gemini_plugin.py
+4
-1
tests/test_lazy/test_models.py
tests/test_lazy/test_models.py
+3
-2
No files found.
tests/test_booster/test_plugin/test_gemini_plugin.py
View file @
c3ca53cf
...
@@ -90,7 +90,10 @@ def check_gemini_plugin(init_method: str = 'none', early_stop: bool = True):
...
@@ -90,7 +90,10 @@ def check_gemini_plugin(init_method: str = 'none', early_stop: bool = True):
'transformers_gpt_double_heads'
,
'torchaudio_hubert_base'
,
'torchaudio_wav2vec2_base'
,
'transformers_gpt_double_heads'
,
'torchaudio_hubert_base'
,
'torchaudio_wav2vec2_base'
,
'transformers_t5_for_conditional_generation'
,
'transformers_t5'
,
'transformers_t5_encoder_model'
,
'transformers_t5_for_conditional_generation'
,
'transformers_t5'
,
'transformers_t5_encoder_model'
,
'transformers_vit'
,
'transformers_vit_for_masked_image_modeling'
,
'transformers_vit'
,
'transformers_vit_for_masked_image_modeling'
,
'transformers_vit_for_image_classification'
'transformers_vit_for_image_classification'
,
'transformers_chatglm'
,
'transformers_chatglm_for_conditional_generation'
,
'transformers_blip2'
,
'transformers_blip2_conditional_gerneration'
,
'transformers_sam'
,
'transformers_whisper'
,
'transformers_whisperForConditionalGeneration'
,
'transformers_whisperWhisperForAudioClassification'
]:
]:
continue
continue
...
...
tests/test_lazy/test_models.py
View file @
c3ca53cf
...
@@ -11,8 +11,9 @@ def test_torchvision_models_lazy_init(subset, default_device):
...
@@ -11,8 +11,9 @@ def test_torchvision_models_lazy_init(subset, default_device):
sub_model_zoo
=
model_zoo
.
get_sub_registry
(
subset
)
sub_model_zoo
=
model_zoo
.
get_sub_registry
(
subset
)
for
name
,
entry
in
sub_model_zoo
.
items
():
for
name
,
entry
in
sub_model_zoo
.
items
():
# TODO(ver217): lazy init does not support weight norm, skip these models
# TODO(ver217): lazy init does not support weight norm, skip these models
if
name
in
(
'torchaudio_wav2vec2_base'
,
'torchaudio_hubert_base'
if
name
in
(
'torchaudio_wav2vec2_base'
,
)
or
name
.
startswith
(
'transformers_llama'
)
or
name
.
startswith
(
'transformers_vit'
):
'torchaudio_hubert_base'
)
or
name
.
startswith
(
'transformers_llama'
)
or
name
.
startswith
(
(
'transformers_vit'
,
'transformers_blip2'
)):
continue
continue
check_lazy_init
(
entry
,
verbose
=
True
,
default_device
=
default_device
)
check_lazy_init
(
entry
,
verbose
=
True
,
default_device
=
default_device
)
...
...
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