Initial upload of fine-tuned model
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- .gitattributes +7 -0
- README.md +62 -0
- adapter_config.json +42 -0
- adapter_model.safetensors +3 -0
- added_tokens.json +3 -0
- chat_template.jinja +47 -0
- checkpoint-100/README.md +209 -0
- checkpoint-100/adapter_config.json +42 -0
- checkpoint-100/adapter_model.safetensors +3 -0
- checkpoint-100/added_tokens.json +3 -0
- checkpoint-100/chat_template.jinja +47 -0
- checkpoint-100/optimizer.pt +3 -0
- checkpoint-100/rng_state.pth +3 -0
- checkpoint-100/scaler.pt +3 -0
- checkpoint-100/scheduler.pt +3 -0
- checkpoint-100/special_tokens_map.json +33 -0
- checkpoint-100/tokenizer.json +3 -0
- checkpoint-100/tokenizer.model +3 -0
- checkpoint-100/tokenizer_config.json +0 -0
- checkpoint-100/trainer_state.json +104 -0
- checkpoint-100/training_args.bin +3 -0
- checkpoint-150/README.md +209 -0
- checkpoint-150/adapter_config.json +42 -0
- checkpoint-150/adapter_model.safetensors +3 -0
- checkpoint-150/added_tokens.json +3 -0
- checkpoint-150/chat_template.jinja +47 -0
- checkpoint-150/optimizer.pt +3 -0
- checkpoint-150/rng_state.pth +3 -0
- checkpoint-150/scaler.pt +3 -0
- checkpoint-150/scheduler.pt +3 -0
- checkpoint-150/special_tokens_map.json +33 -0
- checkpoint-150/tokenizer.json +3 -0
- checkpoint-150/tokenizer.model +3 -0
- checkpoint-150/tokenizer_config.json +0 -0
- checkpoint-150/trainer_state.json +139 -0
- checkpoint-150/training_args.bin +3 -0
- checkpoint-200/README.md +209 -0
- checkpoint-200/adapter_config.json +42 -0
- checkpoint-200/adapter_model.safetensors +3 -0
- checkpoint-200/added_tokens.json +3 -0
- checkpoint-200/chat_template.jinja +47 -0
- checkpoint-200/optimizer.pt +3 -0
- checkpoint-200/rng_state.pth +3 -0
- checkpoint-200/scaler.pt +3 -0
- checkpoint-200/scheduler.pt +3 -0
- checkpoint-200/special_tokens_map.json +33 -0
- checkpoint-200/tokenizer.json +3 -0
- checkpoint-200/tokenizer.model +3 -0
- checkpoint-200/tokenizer_config.json +0 -0
- checkpoint-200/trainer_state.json +174 -0
.gitattributes
CHANGED
@@ -33,3 +33,10 @@ 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 |
+
checkpoint-100/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
37 |
+
checkpoint-150/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
38 |
+
checkpoint-200/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
39 |
+
checkpoint-250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
40 |
+
checkpoint-300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
41 |
+
checkpoint-50/tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
42 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: google/gemma-3-4b-it
|
3 |
+
library_name: peft
|
4 |
+
model_name: outputs_sft
|
5 |
+
tags:
|
6 |
+
- base_model:adapter:google/gemma-3-4b-it
|
7 |
+
- lora
|
8 |
+
- sft
|
9 |
+
- transformers
|
10 |
+
- trl
|
11 |
+
licence: license
|
12 |
+
pipeline_tag: text-generation
|
13 |
+
---
|
14 |
+
|
15 |
+
# Model Card for outputs_sft
|
16 |
+
|
17 |
+
This model is a fine-tuned version of [google/gemma-3-4b-it](https://huggingface.co/google/gemma-3-4b-it).
|
18 |
+
It has been trained using [TRL](https://github.com/huggingface/trl).
|
19 |
+
|
20 |
+
## Quick start
|
21 |
+
|
22 |
+
```python
|
23 |
+
from transformers import pipeline
|
24 |
+
|
25 |
+
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?"
|
26 |
+
generator = pipeline("text-generation", model="mikleeeeemmm/outputs_sft", device="cuda")
|
27 |
+
output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
|
28 |
+
print(output["generated_text"])
|
29 |
+
```
|
30 |
+
|
31 |
+
## Training procedure
|
32 |
+
|
33 |
+
|
34 |
+
|
35 |
+
|
36 |
+
This model was trained with SFT.
|
37 |
+
|
38 |
+
### Framework versions
|
39 |
+
|
40 |
+
- PEFT 0.17.0
|
41 |
+
- TRL: 0.21.0
|
42 |
+
- Transformers: 4.55.0
|
43 |
+
- Pytorch: 2.7.1
|
44 |
+
- Datasets: 3.6.0
|
45 |
+
- Tokenizers: 0.21.4
|
46 |
+
|
47 |
+
## Citations
|
48 |
+
|
49 |
+
|
50 |
+
|
51 |
+
Cite TRL as:
|
52 |
+
|
53 |
+
```bibtex
|
54 |
+
@misc{vonwerra2022trl,
|
55 |
+
title = {{TRL: Transformer Reinforcement Learning}},
|
56 |
+
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},
|
57 |
+
year = 2020,
|
58 |
+
journal = {GitHub repository},
|
59 |
+
publisher = {GitHub},
|
60 |
+
howpublished = {\url{https://github.com/huggingface/trl}}
|
61 |
+
}
|
62 |
+
```
|
adapter_config.json
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "google/gemma-3-4b-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.05,
|
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": 16,
|
25 |
+
"rank_pattern": {},
|
26 |
+
"revision": null,
|
27 |
+
"target_modules": [
|
28 |
+
"v_proj",
|
29 |
+
"q_proj",
|
30 |
+
"up_proj",
|
31 |
+
"o_proj",
|
32 |
+
"down_proj",
|
33 |
+
"gate_proj",
|
34 |
+
"k_proj"
|
35 |
+
],
|
36 |
+
"target_parameters": null,
|
37 |
+
"task_type": "CAUSAL_LM",
|
38 |
+
"trainable_token_indices": null,
|
39 |
+
"use_dora": false,
|
40 |
+
"use_qalora": false,
|
41 |
+
"use_rslora": false
|
42 |
+
}
|
adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:651dc8cfd1927592073c0522813bff1c27cfe9bc9ce611153c5bc0ba630dbd80
|
3 |
+
size 131252288
|
added_tokens.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<image_soft_token>": 262144
|
3 |
+
}
|
chat_template.jinja
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{{ bos_token }}
|
2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
3 |
+
{%- if messages[0]['content'] is string -%}
|
4 |
+
{%- set first_user_prefix = messages[0]['content'] + '
|
5 |
+
|
6 |
+
' -%}
|
7 |
+
{%- else -%}
|
8 |
+
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
|
9 |
+
|
10 |
+
' -%}
|
11 |
+
{%- endif -%}
|
12 |
+
{%- set loop_messages = messages[1:] -%}
|
13 |
+
{%- else -%}
|
14 |
+
{%- set first_user_prefix = "" -%}
|
15 |
+
{%- set loop_messages = messages -%}
|
16 |
+
{%- endif -%}
|
17 |
+
{%- for message in loop_messages -%}
|
18 |
+
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
19 |
+
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
20 |
+
{%- endif -%}
|
21 |
+
{%- if (message['role'] == 'assistant') -%}
|
22 |
+
{%- set role = "model" -%}
|
23 |
+
{%- else -%}
|
24 |
+
{%- set role = message['role'] -%}
|
25 |
+
{%- endif -%}
|
26 |
+
{{ '<start_of_turn>' + role + '
|
27 |
+
' + (first_user_prefix if loop.first else "") }}
|
28 |
+
{%- if message['content'] is string -%}
|
29 |
+
{{ message['content'] | trim }}
|
30 |
+
{%- elif message['content'] is iterable -%}
|
31 |
+
{%- for item in message['content'] -%}
|
32 |
+
{%- if item['type'] == 'image' -%}
|
33 |
+
{{ '<start_of_image>' }}
|
34 |
+
{%- elif item['type'] == 'text' -%}
|
35 |
+
{{ item['text'] | trim }}
|
36 |
+
{%- endif -%}
|
37 |
+
{%- endfor -%}
|
38 |
+
{%- else -%}
|
39 |
+
{{ raise_exception("Invalid content type") }}
|
40 |
+
{%- endif -%}
|
41 |
+
{{ '<end_of_turn>
|
42 |
+
' }}
|
43 |
+
{%- endfor -%}
|
44 |
+
{%- if add_generation_prompt -%}
|
45 |
+
{{'<start_of_turn>model
|
46 |
+
'}}
|
47 |
+
{%- endif -%}
|
checkpoint-100/README.md
ADDED
@@ -0,0 +1,209 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: google/gemma-3-4b-it
|
3 |
+
library_name: peft
|
4 |
+
pipeline_tag: text-generation
|
5 |
+
tags:
|
6 |
+
- base_model:adapter:google/gemma-3-4b-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.0
|
checkpoint-100/adapter_config.json
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "google/gemma-3-4b-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.05,
|
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": 16,
|
25 |
+
"rank_pattern": {},
|
26 |
+
"revision": null,
|
27 |
+
"target_modules": [
|
28 |
+
"v_proj",
|
29 |
+
"q_proj",
|
30 |
+
"up_proj",
|
31 |
+
"o_proj",
|
32 |
+
"down_proj",
|
33 |
+
"gate_proj",
|
34 |
+
"k_proj"
|
35 |
+
],
|
36 |
+
"target_parameters": null,
|
37 |
+
"task_type": "CAUSAL_LM",
|
38 |
+
"trainable_token_indices": null,
|
39 |
+
"use_dora": false,
|
40 |
+
"use_qalora": false,
|
41 |
+
"use_rslora": false
|
42 |
+
}
|
checkpoint-100/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d255c11b50114be201fbbe7be3ab4c47c50ea4d0e01872b5cf6276f212def3ee
|
3 |
+
size 131252288
|
checkpoint-100/added_tokens.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<image_soft_token>": 262144
|
3 |
+
}
|
checkpoint-100/chat_template.jinja
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{{ bos_token }}
|
2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
3 |
+
{%- if messages[0]['content'] is string -%}
|
4 |
+
{%- set first_user_prefix = messages[0]['content'] + '
|
5 |
+
|
6 |
+
' -%}
|
7 |
+
{%- else -%}
|
8 |
+
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
|
9 |
+
|
10 |
+
' -%}
|
11 |
+
{%- endif -%}
|
12 |
+
{%- set loop_messages = messages[1:] -%}
|
13 |
+
{%- else -%}
|
14 |
+
{%- set first_user_prefix = "" -%}
|
15 |
+
{%- set loop_messages = messages -%}
|
16 |
+
{%- endif -%}
|
17 |
+
{%- for message in loop_messages -%}
|
18 |
+
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
19 |
+
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
20 |
+
{%- endif -%}
|
21 |
+
{%- if (message['role'] == 'assistant') -%}
|
22 |
+
{%- set role = "model" -%}
|
23 |
+
{%- else -%}
|
24 |
+
{%- set role = message['role'] -%}
|
25 |
+
{%- endif -%}
|
26 |
+
{{ '<start_of_turn>' + role + '
|
27 |
+
' + (first_user_prefix if loop.first else "") }}
|
28 |
+
{%- if message['content'] is string -%}
|
29 |
+
{{ message['content'] | trim }}
|
30 |
+
{%- elif message['content'] is iterable -%}
|
31 |
+
{%- for item in message['content'] -%}
|
32 |
+
{%- if item['type'] == 'image' -%}
|
33 |
+
{{ '<start_of_image>' }}
|
34 |
+
{%- elif item['type'] == 'text' -%}
|
35 |
+
{{ item['text'] | trim }}
|
36 |
+
{%- endif -%}
|
37 |
+
{%- endfor -%}
|
38 |
+
{%- else -%}
|
39 |
+
{{ raise_exception("Invalid content type") }}
|
40 |
+
{%- endif -%}
|
41 |
+
{{ '<end_of_turn>
|
42 |
+
' }}
|
43 |
+
{%- endfor -%}
|
44 |
+
{%- if add_generation_prompt -%}
|
45 |
+
{{'<start_of_turn>model
|
46 |
+
'}}
|
47 |
+
{%- endif -%}
|
checkpoint-100/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ff919f596c28da80a2715756a2a1c29fcf5462acecd1ff4b4351fb7dfc459713
|
3 |
+
size 61099861
|
checkpoint-100/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f50c284cb9bbcfd82ef73423c6bed2032847bf5a24310bbd6a9ffc5a699215fe
|
3 |
+
size 14645
|
checkpoint-100/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9f7d4b8b497c2cddf690bc46c476db7eff892ebcd9b2d10cf2ff7e87dff831bf
|
3 |
+
size 1383
|
checkpoint-100/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5783b952493c351b41574dfbb08aaf3ef54a893508143a1b549c4721d0b0a03a
|
3 |
+
size 1465
|
checkpoint-100/special_tokens_map.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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": {
|
20 |
+
"content": "<pad>",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false
|
25 |
+
},
|
26 |
+
"unk_token": {
|
27 |
+
"content": "<unk>",
|
28 |
+
"lstrip": false,
|
29 |
+
"normalized": false,
|
30 |
+
"rstrip": false,
|
31 |
+
"single_word": false
|
32 |
+
}
|
33 |
+
}
|
checkpoint-100/tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
|
3 |
+
size 33384568
|
checkpoint-100/tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
|
3 |
+
size 4689074
|
checkpoint-100/tokenizer_config.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
checkpoint-100/trainer_state.json
ADDED
@@ -0,0 +1,104 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": null,
|
3 |
+
"best_metric": null,
|
4 |
+
"best_model_checkpoint": null,
|
5 |
+
"epoch": 0.23781212841854935,
|
6 |
+
"eval_steps": 50,
|
7 |
+
"global_step": 100,
|
8 |
+
"is_hyper_param_search": false,
|
9 |
+
"is_local_process_zero": true,
|
10 |
+
"is_world_process_zero": true,
|
11 |
+
"log_history": [
|
12 |
+
{
|
13 |
+
"epoch": 0.023781212841854936,
|
14 |
+
"grad_norm": 11.536224365234375,
|
15 |
+
"learning_rate": 3e-05,
|
16 |
+
"loss": 5.0938,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.04756242568370987,
|
21 |
+
"grad_norm": 3.4980711936950684,
|
22 |
+
"learning_rate": 0.00013000000000000002,
|
23 |
+
"loss": 3.1992,
|
24 |
+
"step": 20
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"epoch": 0.0713436385255648,
|
28 |
+
"grad_norm": 2.206911325454712,
|
29 |
+
"learning_rate": 0.00019951729686242962,
|
30 |
+
"loss": 2.0348,
|
31 |
+
"step": 30
|
32 |
+
},
|
33 |
+
{
|
34 |
+
"epoch": 0.09512485136741974,
|
35 |
+
"grad_norm": 1.7694568634033203,
|
36 |
+
"learning_rate": 0.00019790828640386162,
|
37 |
+
"loss": 1.9145,
|
38 |
+
"step": 40
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"epoch": 0.11890606420927467,
|
42 |
+
"grad_norm": 1.6293410062789917,
|
43 |
+
"learning_rate": 0.00019629927594529365,
|
44 |
+
"loss": 1.7651,
|
45 |
+
"step": 50
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 0.1426872770511296,
|
49 |
+
"grad_norm": 1.8220889568328857,
|
50 |
+
"learning_rate": 0.00019469026548672567,
|
51 |
+
"loss": 1.8897,
|
52 |
+
"step": 60
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 0.16646848989298454,
|
56 |
+
"grad_norm": 1.842227578163147,
|
57 |
+
"learning_rate": 0.0001930812550281577,
|
58 |
+
"loss": 1.8391,
|
59 |
+
"step": 70
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 0.1902497027348395,
|
63 |
+
"grad_norm": 1.6302684545516968,
|
64 |
+
"learning_rate": 0.00019147224456958972,
|
65 |
+
"loss": 1.7881,
|
66 |
+
"step": 80
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 0.2140309155766944,
|
70 |
+
"grad_norm": 1.7633960247039795,
|
71 |
+
"learning_rate": 0.00018986323411102174,
|
72 |
+
"loss": 1.7444,
|
73 |
+
"step": 90
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.23781212841854935,
|
77 |
+
"grad_norm": 1.9522314071655273,
|
78 |
+
"learning_rate": 0.00018825422365245377,
|
79 |
+
"loss": 1.6647,
|
80 |
+
"step": 100
|
81 |
+
}
|
82 |
+
],
|
83 |
+
"logging_steps": 10,
|
84 |
+
"max_steps": 1263,
|
85 |
+
"num_input_tokens_seen": 0,
|
86 |
+
"num_train_epochs": 3,
|
87 |
+
"save_steps": 50,
|
88 |
+
"stateful_callbacks": {
|
89 |
+
"TrainerControl": {
|
90 |
+
"args": {
|
91 |
+
"should_epoch_stop": false,
|
92 |
+
"should_evaluate": false,
|
93 |
+
"should_log": false,
|
94 |
+
"should_save": true,
|
95 |
+
"should_training_stop": false
|
96 |
+
},
|
97 |
+
"attributes": {}
|
98 |
+
}
|
99 |
+
},
|
100 |
+
"total_flos": 1.149219339026304e+16,
|
101 |
+
"train_batch_size": 4,
|
102 |
+
"trial_name": null,
|
103 |
+
"trial_params": null
|
104 |
+
}
|
checkpoint-100/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:451e610a5f98a61245cca0fea2f818a1a84d53989a901e27173e140c01869cc4
|
3 |
+
size 6097
|
checkpoint-150/README.md
ADDED
@@ -0,0 +1,209 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: google/gemma-3-4b-it
|
3 |
+
library_name: peft
|
4 |
+
pipeline_tag: text-generation
|
5 |
+
tags:
|
6 |
+
- base_model:adapter:google/gemma-3-4b-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.0
|
checkpoint-150/adapter_config.json
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "google/gemma-3-4b-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.05,
|
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": 16,
|
25 |
+
"rank_pattern": {},
|
26 |
+
"revision": null,
|
27 |
+
"target_modules": [
|
28 |
+
"v_proj",
|
29 |
+
"q_proj",
|
30 |
+
"up_proj",
|
31 |
+
"o_proj",
|
32 |
+
"down_proj",
|
33 |
+
"gate_proj",
|
34 |
+
"k_proj"
|
35 |
+
],
|
36 |
+
"target_parameters": null,
|
37 |
+
"task_type": "CAUSAL_LM",
|
38 |
+
"trainable_token_indices": null,
|
39 |
+
"use_dora": false,
|
40 |
+
"use_qalora": false,
|
41 |
+
"use_rslora": false
|
42 |
+
}
|
checkpoint-150/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9405b44a587bf08c55dda62e350717b3ed909f781803dd213116ff8b5d40c3cf
|
3 |
+
size 131252288
|
checkpoint-150/added_tokens.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<image_soft_token>": 262144
|
3 |
+
}
|
checkpoint-150/chat_template.jinja
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{{ bos_token }}
|
2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
3 |
+
{%- if messages[0]['content'] is string -%}
|
4 |
+
{%- set first_user_prefix = messages[0]['content'] + '
|
5 |
+
|
6 |
+
' -%}
|
7 |
+
{%- else -%}
|
8 |
+
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
|
9 |
+
|
10 |
+
' -%}
|
11 |
+
{%- endif -%}
|
12 |
+
{%- set loop_messages = messages[1:] -%}
|
13 |
+
{%- else -%}
|
14 |
+
{%- set first_user_prefix = "" -%}
|
15 |
+
{%- set loop_messages = messages -%}
|
16 |
+
{%- endif -%}
|
17 |
+
{%- for message in loop_messages -%}
|
18 |
+
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
19 |
+
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
20 |
+
{%- endif -%}
|
21 |
+
{%- if (message['role'] == 'assistant') -%}
|
22 |
+
{%- set role = "model" -%}
|
23 |
+
{%- else -%}
|
24 |
+
{%- set role = message['role'] -%}
|
25 |
+
{%- endif -%}
|
26 |
+
{{ '<start_of_turn>' + role + '
|
27 |
+
' + (first_user_prefix if loop.first else "") }}
|
28 |
+
{%- if message['content'] is string -%}
|
29 |
+
{{ message['content'] | trim }}
|
30 |
+
{%- elif message['content'] is iterable -%}
|
31 |
+
{%- for item in message['content'] -%}
|
32 |
+
{%- if item['type'] == 'image' -%}
|
33 |
+
{{ '<start_of_image>' }}
|
34 |
+
{%- elif item['type'] == 'text' -%}
|
35 |
+
{{ item['text'] | trim }}
|
36 |
+
{%- endif -%}
|
37 |
+
{%- endfor -%}
|
38 |
+
{%- else -%}
|
39 |
+
{{ raise_exception("Invalid content type") }}
|
40 |
+
{%- endif -%}
|
41 |
+
{{ '<end_of_turn>
|
42 |
+
' }}
|
43 |
+
{%- endfor -%}
|
44 |
+
{%- if add_generation_prompt -%}
|
45 |
+
{{'<start_of_turn>model
|
46 |
+
'}}
|
47 |
+
{%- endif -%}
|
checkpoint-150/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:42c6d89774daecf8c76e2aba18fc2014e7e2201f6f4dd75abc23dee439de0d79
|
3 |
+
size 61099861
|
checkpoint-150/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3b0b056afe0535359faacec98418a3cb3e3997d356fdf4c6111b7620fd2ee7c0
|
3 |
+
size 14645
|
checkpoint-150/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8311ab85799bfbf19d230574fc3fc59a7809e6d38bf92712ff75c30a60f67438
|
3 |
+
size 1383
|
checkpoint-150/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8be59ca525e2d63f726fd6ecdc41f569c8cea219af35b54aad8ebdd68c5059de
|
3 |
+
size 1465
|
checkpoint-150/special_tokens_map.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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": {
|
20 |
+
"content": "<pad>",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false
|
25 |
+
},
|
26 |
+
"unk_token": {
|
27 |
+
"content": "<unk>",
|
28 |
+
"lstrip": false,
|
29 |
+
"normalized": false,
|
30 |
+
"rstrip": false,
|
31 |
+
"single_word": false
|
32 |
+
}
|
33 |
+
}
|
checkpoint-150/tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
|
3 |
+
size 33384568
|
checkpoint-150/tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
|
3 |
+
size 4689074
|
checkpoint-150/tokenizer_config.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
checkpoint-150/trainer_state.json
ADDED
@@ -0,0 +1,139 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": null,
|
3 |
+
"best_metric": null,
|
4 |
+
"best_model_checkpoint": null,
|
5 |
+
"epoch": 0.356718192627824,
|
6 |
+
"eval_steps": 50,
|
7 |
+
"global_step": 150,
|
8 |
+
"is_hyper_param_search": false,
|
9 |
+
"is_local_process_zero": true,
|
10 |
+
"is_world_process_zero": true,
|
11 |
+
"log_history": [
|
12 |
+
{
|
13 |
+
"epoch": 0.023781212841854936,
|
14 |
+
"grad_norm": 11.536224365234375,
|
15 |
+
"learning_rate": 3e-05,
|
16 |
+
"loss": 5.0938,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.04756242568370987,
|
21 |
+
"grad_norm": 3.4980711936950684,
|
22 |
+
"learning_rate": 0.00013000000000000002,
|
23 |
+
"loss": 3.1992,
|
24 |
+
"step": 20
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"epoch": 0.0713436385255648,
|
28 |
+
"grad_norm": 2.206911325454712,
|
29 |
+
"learning_rate": 0.00019951729686242962,
|
30 |
+
"loss": 2.0348,
|
31 |
+
"step": 30
|
32 |
+
},
|
33 |
+
{
|
34 |
+
"epoch": 0.09512485136741974,
|
35 |
+
"grad_norm": 1.7694568634033203,
|
36 |
+
"learning_rate": 0.00019790828640386162,
|
37 |
+
"loss": 1.9145,
|
38 |
+
"step": 40
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"epoch": 0.11890606420927467,
|
42 |
+
"grad_norm": 1.6293410062789917,
|
43 |
+
"learning_rate": 0.00019629927594529365,
|
44 |
+
"loss": 1.7651,
|
45 |
+
"step": 50
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 0.1426872770511296,
|
49 |
+
"grad_norm": 1.8220889568328857,
|
50 |
+
"learning_rate": 0.00019469026548672567,
|
51 |
+
"loss": 1.8897,
|
52 |
+
"step": 60
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 0.16646848989298454,
|
56 |
+
"grad_norm": 1.842227578163147,
|
57 |
+
"learning_rate": 0.0001930812550281577,
|
58 |
+
"loss": 1.8391,
|
59 |
+
"step": 70
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 0.1902497027348395,
|
63 |
+
"grad_norm": 1.6302684545516968,
|
64 |
+
"learning_rate": 0.00019147224456958972,
|
65 |
+
"loss": 1.7881,
|
66 |
+
"step": 80
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 0.2140309155766944,
|
70 |
+
"grad_norm": 1.7633960247039795,
|
71 |
+
"learning_rate": 0.00018986323411102174,
|
72 |
+
"loss": 1.7444,
|
73 |
+
"step": 90
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.23781212841854935,
|
77 |
+
"grad_norm": 1.9522314071655273,
|
78 |
+
"learning_rate": 0.00018825422365245377,
|
79 |
+
"loss": 1.6647,
|
80 |
+
"step": 100
|
81 |
+
},
|
82 |
+
{
|
83 |
+
"epoch": 0.2615933412604043,
|
84 |
+
"grad_norm": 1.5591909885406494,
|
85 |
+
"learning_rate": 0.00018664521319388576,
|
86 |
+
"loss": 1.6839,
|
87 |
+
"step": 110
|
88 |
+
},
|
89 |
+
{
|
90 |
+
"epoch": 0.2853745541022592,
|
91 |
+
"grad_norm": 1.9646824598312378,
|
92 |
+
"learning_rate": 0.00018503620273531779,
|
93 |
+
"loss": 1.7177,
|
94 |
+
"step": 120
|
95 |
+
},
|
96 |
+
{
|
97 |
+
"epoch": 0.3091557669441142,
|
98 |
+
"grad_norm": 2.0045852661132812,
|
99 |
+
"learning_rate": 0.0001834271922767498,
|
100 |
+
"loss": 1.7352,
|
101 |
+
"step": 130
|
102 |
+
},
|
103 |
+
{
|
104 |
+
"epoch": 0.3329369797859691,
|
105 |
+
"grad_norm": 1.6761493682861328,
|
106 |
+
"learning_rate": 0.00018181818181818183,
|
107 |
+
"loss": 1.5562,
|
108 |
+
"step": 140
|
109 |
+
},
|
110 |
+
{
|
111 |
+
"epoch": 0.356718192627824,
|
112 |
+
"grad_norm": 1.720191478729248,
|
113 |
+
"learning_rate": 0.00018020917135961383,
|
114 |
+
"loss": 1.8305,
|
115 |
+
"step": 150
|
116 |
+
}
|
117 |
+
],
|
118 |
+
"logging_steps": 10,
|
119 |
+
"max_steps": 1263,
|
120 |
+
"num_input_tokens_seen": 0,
|
121 |
+
"num_train_epochs": 3,
|
122 |
+
"save_steps": 50,
|
123 |
+
"stateful_callbacks": {
|
124 |
+
"TrainerControl": {
|
125 |
+
"args": {
|
126 |
+
"should_epoch_stop": false,
|
127 |
+
"should_evaluate": false,
|
128 |
+
"should_log": false,
|
129 |
+
"should_save": true,
|
130 |
+
"should_training_stop": false
|
131 |
+
},
|
132 |
+
"attributes": {}
|
133 |
+
}
|
134 |
+
},
|
135 |
+
"total_flos": 1.732465136211264e+16,
|
136 |
+
"train_batch_size": 4,
|
137 |
+
"trial_name": null,
|
138 |
+
"trial_params": null
|
139 |
+
}
|
checkpoint-150/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:451e610a5f98a61245cca0fea2f818a1a84d53989a901e27173e140c01869cc4
|
3 |
+
size 6097
|
checkpoint-200/README.md
ADDED
@@ -0,0 +1,209 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: google/gemma-3-4b-it
|
3 |
+
library_name: peft
|
4 |
+
pipeline_tag: text-generation
|
5 |
+
tags:
|
6 |
+
- base_model:adapter:google/gemma-3-4b-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.0
|
checkpoint-200/adapter_config.json
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "google/gemma-3-4b-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.05,
|
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": 16,
|
25 |
+
"rank_pattern": {},
|
26 |
+
"revision": null,
|
27 |
+
"target_modules": [
|
28 |
+
"v_proj",
|
29 |
+
"q_proj",
|
30 |
+
"up_proj",
|
31 |
+
"o_proj",
|
32 |
+
"down_proj",
|
33 |
+
"gate_proj",
|
34 |
+
"k_proj"
|
35 |
+
],
|
36 |
+
"target_parameters": null,
|
37 |
+
"task_type": "CAUSAL_LM",
|
38 |
+
"trainable_token_indices": null,
|
39 |
+
"use_dora": false,
|
40 |
+
"use_qalora": false,
|
41 |
+
"use_rslora": false
|
42 |
+
}
|
checkpoint-200/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:320d8d7d1e915ae0824d8f6ef4c84f5911b85e49de930e686756814e981f9acc
|
3 |
+
size 131252288
|
checkpoint-200/added_tokens.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"<image_soft_token>": 262144
|
3 |
+
}
|
checkpoint-200/chat_template.jinja
ADDED
@@ -0,0 +1,47 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{{ bos_token }}
|
2 |
+
{%- if messages[0]['role'] == 'system' -%}
|
3 |
+
{%- if messages[0]['content'] is string -%}
|
4 |
+
{%- set first_user_prefix = messages[0]['content'] + '
|
5 |
+
|
6 |
+
' -%}
|
7 |
+
{%- else -%}
|
8 |
+
{%- set first_user_prefix = messages[0]['content'][0]['text'] + '
|
9 |
+
|
10 |
+
' -%}
|
11 |
+
{%- endif -%}
|
12 |
+
{%- set loop_messages = messages[1:] -%}
|
13 |
+
{%- else -%}
|
14 |
+
{%- set first_user_prefix = "" -%}
|
15 |
+
{%- set loop_messages = messages -%}
|
16 |
+
{%- endif -%}
|
17 |
+
{%- for message in loop_messages -%}
|
18 |
+
{%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
|
19 |
+
{{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
|
20 |
+
{%- endif -%}
|
21 |
+
{%- if (message['role'] == 'assistant') -%}
|
22 |
+
{%- set role = "model" -%}
|
23 |
+
{%- else -%}
|
24 |
+
{%- set role = message['role'] -%}
|
25 |
+
{%- endif -%}
|
26 |
+
{{ '<start_of_turn>' + role + '
|
27 |
+
' + (first_user_prefix if loop.first else "") }}
|
28 |
+
{%- if message['content'] is string -%}
|
29 |
+
{{ message['content'] | trim }}
|
30 |
+
{%- elif message['content'] is iterable -%}
|
31 |
+
{%- for item in message['content'] -%}
|
32 |
+
{%- if item['type'] == 'image' -%}
|
33 |
+
{{ '<start_of_image>' }}
|
34 |
+
{%- elif item['type'] == 'text' -%}
|
35 |
+
{{ item['text'] | trim }}
|
36 |
+
{%- endif -%}
|
37 |
+
{%- endfor -%}
|
38 |
+
{%- else -%}
|
39 |
+
{{ raise_exception("Invalid content type") }}
|
40 |
+
{%- endif -%}
|
41 |
+
{{ '<end_of_turn>
|
42 |
+
' }}
|
43 |
+
{%- endfor -%}
|
44 |
+
{%- if add_generation_prompt -%}
|
45 |
+
{{'<start_of_turn>model
|
46 |
+
'}}
|
47 |
+
{%- endif -%}
|
checkpoint-200/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d90142d6453774c5fa16873258da4d4c5ba9db276e0d35c9b574c76a7d8555a5
|
3 |
+
size 61099861
|
checkpoint-200/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ce633e64d01c168f214c2e23e24a979511c5ede6c753527d2b0d7043350aaf9d
|
3 |
+
size 14645
|
checkpoint-200/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a871f91401a0bd517d31d61a5fd51ea6df5bf7fd306387cba6f306497b3414d2
|
3 |
+
size 1383
|
checkpoint-200/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a05f6620774ccc815da5c955d83e92bdb12a4e00d205bcfb1a2609cc3cfc4e96
|
3 |
+
size 1465
|
checkpoint-200/special_tokens_map.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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": {
|
20 |
+
"content": "<pad>",
|
21 |
+
"lstrip": false,
|
22 |
+
"normalized": false,
|
23 |
+
"rstrip": false,
|
24 |
+
"single_word": false
|
25 |
+
},
|
26 |
+
"unk_token": {
|
27 |
+
"content": "<unk>",
|
28 |
+
"lstrip": false,
|
29 |
+
"normalized": false,
|
30 |
+
"rstrip": false,
|
31 |
+
"single_word": false
|
32 |
+
}
|
33 |
+
}
|
checkpoint-200/tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
|
3 |
+
size 33384568
|
checkpoint-200/tokenizer.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
|
3 |
+
size 4689074
|
checkpoint-200/tokenizer_config.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
checkpoint-200/trainer_state.json
ADDED
@@ -0,0 +1,174 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": null,
|
3 |
+
"best_metric": null,
|
4 |
+
"best_model_checkpoint": null,
|
5 |
+
"epoch": 0.4756242568370987,
|
6 |
+
"eval_steps": 50,
|
7 |
+
"global_step": 200,
|
8 |
+
"is_hyper_param_search": false,
|
9 |
+
"is_local_process_zero": true,
|
10 |
+
"is_world_process_zero": true,
|
11 |
+
"log_history": [
|
12 |
+
{
|
13 |
+
"epoch": 0.023781212841854936,
|
14 |
+
"grad_norm": 11.536224365234375,
|
15 |
+
"learning_rate": 3e-05,
|
16 |
+
"loss": 5.0938,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.04756242568370987,
|
21 |
+
"grad_norm": 3.4980711936950684,
|
22 |
+
"learning_rate": 0.00013000000000000002,
|
23 |
+
"loss": 3.1992,
|
24 |
+
"step": 20
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"epoch": 0.0713436385255648,
|
28 |
+
"grad_norm": 2.206911325454712,
|
29 |
+
"learning_rate": 0.00019951729686242962,
|
30 |
+
"loss": 2.0348,
|
31 |
+
"step": 30
|
32 |
+
},
|
33 |
+
{
|
34 |
+
"epoch": 0.09512485136741974,
|
35 |
+
"grad_norm": 1.7694568634033203,
|
36 |
+
"learning_rate": 0.00019790828640386162,
|
37 |
+
"loss": 1.9145,
|
38 |
+
"step": 40
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"epoch": 0.11890606420927467,
|
42 |
+
"grad_norm": 1.6293410062789917,
|
43 |
+
"learning_rate": 0.00019629927594529365,
|
44 |
+
"loss": 1.7651,
|
45 |
+
"step": 50
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 0.1426872770511296,
|
49 |
+
"grad_norm": 1.8220889568328857,
|
50 |
+
"learning_rate": 0.00019469026548672567,
|
51 |
+
"loss": 1.8897,
|
52 |
+
"step": 60
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 0.16646848989298454,
|
56 |
+
"grad_norm": 1.842227578163147,
|
57 |
+
"learning_rate": 0.0001930812550281577,
|
58 |
+
"loss": 1.8391,
|
59 |
+
"step": 70
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 0.1902497027348395,
|
63 |
+
"grad_norm": 1.6302684545516968,
|
64 |
+
"learning_rate": 0.00019147224456958972,
|
65 |
+
"loss": 1.7881,
|
66 |
+
"step": 80
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 0.2140309155766944,
|
70 |
+
"grad_norm": 1.7633960247039795,
|
71 |
+
"learning_rate": 0.00018986323411102174,
|
72 |
+
"loss": 1.7444,
|
73 |
+
"step": 90
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.23781212841854935,
|
77 |
+
"grad_norm": 1.9522314071655273,
|
78 |
+
"learning_rate": 0.00018825422365245377,
|
79 |
+
"loss": 1.6647,
|
80 |
+
"step": 100
|
81 |
+
},
|
82 |
+
{
|
83 |
+
"epoch": 0.2615933412604043,
|
84 |
+
"grad_norm": 1.5591909885406494,
|
85 |
+
"learning_rate": 0.00018664521319388576,
|
86 |
+
"loss": 1.6839,
|
87 |
+
"step": 110
|
88 |
+
},
|
89 |
+
{
|
90 |
+
"epoch": 0.2853745541022592,
|
91 |
+
"grad_norm": 1.9646824598312378,
|
92 |
+
"learning_rate": 0.00018503620273531779,
|
93 |
+
"loss": 1.7177,
|
94 |
+
"step": 120
|
95 |
+
},
|
96 |
+
{
|
97 |
+
"epoch": 0.3091557669441142,
|
98 |
+
"grad_norm": 2.0045852661132812,
|
99 |
+
"learning_rate": 0.0001834271922767498,
|
100 |
+
"loss": 1.7352,
|
101 |
+
"step": 130
|
102 |
+
},
|
103 |
+
{
|
104 |
+
"epoch": 0.3329369797859691,
|
105 |
+
"grad_norm": 1.6761493682861328,
|
106 |
+
"learning_rate": 0.00018181818181818183,
|
107 |
+
"loss": 1.5562,
|
108 |
+
"step": 140
|
109 |
+
},
|
110 |
+
{
|
111 |
+
"epoch": 0.356718192627824,
|
112 |
+
"grad_norm": 1.720191478729248,
|
113 |
+
"learning_rate": 0.00018020917135961383,
|
114 |
+
"loss": 1.8305,
|
115 |
+
"step": 150
|
116 |
+
},
|
117 |
+
{
|
118 |
+
"epoch": 0.380499405469679,
|
119 |
+
"grad_norm": 1.689537763595581,
|
120 |
+
"learning_rate": 0.00017860016090104586,
|
121 |
+
"loss": 1.6121,
|
122 |
+
"step": 160
|
123 |
+
},
|
124 |
+
{
|
125 |
+
"epoch": 0.4042806183115339,
|
126 |
+
"grad_norm": 1.6469579935073853,
|
127 |
+
"learning_rate": 0.0001769911504424779,
|
128 |
+
"loss": 1.5827,
|
129 |
+
"step": 170
|
130 |
+
},
|
131 |
+
{
|
132 |
+
"epoch": 0.4280618311533888,
|
133 |
+
"grad_norm": 1.637831449508667,
|
134 |
+
"learning_rate": 0.0001753821399839099,
|
135 |
+
"loss": 1.6039,
|
136 |
+
"step": 180
|
137 |
+
},
|
138 |
+
{
|
139 |
+
"epoch": 0.4518430439952438,
|
140 |
+
"grad_norm": 2.1786320209503174,
|
141 |
+
"learning_rate": 0.00017377312952534193,
|
142 |
+
"loss": 1.7008,
|
143 |
+
"step": 190
|
144 |
+
},
|
145 |
+
{
|
146 |
+
"epoch": 0.4756242568370987,
|
147 |
+
"grad_norm": 1.634881615638733,
|
148 |
+
"learning_rate": 0.00017216411906677395,
|
149 |
+
"loss": 1.5673,
|
150 |
+
"step": 200
|
151 |
+
}
|
152 |
+
],
|
153 |
+
"logging_steps": 10,
|
154 |
+
"max_steps": 1263,
|
155 |
+
"num_input_tokens_seen": 0,
|
156 |
+
"num_train_epochs": 3,
|
157 |
+
"save_steps": 50,
|
158 |
+
"stateful_callbacks": {
|
159 |
+
"TrainerControl": {
|
160 |
+
"args": {
|
161 |
+
"should_epoch_stop": false,
|
162 |
+
"should_evaluate": false,
|
163 |
+
"should_log": false,
|
164 |
+
"should_save": true,
|
165 |
+
"should_training_stop": false
|
166 |
+
},
|
167 |
+
"attributes": {}
|
168 |
+
}
|
169 |
+
},
|
170 |
+
"total_flos": 2.30437162755072e+16,
|
171 |
+
"train_batch_size": 4,
|
172 |
+
"trial_name": null,
|
173 |
+
"trial_params": null
|
174 |
+
}
|