Hugging Face's logo

trl-internal-testing
/
tiny-Idefics2ForConditionalGeneration

Image-to-Text
Transformers
Safetensors
idefics2
trl
text-generation-inference
Model card Files
xet
Community
qgallouedec HF Staff commited on
Commit
8dc21a4
·
verified ·
1 Parent(s): 49d38ea

Upload Idefics2ForConditionalGeneration

Browse files
Files changed (3) hide show
  1. config.json +4 -2
  2. generation_config.json +1 -1
  3. model.safetensors +1 -1
config.json CHANGED
@@ -2,6 +2,7 @@
2
  "architectures": [
3
  "Idefics2ForConditionalGeneration"
4
  ],
 
5
  "image_token_id": 32001,
6
  "model_type": "idefics2",
7
  "perceiver_config": {
@@ -24,6 +25,7 @@
24
  "hidden_size": 16,
25
  "initializer_range": 0.02,
26
  "intermediate_size": 14336,
 
27
  "max_position_embeddings": 32768,
28
  "model_type": "mistral",
29
  "num_attention_heads": 4,
@@ -37,11 +39,11 @@
37
  "vocab_size": 32003
38
  },
39
  "tie_word_embeddings": false,
40
- "torch_dtype": "bfloat16",
41
- "transformers_version": "4.56.0.dev0",
42
  "use_cache": true,
43
  "vision_config": {
44
  "attention_dropout": 0.0,
 
45
  "hidden_act": "gelu_pytorch_tanh",
46
  "hidden_size": 16,
47
  "image_size": 980,
 
2
  "architectures": [
3
  "Idefics2ForConditionalGeneration"
4
  ],
5
+ "dtype": "bfloat16",
6
  "image_token_id": 32001,
7
  "model_type": "idefics2",
8
  "perceiver_config": {
 
25
  "hidden_size": 16,
26
  "initializer_range": 0.02,
27
  "intermediate_size": 14336,
28
+ "layer_types": null,
29
  "max_position_embeddings": 32768,
30
  "model_type": "mistral",
31
  "num_attention_heads": 4,
 
39
  "vocab_size": 32003
40
  },
41
  "tie_word_embeddings": false,
42
+ "transformers_version": "4.57.0.dev0",
 
43
  "use_cache": true,
44
  "vision_config": {
45
  "attention_dropout": 0.0,
46
+ "embed_dim": 32,
47
  "hidden_act": "gelu_pytorch_tanh",
48
  "hidden_size": 16,
49
  "image_size": 980,
generation_config.json CHANGED
@@ -3,5 +3,5 @@
3
  "bos_token_id": 1,
4
  "eos_token_id": 2,
5
  "pad_token_id": 0,
6
- "transformers_version": "4.56.0.dev0"
7
  }
 
3
  "bos_token_id": 1,
4
  "eos_token_id": 2,
5
  "pad_token_id": 0,
6
+ "transformers_version": "4.57.0.dev0"
7
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:314880cdb4a482295e68fba43f886af05e3e079c85ad8df78dde08e35934ef30
3
  size 7329776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:911a802f439e9545cb62df2754b7152f2348e2fff7098fa518d784f137c9dfb2
3
  size 7329776