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
|