ccloud0525 commited on
Commit
bd420ab
·
1 Parent(s): cacd498

feat: "first commit"

Browse files
Files changed (1) hide show
  1. 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,