lazy import CONFIG_MAPPING
Browse files
configuration_interns1.py
CHANGED
@@ -15,7 +15,7 @@
|
|
15 |
|
16 |
|
17 |
from transformers.configuration_utils import PretrainedConfig
|
18 |
-
from transformers import
|
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
|