2026hbk commited on
Commit
73d430f
·
verified ·
1 Parent(s): fa0e03c

Upload 9 files

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: HuggingFaceTB/SmolVLM2-500M-Video-Instruct
3
+ library_name: transformers
4
+ model_name: vlmfu
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
+ ---
11
+
12
+ # Model Card for vlmfu
13
+
14
+ This model is a fine-tuned version of [HuggingFaceTB/SmolVLM2-500M-Video-Instruct](https://huggingface.co/HuggingFaceTB/SmolVLM2-500M-Video-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
+
17
+ ## Quick start
18
+
19
+ ```python
20
+ from transformers import pipeline
21
+
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="None", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
+
28
+ ## Training procedure
29
+
30
+
31
+
32
+
33
+ This model was trained with SFT.
34
+
35
+ ### Framework versions
36
+
37
+ - TRL: 0.27.1
38
+ - Transformers: 5.0.0
39
+ - Pytorch: 2.10.0+cu129
40
+ - Datasets: 4.5.0
41
+ - Tokenizers: 0.22.2
42
+
43
+ ## Citations
44
+
45
+
46
+
47
+ Cite TRL as:
48
+
49
+ ```bibtex
50
+ @misc{vonwerra2022trl,
51
+ title = {{TRL: Transformer Reinforcement Learning}},
52
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
53
+ year = 2020,
54
+ journal = {GitHub repository},
55
+ publisher = {GitHub},
56
+ howpublished = {\url{https://github.com/huggingface/trl}}
57
+ }
58
+ ```
chat_template.jinja ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ <|im_start|>{% for message in messages %}{{message['role'] | capitalize}}{% if message['content'][0]['type'] == 'image' %}{{':'}}{% else %}{{': '}}{% endif %}{% for line in message['content'] %}{% if line['type'] == 'text' %}{{line['text']}}{% elif line['type'] == 'image' %}{{ '<image>' }}{% endif %}{% endfor %}<end_of_utterance>
2
+ {% endfor %}{% if add_generation_prompt %}{{ 'Assistant:' }}{% endif %}
config.json ADDED
@@ -0,0 +1,164 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "SmolVLMForConditionalGeneration"
4
+ ],
5
+ "bos_token_id": 1,
6
+ "dtype": "bfloat16",
7
+ "eos_token_id": 49279,
8
+ "image_token_id": 49190,
9
+ "model_type": "smolvlm",
10
+ "pad_token_id": 2,
11
+ "scale_factor": 4,
12
+ "text_config": {
13
+ "_flash_attn_2_enabled": true,
14
+ "_name_or_path": "None",
15
+ "architectures": [
16
+ "VLlama3ForCausalLM"
17
+ ],
18
+ "attention_bias": false,
19
+ "attention_dropout": 0.0,
20
+ "bos_token_id": 1,
21
+ "dtype": "bfloat16",
22
+ "eos_token_id": 2,
23
+ "head_dim": 64,
24
+ "hidden_act": "silu",
25
+ "hidden_size": 960,
26
+ "initializer_range": 0.02,
27
+ "intermediate_size": 2560,
28
+ "is_llama_config": true,
29
+ "max_position_embeddings": 8192,
30
+ "mlp_bias": false,
31
+ "model_type": "llama",
32
+ "neftune_noise_alpha": 0.0,
33
+ "num_attention_heads": 15,
34
+ "num_hidden_layers": 32,
35
+ "num_key_value_heads": 5,
36
+ "pad_token_id": 2,
37
+ "perceiver_config": {
38
+ "_attn_implementation_autoset": false,
39
+ "_name_or_path": "",
40
+ "add_cross_attention": false,
41
+ "architectures": null,
42
+ "attention_dropout": 0.0,
43
+ "bad_words_ids": null,
44
+ "begin_suppress_tokens": null,
45
+ "bos_token_id": null,
46
+ "chunk_size_feed_forward": 0,
47
+ "cross_attention_hidden_size": null,
48
+ "decoder_start_token_id": null,
49
+ "diversity_penalty": 0.0,
50
+ "do_sample": false,
51
+ "early_stopping": false,
52
+ "encoder_no_repeat_ngram_size": 0,
53
+ "eos_token_id": null,
54
+ "exponential_decay_length_penalty": null,
55
+ "finetuning_task": null,
56
+ "forced_bos_token_id": null,
57
+ "forced_eos_token_id": null,
58
+ "hidden_act": "silu",
59
+ "id2label": {
60
+ "0": "LABEL_0",
61
+ "1": "LABEL_1"
62
+ },
63
+ "is_decoder": false,
64
+ "is_encoder_decoder": false,
65
+ "label2id": {
66
+ "LABEL_0": 0,
67
+ "LABEL_1": 1
68
+ },
69
+ "length_penalty": 1.0,
70
+ "max_length": 20,
71
+ "min_length": 0,
72
+ "model_type": "vllama3",
73
+ "no_repeat_ngram_size": 0,
74
+ "num_beam_groups": 1,
75
+ "num_beams": 1,
76
+ "num_key_value_heads": 1,
77
+ "num_return_sequences": 1,
78
+ "output_attentions": false,
79
+ "output_hidden_states": false,
80
+ "output_scores": false,
81
+ "pad_token_id": null,
82
+ "prefix": null,
83
+ "problem_type": null,
84
+ "pruned_heads": {},
85
+ "qk_layer_norms_perceiver": false,
86
+ "remove_invalid_values": false,
87
+ "repetition_penalty": 1.0,
88
+ "resampler_depth": 6,
89
+ "resampler_head_dim": 96,
90
+ "resampler_n_heads": 16,
91
+ "resampler_n_latents": 64,
92
+ "return_dict": true,
93
+ "return_dict_in_generate": false,
94
+ "sep_token_id": null,
95
+ "suppress_tokens": null,
96
+ "task_specific_params": null,
97
+ "temperature": 1.0,
98
+ "tf_legacy_loss": false,
99
+ "tie_encoder_decoder": false,
100
+ "tie_word_embeddings": true,
101
+ "tokenizer_class": null,
102
+ "top_k": 50,
103
+ "top_p": 1.0,
104
+ "torch_dtype": null,
105
+ "torchscript": false,
106
+ "transformers_version": "4.46.0",
107
+ "typical_p": 1.0,
108
+ "use_bfloat16": false
109
+ },
110
+ "pixel_shuffle_factor": 4,
111
+ "pretraining_tp": 1,
112
+ "qk_layer_norms": false,
113
+ "rms_norm_eps": 1e-05,
114
+ "rope_interleaved": false,
115
+ "rope_parameters": {
116
+ "rope_theta": 100000,
117
+ "rope_type": "default"
118
+ },
119
+ "tie_word_embeddings": false,
120
+ "transformers.js_config": {
121
+ "kv_cache_dtype": {
122
+ "fp16": "float16",
123
+ "q4f16": "float16"
124
+ }
125
+ },
126
+ "use_cache": true,
127
+ "use_resampler": false,
128
+ "vocab_size": 49280
129
+ },
130
+ "tie_word_embeddings": false,
131
+ "transformers.js_config": {
132
+ "kv_cache_dtype": {
133
+ "fp16": "float16",
134
+ "q4f16": "float16"
135
+ }
136
+ },
137
+ "transformers_version": "5.0.0",
138
+ "use_cache": false,
139
+ "use_reentrant_checkpointing": false,
140
+ "vision_config": {
141
+ "attention_dropout": 0.0,
142
+ "dtype": "bfloat16",
143
+ "hidden_act": "gelu_pytorch_tanh",
144
+ "hidden_size": 768,
145
+ "image_size": 512,
146
+ "initializer_range": 0.02,
147
+ "intermediate_size": 3072,
148
+ "layer_norm_eps": 1e-06,
149
+ "max_image_size": {
150
+ "longest_edge": 512
151
+ },
152
+ "model_type": "smolvlm_vision",
153
+ "num_attention_heads": 12,
154
+ "num_channels": 3,
155
+ "num_hidden_layers": 12,
156
+ "patch_size": 16,
157
+ "size": {
158
+ "longest_edge": 2048
159
+ },
160
+ "tie_word_embeddings": false,
161
+ "use_base_siglip": false
162
+ },
163
+ "vocab_size": 49280
164
+ }
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 1,
4
+ "eos_token_id": [
5
+ 49279,
6
+ 49279
7
+ ],
8
+ "pad_token_id": 2,
9
+ "transformers_version": "5.0.0"
10
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:066294d5273a9295d3a6edc78a49b6245082d7b7141d0f518d1b2baba5299efc
3
+ size 1015025832
processor_config.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "image_processor": {
3
+ "do_convert_rgb": true,
4
+ "do_image_splitting": true,
5
+ "do_normalize": true,
6
+ "do_pad": true,
7
+ "do_rescale": true,
8
+ "do_resize": true,
9
+ "image_mean": [
10
+ 0.5,
11
+ 0.5,
12
+ 0.5
13
+ ],
14
+ "image_processor_type": "SmolVLMImageProcessor",
15
+ "image_std": [
16
+ 0.5,
17
+ 0.5,
18
+ 0.5
19
+ ],
20
+ "max_image_size": {
21
+ "longest_edge": 512
22
+ },
23
+ "resample": 1,
24
+ "rescale_factor": 0.00392156862745098,
25
+ "size": {
26
+ "longest_edge": 2048
27
+ },
28
+ "video_sampling": {
29
+ "fps": 1,
30
+ "max_frames": 64,
31
+ "video_size": {
32
+ "longest_edge": 512
33
+ }
34
+ }
35
+ },
36
+ "image_seq_len": 64,
37
+ "processor_class": "SmolVLMProcessor",
38
+ "video_processor": {
39
+ "data_format": "channels_first",
40
+ "default_to_square": true,
41
+ "do_convert_rgb": true,
42
+ "do_image_splitting": true,
43
+ "do_normalize": true,
44
+ "do_pad": true,
45
+ "do_rescale": true,
46
+ "do_resize": true,
47
+ "do_sample_frames": false,
48
+ "fps": 1,
49
+ "image_mean": [
50
+ 0.5,
51
+ 0.5,
52
+ 0.5
53
+ ],
54
+ "image_processor_type": "SmolVLMImageProcessor",
55
+ "image_std": [
56
+ 0.5,
57
+ 0.5,
58
+ 0.5
59
+ ],
60
+ "max_image_size": {
61
+ "longest_edge": 512
62
+ },
63
+ "num_frames": 64,
64
+ "resample": 1,
65
+ "rescale_factor": 0.00392156862745098,
66
+ "return_metadata": false,
67
+ "size": {
68
+ "longest_edge": 2048
69
+ },
70
+ "video_processor_type": "SmolVLMVideoProcessor",
71
+ "video_sampling": {
72
+ "fps": 1,
73
+ "max_frames": 64,
74
+ "video_size": {
75
+ "longest_edge": 2048
76
+ }
77
+ }
78
+ }
79
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "backend": "tokenizers",
4
+ "bos_token": "<|im_start|>",
5
+ "clean_up_tokenization_spaces": false,
6
+ "end_of_utterance_token": "<end_of_utterance>",
7
+ "eos_token": "<end_of_utterance>",
8
+ "extra_special_tokens": [
9
+ "<fake_token_around_image>",
10
+ "<image>",
11
+ "<end_of_utterance>"
12
+ ],
13
+ "fake_image_token": "<fake_token_around_image>",
14
+ "global_image_token": "<global-img>",
15
+ "image_token": "<image>",
16
+ "is_local": false,
17
+ "legacy": false,
18
+ "model_max_length": 8192,
19
+ "model_specific_special_tokens": {
20
+ "end_of_utterance_token": "<end_of_utterance>",
21
+ "fake_image_token": "<fake_token_around_image>",
22
+ "global_image_token": "<global-img>",
23
+ "image_token": "<image>"
24
+ },
25
+ "pad_token": "<|im_end|>",
26
+ "processor_class": "SmolVLMProcessor",
27
+ "tokenizer_class": "TokenizersBackend",
28
+ "truncation_side": "left",
29
+ "unk_token": "<|endoftext|>",
30
+ "vocab_size": 49152
31
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f354a237fc57bc12f7f78a490f0b9fd8c67b34c09097ad0da6dc6a9edf4cfd50
3
+ size 5649