ccloud0525 commited on
Commit
853e849
·
1 Parent(s): 75a8435

feat: "first commit"

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. configuration_aurora.py +1 -9
config.json CHANGED
@@ -5,7 +5,7 @@
5
  "model_type": "aurora",
6
  "auto_map": {
7
  "AutoConfig": "configuration_aurora.AuroraConfig",
8
- "AutoModelForCausalLM": "modeling_aurora.AuroraForPrediction"
9
  },
10
 
11
  "hidden_size": 256,
 
5
  "model_type": "aurora",
6
  "auto_map": {
7
  "AutoConfig": "configuration_aurora.AuroraConfig",
8
+ "AutoModel": "modeling_aurora.AuroraForPrediction"
9
  },
10
 
11
  "hidden_size": 256,
configuration_aurora.py CHANGED
@@ -69,28 +69,20 @@ class AuroraConfig(PretrainedConfig):
69
 
70
  if vision_config is None:
71
  self.vision_config = ViTConfig()
72
- print("vision config is None")
73
  elif isinstance(vision_config, dict):
74
  self.vision_config = ViTConfig(**vision_config)
75
- print("vision config is dict")
76
 
77
  else:
78
  self.vision_config = vision_config
79
- print("vision config is valued")
80
-
81
 
82
  if text_config is None:
83
  self.text_config = BertConfig()
84
- print("bert config is None")
85
 
86
  elif isinstance(text_config, dict):
87
  self.text_config = BertConfig(**text_config)
88
- print("bert config is dict")
89
 
90
  else:
91
  self.text_config = text_config
92
- print("bert config is valued")
93
-
94
 
95
  super().__init__(
96
  **kwargs,
@@ -113,4 +105,4 @@ class AuroraConfig(PretrainedConfig):
113
  else:
114
  output["text_config"] = self.text_config
115
 
116
- return output
 
69
 
70
  if vision_config is None:
71
  self.vision_config = ViTConfig()
 
72
  elif isinstance(vision_config, dict):
73
  self.vision_config = ViTConfig(**vision_config)
 
74
 
75
  else:
76
  self.vision_config = vision_config
 
 
77
 
78
  if text_config is None:
79
  self.text_config = BertConfig()
 
80
 
81
  elif isinstance(text_config, dict):
82
  self.text_config = BertConfig(**text_config)
 
83
 
84
  else:
85
  self.text_config = text_config
 
 
86
 
87
  super().__init__(
88
  **kwargs,
 
105
  else:
106
  output["text_config"] = self.text_config
107
 
108
+ return output