planeB commited on
Commit
292b8bc
·
verified ·
1 Parent(s): c03bc2a

Add files using upload-large-folder tool

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ model_name: user_gemma_3_12b_it
4
+ tags:
5
+ - generated_from_trainer
6
+ - sft
7
+ - trl
8
+ licence: license
9
+ ---
10
+
11
+ # Model Card for user_gemma_3_12b_it
12
+
13
+ This model is a fine-tuned version of [None](https://huggingface.co/None).
14
+ It has been trained using [TRL](https://github.com/huggingface/trl).
15
+
16
+ ## Quick start
17
+
18
+ ```python
19
+ from transformers import pipeline
20
+
21
+ 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?"
22
+ generator = pipeline("text-generation", model="None", device="cuda")
23
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
24
+ print(output["generated_text"])
25
+ ```
26
+
27
+ ## Training procedure
28
+
29
+
30
+
31
+
32
+ This model was trained with SFT.
33
+
34
+ ### Framework versions
35
+
36
+ - TRL: 0.25.1
37
+ - Transformers: 4.57.3
38
+ - Pytorch: 2.9.0
39
+ - Datasets: 4.2.0
40
+ - Tokenizers: 0.22.1
41
+
42
+ ## Citations
43
+
44
+
45
+
46
+ Cite TRL as:
47
+
48
+ ```bibtex
49
+ @misc{vonwerra2022trl,
50
+ title = {{TRL: Transformer Reinforcement Learning}},
51
+ 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},
52
+ year = 2020,
53
+ journal = {GitHub repository},
54
+ publisher = {GitHub},
55
+ howpublished = {\url{https://github.com/huggingface/trl}}
56
+ }
57
+ ```
adapter/README.md ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: /leonardo_scratch/large/userexternal/apetruzz/ale_priv/base_models/gemma-3-12b-it
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:/leonardo_scratch/large/userexternal/apetruzz/ale_priv/base_models/gemma-3-12b-it
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ ---
12
+
13
+ # Model Card for Model ID
14
+
15
+ <!-- Provide a quick summary of what the model is/does. -->
16
+
17
+
18
+
19
+ ## Model Details
20
+
21
+ ### Model Description
22
+
23
+ <!-- Provide a longer summary of what this model is. -->
24
+
25
+
26
+
27
+ - **Developed by:** [More Information Needed]
28
+ - **Funded by [optional]:** [More Information Needed]
29
+ - **Shared by [optional]:** [More Information Needed]
30
+ - **Model type:** [More Information Needed]
31
+ - **Language(s) (NLP):** [More Information Needed]
32
+ - **License:** [More Information Needed]
33
+ - **Finetuned from model [optional]:** [More Information Needed]
34
+
35
+ ### Model Sources [optional]
36
+
37
+ <!-- Provide the basic links for the model. -->
38
+
39
+ - **Repository:** [More Information Needed]
40
+ - **Paper [optional]:** [More Information Needed]
41
+ - **Demo [optional]:** [More Information Needed]
42
+
43
+ ## Uses
44
+
45
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
46
+
47
+ ### Direct Use
48
+
49
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
50
+
51
+ [More Information Needed]
52
+
53
+ ### Downstream Use [optional]
54
+
55
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
56
+
57
+ [More Information Needed]
58
+
59
+ ### Out-of-Scope Use
60
+
61
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
62
+
63
+ [More Information Needed]
64
+
65
+ ## Bias, Risks, and Limitations
66
+
67
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
68
+
69
+ [More Information Needed]
70
+
71
+ ### Recommendations
72
+
73
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
74
+
75
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
76
+
77
+ ## How to Get Started with the Model
78
+
79
+ Use the code below to get started with the model.
80
+
81
+ [More Information Needed]
82
+
83
+ ## Training Details
84
+
85
+ ### Training Data
86
+
87
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
88
+
89
+ [More Information Needed]
90
+
91
+ ### Training Procedure
92
+
93
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
94
+
95
+ #### Preprocessing [optional]
96
+
97
+ [More Information Needed]
98
+
99
+
100
+ #### Training Hyperparameters
101
+
102
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
103
+
104
+ #### Speeds, Sizes, Times [optional]
105
+
106
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
107
+
108
+ [More Information Needed]
109
+
110
+ ## Evaluation
111
+
112
+ <!-- This section describes the evaluation protocols and provides the results. -->
113
+
114
+ ### Testing Data, Factors & Metrics
115
+
116
+ #### Testing Data
117
+
118
+ <!-- This should link to a Dataset Card if possible. -->
119
+
120
+ [More Information Needed]
121
+
122
+ #### Factors
123
+
124
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
125
+
126
+ [More Information Needed]
127
+
128
+ #### Metrics
129
+
130
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
131
+
132
+ [More Information Needed]
133
+
134
+ ### Results
135
+
136
+ [More Information Needed]
137
+
138
+ #### Summary
139
+
140
+
141
+
142
+ ## Model Examination [optional]
143
+
144
+ <!-- Relevant interpretability work for the model goes here -->
145
+
146
+ [More Information Needed]
147
+
148
+ ## Environmental Impact
149
+
150
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
151
+
152
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
153
+
154
+ - **Hardware Type:** [More Information Needed]
155
+ - **Hours used:** [More Information Needed]
156
+ - **Cloud Provider:** [More Information Needed]
157
+ - **Compute Region:** [More Information Needed]
158
+ - **Carbon Emitted:** [More Information Needed]
159
+
160
+ ## Technical Specifications [optional]
161
+
162
+ ### Model Architecture and Objective
163
+
164
+ [More Information Needed]
165
+
166
+ ### Compute Infrastructure
167
+
168
+ [More Information Needed]
169
+
170
+ #### Hardware
171
+
172
+ [More Information Needed]
173
+
174
+ #### Software
175
+
176
+ [More Information Needed]
177
+
178
+ ## Citation [optional]
179
+
180
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
181
+
182
+ **BibTeX:**
183
+
184
+ [More Information Needed]
185
+
186
+ **APA:**
187
+
188
+ [More Information Needed]
189
+
190
+ ## Glossary [optional]
191
+
192
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
193
+
194
+ [More Information Needed]
195
+
196
+ ## More Information [optional]
197
+
198
+ [More Information Needed]
199
+
200
+ ## Model Card Authors [optional]
201
+
202
+ [More Information Needed]
203
+
204
+ ## Model Card Contact
205
+
206
+ [More Information Needed]
207
+ ### Framework versions
208
+
209
+ - PEFT 0.17.1
adapter/adapter_config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "/leonardo_scratch/large/userexternal/apetruzz/ale_priv/base_models/gemma-3-12b-it",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": false,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "qalora_group_size": 16,
24
+ "r": 128,
25
+ "rank_pattern": {},
26
+ "revision": null,
27
+ "target_modules": [
28
+ "k_proj",
29
+ "out_proj",
30
+ "up_proj",
31
+ "fc2",
32
+ "gate_proj",
33
+ "q_proj",
34
+ "v_proj",
35
+ "fc1",
36
+ "down_proj",
37
+ "o_proj"
38
+ ],
39
+ "target_parameters": null,
40
+ "task_type": "CAUSAL_LM",
41
+ "trainable_token_indices": null,
42
+ "use_dora": false,
43
+ "use_qalora": false,
44
+ "use_rslora": false
45
+ }
adapter/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:49e1d2fa2f2aad6c7b78769704565cd1c9cb8fb5ff65e500cc95237a728d7b6e
3
+ size 2373462152
chat_template.jinja ADDED
@@ -0,0 +1,84 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{ bos_token }}
2
+ {%- set system_prompt = "You are an advanced user simulator. Your objective is to generate realistic and coherent user responses in a dialogue with a recommendation system based on the user persona, interaction history and target item provided.
3
+ Use the following guidelines:
4
+ 1. Adhere to the user persona: The user's behavior, tone, and responses must align with the specified persona.
5
+ 2. Use the tone inferred from the user's past reviews in the interaction history.
6
+ 3. Ensure that the user's responses are contextually relevant to the ongoing conversation and the target item.
7
+ 4. Maintain coherence and natural flow in the dialogue." -%}
8
+
9
+ {%- set first_user_prefix = system_prompt + '\n\n' -%}
10
+ {%- set loop_messages = messages -%}
11
+
12
+ {%- for message in loop_messages -%}
13
+ {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
14
+ {{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
15
+ {%- endif -%}
16
+ {%- if (message['role'] == 'assistant') -%}
17
+ {%- set role = "model" -%}
18
+ {%- elif (message['role'] == 'system') -%}
19
+ {{ raise_exception("This template is hardcoded and does not accept 'system' messages.") }}
20
+ {%- else -%}
21
+ {%- set role = message['role'] -%}
22
+ {%- endif -%}
23
+ {{ '<start_of_turn>' + role + '\n' + (first_user_prefix if loop.first else "") }}
24
+
25
+ {# --- START: SENIOR ENGINEER REFACTOR FOR DATA PARSING --- #}
26
+ {%- set content_data = message.content -%}
27
+ {%- set is_ninja_payload = false -%}
28
+
29
+ {# Check if content is a string that looks like JSON (starts with { and contains target_item) #}
30
+ {%- if message.content is string -%}
31
+ {%- if message.content | trim | first == '{' and 'target_item' in message.content -%}
32
+ {%- set content_data = message.content | fromjson -%}
33
+ {%- set is_ninja_payload = true -%}
34
+ {%- endif -%}
35
+ {%- elif message.content is mapping -%}
36
+ {# Fallback for internal testing where dicts might be passed directly #}
37
+ {%- set is_ninja_payload = true -%}
38
+ {%- endif -%}
39
+ {# --- END: DATA PARSING --- #}
40
+
41
+ {%- if not is_ninja_payload -%}
42
+ {# This handles regular assistant/model responses or simple user strings #}
43
+ {{ message.content | trim }}
44
+ {%- else -%}
45
+ {# This handles the complex Ninja User Simulator inputs #}
46
+ # Category
47
+ {{ content_data.category | trim }}{{ "\n\n" }}
48
+ # User Persona
49
+ {{ content_data.user_persona | trim }}{{ "\n" }}
50
+ {%- if content_data.interacted_items | length > 0 -%}
51
+ {{ "\n\n" }}# Interaction history
52
+ {%- for item in content_data.interacted_items -%}
53
+ {{ "\n" }}## Item Name: {{ item.item_name | trim }}
54
+ ## Description: {{ item.description | trim }}
55
+ ## Visual Description: {{ item.visual_description | trim }}
56
+ ## Review: {{ item.review | trim }}{{ "\n" }}
57
+ {%- endfor -%}
58
+ {%- endif -%}
59
+ {{ "\n" }}
60
+ # Target item
61
+ ## Item Name: {{ content_data.target_item.item_name | trim }}
62
+ ## Description: {{ content_data.target_item.description | trim }}
63
+ ## Visual Description: {{ content_data.target_item.visual_description | trim }}
64
+ ## Review: {{ content_data.target_item.review | trim }}{{ "\n" }}
65
+ {%- if content_data.messages | length > 0 -%}
66
+ {{ "\n\n" }}# Conversation so far
67
+ {%- for turn in content_data.messages[-2:] -%}
68
+ {%- if turn['role'] == 'assistant' -%}
69
+ {{ "\n" }}CRS: "{{turn['content'] | trim}}"
70
+ {%- else -%}
71
+ {{ "\n" }}USR: "{{turn['content'] | trim}}"
72
+ {%- endif -%}
73
+ {%- endfor -%}
74
+ {{ "\n" }}
75
+ {{ "Generate the next user response based on the above information. Remember to accept only the target item. Reject any other item." }}
76
+ {%- else -%}
77
+ {{ "\nGenerate the first user message based on the above information." }}
78
+ {%- endif -%}
79
+ {%- endif -%}
80
+ {{ '<end_of_turn>\n' }}
81
+ {%- endfor -%}
82
+ {%- if add_generation_prompt -%}
83
+ {{'<start_of_turn>model\n'}}
84
+ {%- endif -%}
config.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma3ForConditionalGeneration"
4
+ ],
5
+ "boi_token_index": 255999,
6
+ "dtype": "float32",
7
+ "eoi_token_index": 256000,
8
+ "eos_token_id": [
9
+ 1,
10
+ 106
11
+ ],
12
+ "image_token_index": 262144,
13
+ "initializer_range": 0.02,
14
+ "mm_tokens_per_image": 256,
15
+ "model_type": "gemma3",
16
+ "text_config": {
17
+ "_sliding_window_pattern": 6,
18
+ "attention_bias": false,
19
+ "attention_dropout": 0.0,
20
+ "attn_logit_softcapping": null,
21
+ "dtype": "float32",
22
+ "final_logit_softcapping": null,
23
+ "head_dim": 256,
24
+ "hidden_activation": "gelu_pytorch_tanh",
25
+ "hidden_size": 3840,
26
+ "initializer_range": 0.02,
27
+ "intermediate_size": 15360,
28
+ "layer_types": [
29
+ "sliding_attention",
30
+ "sliding_attention",
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "full_attention",
35
+ "sliding_attention",
36
+ "sliding_attention",
37
+ "sliding_attention",
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "full_attention",
41
+ "sliding_attention",
42
+ "sliding_attention",
43
+ "sliding_attention",
44
+ "sliding_attention",
45
+ "sliding_attention",
46
+ "full_attention",
47
+ "sliding_attention",
48
+ "sliding_attention",
49
+ "sliding_attention",
50
+ "sliding_attention",
51
+ "sliding_attention",
52
+ "full_attention",
53
+ "sliding_attention",
54
+ "sliding_attention",
55
+ "sliding_attention",
56
+ "sliding_attention",
57
+ "sliding_attention",
58
+ "full_attention",
59
+ "sliding_attention",
60
+ "sliding_attention",
61
+ "sliding_attention",
62
+ "sliding_attention",
63
+ "sliding_attention",
64
+ "full_attention",
65
+ "sliding_attention",
66
+ "sliding_attention",
67
+ "sliding_attention",
68
+ "sliding_attention",
69
+ "sliding_attention",
70
+ "full_attention",
71
+ "sliding_attention",
72
+ "sliding_attention",
73
+ "sliding_attention",
74
+ "sliding_attention",
75
+ "sliding_attention",
76
+ "full_attention"
77
+ ],
78
+ "max_position_embeddings": 131072,
79
+ "model_type": "gemma3_text",
80
+ "num_attention_heads": 16,
81
+ "num_hidden_layers": 48,
82
+ "num_key_value_heads": 8,
83
+ "query_pre_attn_scalar": 256,
84
+ "rms_norm_eps": 1e-06,
85
+ "rope_local_base_freq": 10000.0,
86
+ "rope_scaling": {
87
+ "factor": 8.0,
88
+ "rope_type": "linear"
89
+ },
90
+ "rope_theta": 1000000.0,
91
+ "sliding_window": 1024,
92
+ "use_bidirectional_attention": false,
93
+ "use_cache": true,
94
+ "vocab_size": 262208
95
+ },
96
+ "transformers_version": "4.57.3",
97
+ "vision_config": {
98
+ "attention_dropout": 0.0,
99
+ "dtype": "float32",
100
+ "hidden_act": "gelu_pytorch_tanh",
101
+ "hidden_size": 1152,
102
+ "image_size": 896,
103
+ "intermediate_size": 4304,
104
+ "layer_norm_eps": 1e-06,
105
+ "model_type": "siglip_vision_model",
106
+ "num_attention_heads": 16,
107
+ "num_channels": 3,
108
+ "num_hidden_layers": 27,
109
+ "patch_size": 14,
110
+ "vision_use_head": false
111
+ }
112
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "pad_token_id": 0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "4.57.3"
13
+ }
model-00001-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99057b5bd417ab07add86d2f844f8dee5ad5df5d107838ff61e9d344093d4f13
3
+ size 1685223128
model-00002-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d511e7e1392df02702ca4ec2fc472df37928bb18674c20400ea659dd5ddc694
3
+ size 4987027384
model-00003-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e0522cd83a083dd49cde3f407b9472c3952473bbae512aef60faa2aea865a26
3
+ size 4844749824
model-00004-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44537520bc22c26e0b3574a6b072e409f9dae12ad37208caf5f6a9ae74879e75
3
+ size 4954909736
model-00005-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d12e8a3437d3ed1683c04454c07d2dff4f74dc6b0f1dd9fb35fe1b5e639fe23e
3
+ size 4907664584
model-00006-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:324812785c3b00806888a88fe5744849660705818b2c4540143cabac33d50d88
3
+ size 4954909792
model-00007-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d85b9b7d5bd216d7f6319215089292340b3c93d1c7b74fbb893b40c677d9fbe
3
+ size 4907664584
model-00008-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7c95032614bb0ac94e4f3dd486cf846c95d641ab4d90d518a0ac4212dc195d0
3
+ size 4954909792
model-00009-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e401a970e71a6e12a8244084c1fa1ff1f5164867bb114aa3e01cee27d3074649
3
+ size 4907664584
model-00010-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6a4a9f3078f6aee14b39cca470c8ee607ed9072591c7782695b1f34524ae06b
3
+ size 4954909792
model-00011-of-00011.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:215ba273c28e86561d3ffd6df72652e98f85cb25a20329c462457b56df894dc1
3
+ size 2689808472
model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
special_tokens_map.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "boi_token": "<start_of_image>",
3
+ "bos_token": {
4
+ "content": "<bos>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "eoi_token": "<end_of_image>",
11
+ "eos_token": {
12
+ "content": "<eos>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false
17
+ },
18
+ "image_token": "<image_soft_token>",
19
+ "pad_token": "<eos>",
20
+ "unk_token": {
21
+ "content": "<unk>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ }
27
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
3
+ size 33384568
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff
 
training_config.yaml ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # config.yaml
2
+
3
+ # General settings for the training run
4
+ general:
5
+ output_dir: "/leonardo_scratch/large/userexternal/apetruzz/ale_priv/SpecialIssue/results/v3/user_gemma_3_12b_it" # Directory to save the final model adapters
6
+
7
+ # Model configuration
8
+ model:
9
+ name: "/leonardo_scratch/large/userexternal/apetruzz/ale_priv/base_models/gemma-3-12b-it" # Base model from Hugging Face Hub
10
+ max_seq_length: 2048 # Maximum sequence length for the tokenizer and model
11
+ trust_remote_code: true
12
+ chat_template_file: "/leonardo_work/IscrC_SYMBREC/ale/UserSimTraining/data/chat_template.jinja"
13
+
14
+ # Dataset configuration
15
+ dataset:
16
+ name: "/leonardo_work/IscrC_SYMBREC/ale/UserSimTraining/data/all_processed_prompts_new.jsonl" # Dataset from Hugging Face Hub or local path
17
+ text_field: "prompt" # The name of the column in the dataset that contains the text
18
+
19
+ # PEFT (LoRA) configuration
20
+ peft_config:
21
+ lora_alpha: 32
22
+ lora_dropout: 0.1
23
+ r: 128
24
+ bias: "none"
25
+ task_type: "CAUSAL_LM"
26
+ target_modules: "all-linear"
27
+
28
+ # SFTTrainer-specific arguments
29
+ trainer_args:
30
+ packing: false
31
+
32
+ # Logging configuration
33
+ logging:
34
+ use_wandb: false # Set to true to enable Weights & Biases logging
35
+
36
+ # Hugging Face TrainingArguments
37
+ # See https://huggingface.co/docs/transformers/main_classes/trainer#transformers.TrainingArguments
38
+ training_args:
39
+ num_train_epochs: 5
40
+ per_device_train_batch_size: 8
41
+ gradient_accumulation_steps: 1
42
+ optim: "adamw_torch"
43
+ logging_steps: 25
44
+ learning_rate: 0.0002 # 2e-4
45
+ weight_decay: 0.001
46
+ fp16: false
47
+ bf16: true
48
+ max_grad_norm: 1.0
49
+ max_steps: -1
50
+ warmup_ratio: 0.05
51
+ lr_scheduler_type: "constant"
52
+ #evaluation_strategy: "epoch"
53
+ save_strategy: "epoch"
54
+