yuchenxie commited on
Commit
232b8ae
1 Parent(s): 1b096c2

Update configuration_arlow_gpt.py

Browse files
Files changed (1) hide show
  1. configuration_arlow_gpt.py +5 -1
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