ydshieh commited on
Commit
b31314b
1 Parent(s): b04e4c6

update config script

Browse files
Files changed (1) hide show
  1. vit_gpt2/configuration_vit_gpt2.py +1 -1
vit_gpt2/configuration_vit_gpt2.py CHANGED
@@ -35,7 +35,7 @@ class ViTGPT2Config(PretrainedConfig):
35
  self.forced_eos_token_id = self.text_config.eos_token_id
36
 
37
  @classmethod
38
- def from_vit_gpt2_configs(cls, vision_config: ViTConfig, text_config: GPT2Config, **kwargs):
39
 
40
  return cls(vision_config_dict=vision_config.to_dict(), text_config_dict=text_config.to_dict(), **kwargs)
41
 
 
35
  self.forced_eos_token_id = self.text_config.eos_token_id
36
 
37
  @classmethod
38
+ def from_vision_text_configs(cls, vision_config: ViTConfig, text_config: GPT2Config, **kwargs):
39
 
40
  return cls(vision_config_dict=vision_config.to_dict(), text_config_dict=text_config.to_dict(), **kwargs)
41