Update configuration_arlow_gpt.py
Browse files
configuration_arlow_gpt.py
CHANGED
@@ -1,6 +1,10 @@
|
|
1 |
from transformers import PretrainedConfig
|
2 |
from typing import Dict, Optional
|
3 |
|
|
|
|
|
|
|
|
|
4 |
class ArlowGPTConfig(PretrainedConfig):
|
5 |
model_type = "arlow_gpt"
|
6 |
|
@@ -20,4 +24,4 @@ class ArlowGPTConfig(PretrainedConfig):
|
|
20 |
self.clip_config = clip_config
|
21 |
self.gpt2_config = gpt2_config
|
22 |
self.projection_dim = projection_dim
|
23 |
-
self.vocab_size = vocab_size
|
|
|
1 |
from transformers import PretrainedConfig
|
2 |
from typing import Dict, Optional
|
3 |
|
4 |
+
ARLOW_GPT_PRETRAINED_CONFIG_ARCHIVE_MAP = {
|
5 |
+
"yuchenxie/arlow-gpt": "https://huggingface.co/yuchenxie/arlow-gpt/resolve/main/config.json",
|
6 |
+
}
|
7 |
+
|
8 |
class ArlowGPTConfig(PretrainedConfig):
|
9 |
model_type = "arlow_gpt"
|
10 |
|
|
|
24 |
self.clip_config = clip_config
|
25 |
self.gpt2_config = gpt2_config
|
26 |
self.projection_dim = projection_dim
|
27 |
+
self.vocab_size = vocab_size
|