TheMrguiller commited on
Commit
b195c4e
1 Parent(s): fb99c5d

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-8991",
3
+ "architectures": [
4
+ "FlamingoModel"
5
+ ],
6
+ "clip_model_type": "openai/clip-vit-large-patch14",
7
+ "dim": 768,
8
+ "dim_visual": 1024,
9
+ "freeze_language_model": true,
10
+ "freeze_vision_model": true,
11
+ "lm": "facebook/opt-125m",
12
+ "resampler_act": "sqrelu",
13
+ "resampler_depth": 6,
14
+ "resampler_dim_head": 64,
15
+ "resampler_ff_mult": 4,
16
+ "resampler_heads": 8,
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": 64,
23
+ "xattn_every": 1,
24
+ "xattn_ff_mult": 4,
25
+ "xattn_heads": 8
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:6db913056544134338200af76151fb698093185038111617f005fa8616caf45b
3
+ size 2280769341