jack-zxy commited on
Commit
a61bd1e
·
1 Parent(s): 8aa545a

lazy import CONFIG_MAPPING

Browse files
Files changed (1) hide show
  1. configuration_interns1.py +3 -1
configuration_interns1.py CHANGED
@@ -15,7 +15,7 @@
15
 
16
 
17
  from transformers.configuration_utils import PretrainedConfig
18
- from transformers import CONFIG_MAPPING, AutoConfig
19
 
20
 
21
  class InternS1VisionConfig(PretrainedConfig):
@@ -208,9 +208,11 @@ class InternS1Config(PretrainedConfig):
208
  self.vision_config = InternS1VisionConfig()
209
 
210
  if isinstance(text_config, dict):
 
211
  text_config["model_type"] = text_config["model_type"] if "model_type" in text_config else "qwen2" # todo
212
  text_config = CONFIG_MAPPING[text_config["model_type"]](**text_config)
213
  elif text_config is None:
 
214
  text_config = CONFIG_MAPPING["qwen2"]() # todo
215
 
216
  self.text_config = text_config
 
15
 
16
 
17
  from transformers.configuration_utils import PretrainedConfig
18
+ from transformers import AutoConfig
19
 
20
 
21
  class InternS1VisionConfig(PretrainedConfig):
 
208
  self.vision_config = InternS1VisionConfig()
209
 
210
  if isinstance(text_config, dict):
211
+ from transformers import CONFIG_MAPPING
212
  text_config["model_type"] = text_config["model_type"] if "model_type" in text_config else "qwen2" # todo
213
  text_config = CONFIG_MAPPING[text_config["model_type"]](**text_config)
214
  elif text_config is None:
215
+ from transformers import CONFIG_MAPPING
216
  text_config = CONFIG_MAPPING["qwen2"]() # todo
217
 
218
  self.text_config = text_config