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

feat: "first commit"

Browse files
Files changed (2) hide show
  1. config.json +1 -1
  2. configuration_aurora.py +0 -3
config.json CHANGED
@@ -5,7 +5,7 @@
5
  "model_type": "aurora",
6
  "auto_map": {
7
  "AutoConfig": "configuration_aurora.AuroraConfig",
8
- "AutoModel": "modeling_aurora.AuroraForPrediction"
9
  },
10
 
11
  "hidden_size": 256,
 
5
  "model_type": "aurora",
6
  "auto_map": {
7
  "AutoConfig": "configuration_aurora.AuroraConfig",
8
+ "AutoModelForCausalLM": "modeling_aurora.AuroraForPrediction"
9
  },
10
 
11
  "hidden_size": 256,
configuration_aurora.py CHANGED
@@ -71,16 +71,13 @@ class AuroraConfig(PretrainedConfig):
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
 
 
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