TheMrguiller commited on
Commit
fddcdd9
1 Parent(s): 6d5c305

Upload model

Browse files
Files changed (3) hide show
  1. config.json +26 -0
  2. generation_config.json +4 -0
  3. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/home/d4k/Documents/guillermo/MUCSI_Modal/flamingo-train_task/training/flamingo-Bilbao/checkpoint-5160",
3
+ "architectures": [
4
+ "FlamingoModel"
5
+ ],
6
+ "clip_model_type": "openai/clip-vit-large-patch14",
7
+ "dim": 1024,
8
+ "dim_visual": 1024,
9
+ "freeze_language_model": true,
10
+ "freeze_vision_model": true,
11
+ "lm": "facebook/opt-350m",
12
+ "resampler_act": "sqrelu",
13
+ "resampler_depth": 6,
14
+ "resampler_dim_head": 128,
15
+ "resampler_ff_mult": 4,
16
+ "resampler_heads": 16,
17
+ "resampler_num_latents": 64,
18
+ "resampler_num_time_embeds": 4,
19
+ "torch_dtype": "float32",
20
+ "transformers_version": "4.29.2",
21
+ "xattn_act": "sqrelu",
22
+ "xattn_dim_head": 128,
23
+ "xattn_every": 1,
24
+ "xattn_ff_mult": 4,
25
+ "xattn_heads": 16
26
+ }
generation_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "transformers_version": "4.29.2"
4
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f59999333c02f68da7019eebd8af53ea09731e79c48a365aa0479dead56cd91
3
+ size 4551988605