yuchenxie commited on
Commit
b1f7a0f
1 Parent(s): ed83564

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -3
config.json CHANGED
@@ -1,10 +1,12 @@
1
  {
2
  "architectures": [
3
- "ArlowGPTForCausalLM"
 
4
  ],
5
  "auto_map": {
6
  "AutoConfig": "configuration_arlow_gpt.ArlowGPTConfig",
7
- "AutoModelForCausalLM": "modeling_arlow_gpt.ArlowGPTForCausalLM"
 
8
  },
9
  "clip_config": null,
10
  "clip_model_name": "yuchenxie/CLiP",
@@ -14,5 +16,6 @@
14
  "projection_dim": 768,
15
  "torch_dtype": "float32",
16
  "transformers_version": "4.46.2",
17
- "vocab_size": 50257
 
18
  }
 
1
  {
2
  "architectures": [
3
+ "ArlowGPTForCausalLM",
4
+ "ArlowGPTForImageTextToText"
5
  ],
6
  "auto_map": {
7
  "AutoConfig": "configuration_arlow_gpt.ArlowGPTConfig",
8
+ "AutoModelForCausalLM": "modeling_arlow_gpt.ArlowGPTForCausalLM",
9
+ "AutoModelForImageTextToText": "modeling_arlow_gpt.ArlowGPTForImageTextToText"
10
  },
11
  "clip_config": null,
12
  "clip_model_name": "yuchenxie/CLiP",
 
16
  "projection_dim": 768,
17
  "torch_dtype": "float32",
18
  "transformers_version": "4.46.2",
19
+ "vocab_size": 50257,
20
+ "_name_or_path": "yuchenxie/GPT-2V"
21
  }