ccloud0525 commited on
Commit ·
bd420ab
1
Parent(s): cacd498
feat: "first commit"
Browse files- configuration_aurora.py +4 -0
configuration_aurora.py
CHANGED
|
@@ -71,11 +71,15 @@ class AuroraConfig(PretrainedConfig):
|
|
| 71 |
self.vision_config = ViTConfig()
|
| 72 |
elif isinstance(vision_config, dict):
|
| 73 |
self.vision_config = ViTConfig(**vision_config)
|
|
|
|
|
|
|
| 74 |
|
| 75 |
if text_config is None:
|
| 76 |
self.text_config = BertConfig()
|
| 77 |
elif isinstance(text_config, dict):
|
| 78 |
self.text_config = BertConfig(**text_config)
|
|
|
|
|
|
|
| 79 |
|
| 80 |
super().__init__(
|
| 81 |
**kwargs,
|
|
|
|
| 71 |
self.vision_config = ViTConfig()
|
| 72 |
elif isinstance(vision_config, dict):
|
| 73 |
self.vision_config = ViTConfig(**vision_config)
|
| 74 |
+
else:
|
| 75 |
+
self.vision_config = vision_config
|
| 76 |
|
| 77 |
if text_config is None:
|
| 78 |
self.text_config = BertConfig()
|
| 79 |
elif isinstance(text_config, dict):
|
| 80 |
self.text_config = BertConfig(**text_config)
|
| 81 |
+
else:
|
| 82 |
+
self.text_config = text_config
|
| 83 |
|
| 84 |
super().__init__(
|
| 85 |
**kwargs,
|