ydshieh
commited on
Commit
•
b31314b
1
Parent(s):
b04e4c6
update config script
Browse files
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
|
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 |
|