github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-04-28 19:55:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] skip some not compatible models

Browse Source
This commit is contained in:
FoolPlayer 2023-08-02 14:53:26 +08:00 committed by Hongxin Liu
parent 726541afe2
commit c3ca53cf05
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_booster/test_plugin/test_gemini_plugin.py
View File

@ -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

5
tests/test_lazy/test_models.py
View File

@ -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)