Spaces:
Running
Running
Перевод на FastAPI POST API
Browse files- .idea/.gitignore +3 -0
- .idea/capybara_fas_ai.iml +8 -0
- .idea/inspectionProfiles/Project_Default.xml +34 -0
- .idea/inspectionProfiles/profiles_settings.xml +6 -0
- .idea/misc.xml +7 -0
- .idea/modules.xml +8 -0
- .idea/vcs.xml +6 -0
- README.md +199 -11
- app.py +37 -0
- capybara-finetuned/.DS_Store +0 -0
- capybara-finetuned/README.md +202 -0
- capybara-finetuned/adapter_config.json +34 -0
- capybara-finetuned/adapter_model.safetensors +3 -0
- capybara-finetuned/checkpoint-193/README.md +202 -0
- capybara-finetuned/checkpoint-193/adapter_config.json +34 -0
- capybara-finetuned/checkpoint-193/adapter_model.safetensors +3 -0
- capybara-finetuned/checkpoint-193/optimizer.pt +3 -0
- capybara-finetuned/checkpoint-193/rng_state.pth +3 -0
- capybara-finetuned/checkpoint-193/scaler.pt +3 -0
- capybara-finetuned/checkpoint-193/scheduler.pt +3 -0
- capybara-finetuned/checkpoint-193/trainer_state.json +167 -0
- capybara-finetuned/checkpoint-193/training_args.bin +3 -0
- capybara-finetuned/checkpoint-386/README.md +202 -0
- capybara-finetuned/checkpoint-386/adapter_config.json +34 -0
- capybara-finetuned/checkpoint-386/adapter_model.safetensors +3 -0
- capybara-finetuned/checkpoint-386/optimizer.pt +3 -0
- capybara-finetuned/checkpoint-386/rng_state.pth +3 -0
- capybara-finetuned/checkpoint-386/scaler.pt +3 -0
- capybara-finetuned/checkpoint-386/scheduler.pt +3 -0
- capybara-finetuned/checkpoint-386/trainer_state.json +300 -0
- capybara-finetuned/checkpoint-386/training_args.bin +3 -0
- capybara-finetuned/checkpoint-579/README.md +202 -0
- capybara-finetuned/checkpoint-579/adapter_config.json +34 -0
- capybara-finetuned/checkpoint-579/adapter_model.safetensors +3 -0
- capybara-finetuned/checkpoint-579/optimizer.pt +3 -0
- capybara-finetuned/checkpoint-579/rng_state.pth +3 -0
- capybara-finetuned/checkpoint-579/scaler.pt +3 -0
- capybara-finetuned/checkpoint-579/scheduler.pt +3 -0
- capybara-finetuned/checkpoint-579/trainer_state.json +433 -0
- capybara-finetuned/checkpoint-579/training_args.bin +3 -0
- capybara-finetuned/special_tokens_map.json +30 -0
- capybara-finetuned/tokenizer.json +0 -0
- capybara-finetuned/tokenizer_config.json +239 -0
- capybara-finetuned/training_args.bin +3 -0
- requirements.txt +7 -0
.idea/.gitignore
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
# Default ignored files
|
2 |
+
/shelf/
|
3 |
+
/workspace.xml
|
.idea/capybara_fas_ai.iml
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
2 |
+
<module type="PYTHON_MODULE" version="4">
|
3 |
+
<component name="NewModuleRootManager">
|
4 |
+
<content url="file://$MODULE_DIR$" />
|
5 |
+
<orderEntry type="jdk" jdkName="Python 3.9 (github_api_test)" jdkType="Python SDK" />
|
6 |
+
<orderEntry type="sourceFolder" forTests="false" />
|
7 |
+
</component>
|
8 |
+
</module>
|
.idea/inspectionProfiles/Project_Default.xml
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<component name="InspectionProjectProfileManager">
|
2 |
+
<profile version="1.0">
|
3 |
+
<option name="myName" value="Project Default" />
|
4 |
+
<inspection_tool class="Annotator" enabled="false" level="ERROR" enabled_by_default="false" />
|
5 |
+
<inspection_tool class="EditorConfigRootDeclarationUniqueness" enabled="false" level="ERROR" enabled_by_default="false" />
|
6 |
+
<inspection_tool class="PyCompatibilityInspection" enabled="true" level="WARNING" enabled_by_default="true">
|
7 |
+
<option name="ourVersions">
|
8 |
+
<value>
|
9 |
+
<list size="1">
|
10 |
+
<item index="0" class="java.lang.String" itemvalue="3.13" />
|
11 |
+
</list>
|
12 |
+
</value>
|
13 |
+
</option>
|
14 |
+
</inspection_tool>
|
15 |
+
<inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
|
16 |
+
<option name="ignoredPackages">
|
17 |
+
<value>
|
18 |
+
<list size="10">
|
19 |
+
<item index="0" class="java.lang.String" itemvalue="httpx" />
|
20 |
+
<item index="1" class="java.lang.String" itemvalue="pydantic" />
|
21 |
+
<item index="2" class="java.lang.String" itemvalue="aiogram" />
|
22 |
+
<item index="3" class="java.lang.String" itemvalue="psycopg2-binary" />
|
23 |
+
<item index="4" class="java.lang.String" itemvalue="python-dotenv" />
|
24 |
+
<item index="5" class="java.lang.String" itemvalue="pillow" />
|
25 |
+
<item index="6" class="java.lang.String" itemvalue="aiohttp" />
|
26 |
+
<item index="7" class="java.lang.String" itemvalue="python-multipart" />
|
27 |
+
<item index="8" class="java.lang.String" itemvalue="gunicorn" />
|
28 |
+
<item index="9" class="java.lang.String" itemvalue="django-formtools" />
|
29 |
+
</list>
|
30 |
+
</value>
|
31 |
+
</option>
|
32 |
+
</inspection_tool>
|
33 |
+
</profile>
|
34 |
+
</component>
|
.idea/inspectionProfiles/profiles_settings.xml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<component name="InspectionProjectProfileManager">
|
2 |
+
<settings>
|
3 |
+
<option name="USE_PROJECT_PROFILE" value="false" />
|
4 |
+
<version value="1.0" />
|
5 |
+
</settings>
|
6 |
+
</component>
|
.idea/misc.xml
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
2 |
+
<project version="4">
|
3 |
+
<component name="Black">
|
4 |
+
<option name="sdkName" value="Python 3.9 (github_api_test)" />
|
5 |
+
</component>
|
6 |
+
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (github_api_test)" project-jdk-type="Python SDK" />
|
7 |
+
</project>
|
.idea/modules.xml
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
2 |
+
<project version="4">
|
3 |
+
<component name="ProjectModuleManager">
|
4 |
+
<modules>
|
5 |
+
<module fileurl="file://$PROJECT_DIR$/.idea/capybara_fas_ai.iml" filepath="$PROJECT_DIR$/.idea/capybara_fas_ai.iml" />
|
6 |
+
</modules>
|
7 |
+
</component>
|
8 |
+
</project>
|
.idea/vcs.xml
ADDED
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<?xml version="1.0" encoding="UTF-8"?>
|
2 |
+
<project version="4">
|
3 |
+
<component name="VcsDirectoryMappings">
|
4 |
+
<mapping directory="" vcs="Git" />
|
5 |
+
</component>
|
6 |
+
</project>
|
README.md
CHANGED
@@ -1,14 +1,202 @@
|
|
1 |
---
|
2 |
-
|
3 |
-
|
4 |
-
colorFrom: blue
|
5 |
-
colorTo: green
|
6 |
-
sdk: gradio
|
7 |
-
sdk_version: 5.25.0
|
8 |
-
app_file: app.py
|
9 |
-
pinned: false
|
10 |
-
license: other
|
11 |
-
short_description: my_ai_1
|
12 |
---
|
13 |
|
14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
base_model: NousResearch/Nous-Capybara-3B-V1.9
|
3 |
+
library_name: peft
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
---
|
5 |
|
6 |
+
# Model Card for Model ID
|
7 |
+
|
8 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Model Description
|
15 |
+
|
16 |
+
<!-- Provide a longer summary of what this model is. -->
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
- **Developed by:** [More Information Needed]
|
21 |
+
- **Funded by [optional]:** [More Information Needed]
|
22 |
+
- **Shared by [optional]:** [More Information Needed]
|
23 |
+
- **Model type:** [More Information Needed]
|
24 |
+
- **Language(s) (NLP):** [More Information Needed]
|
25 |
+
- **License:** [More Information Needed]
|
26 |
+
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
+
|
28 |
+
### Model Sources [optional]
|
29 |
+
|
30 |
+
<!-- Provide the basic links for the model. -->
|
31 |
+
|
32 |
+
- **Repository:** [More Information Needed]
|
33 |
+
- **Paper [optional]:** [More Information Needed]
|
34 |
+
- **Demo [optional]:** [More Information Needed]
|
35 |
+
|
36 |
+
## Uses
|
37 |
+
|
38 |
+
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
39 |
+
|
40 |
+
### Direct Use
|
41 |
+
|
42 |
+
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
43 |
+
|
44 |
+
[More Information Needed]
|
45 |
+
|
46 |
+
### Downstream Use [optional]
|
47 |
+
|
48 |
+
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
49 |
+
|
50 |
+
[More Information Needed]
|
51 |
+
|
52 |
+
### Out-of-Scope Use
|
53 |
+
|
54 |
+
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
55 |
+
|
56 |
+
[More Information Needed]
|
57 |
+
|
58 |
+
## Bias, Risks, and Limitations
|
59 |
+
|
60 |
+
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
61 |
+
|
62 |
+
[More Information Needed]
|
63 |
+
|
64 |
+
### Recommendations
|
65 |
+
|
66 |
+
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
67 |
+
|
68 |
+
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
69 |
+
|
70 |
+
## How to Get Started with the Model
|
71 |
+
|
72 |
+
Use the code below to get started with the model.
|
73 |
+
|
74 |
+
[More Information Needed]
|
75 |
+
|
76 |
+
## Training Details
|
77 |
+
|
78 |
+
### Training Data
|
79 |
+
|
80 |
+
<!-- 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. -->
|
81 |
+
|
82 |
+
[More Information Needed]
|
83 |
+
|
84 |
+
### Training Procedure
|
85 |
+
|
86 |
+
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
87 |
+
|
88 |
+
#### Preprocessing [optional]
|
89 |
+
|
90 |
+
[More Information Needed]
|
91 |
+
|
92 |
+
|
93 |
+
#### Training Hyperparameters
|
94 |
+
|
95 |
+
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
96 |
+
|
97 |
+
#### Speeds, Sizes, Times [optional]
|
98 |
+
|
99 |
+
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
|
100 |
+
|
101 |
+
[More Information Needed]
|
102 |
+
|
103 |
+
## Evaluation
|
104 |
+
|
105 |
+
<!-- This section describes the evaluation protocols and provides the results. -->
|
106 |
+
|
107 |
+
### Testing Data, Factors & Metrics
|
108 |
+
|
109 |
+
#### Testing Data
|
110 |
+
|
111 |
+
<!-- This should link to a Dataset Card if possible. -->
|
112 |
+
|
113 |
+
[More Information Needed]
|
114 |
+
|
115 |
+
#### Factors
|
116 |
+
|
117 |
+
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
118 |
+
|
119 |
+
[More Information Needed]
|
120 |
+
|
121 |
+
#### Metrics
|
122 |
+
|
123 |
+
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
124 |
+
|
125 |
+
[More Information Needed]
|
126 |
+
|
127 |
+
### Results
|
128 |
+
|
129 |
+
[More Information Needed]
|
130 |
+
|
131 |
+
#### Summary
|
132 |
+
|
133 |
+
|
134 |
+
|
135 |
+
## Model Examination [optional]
|
136 |
+
|
137 |
+
<!-- Relevant interpretability work for the model goes here -->
|
138 |
+
|
139 |
+
[More Information Needed]
|
140 |
+
|
141 |
+
## Environmental Impact
|
142 |
+
|
143 |
+
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
144 |
+
|
145 |
+
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).
|
146 |
+
|
147 |
+
- **Hardware Type:** [More Information Needed]
|
148 |
+
- **Hours used:** [More Information Needed]
|
149 |
+
- **Cloud Provider:** [More Information Needed]
|
150 |
+
- **Compute Region:** [More Information Needed]
|
151 |
+
- **Carbon Emitted:** [More Information Needed]
|
152 |
+
|
153 |
+
## Technical Specifications [optional]
|
154 |
+
|
155 |
+
### Model Architecture and Objective
|
156 |
+
|
157 |
+
[More Information Needed]
|
158 |
+
|
159 |
+
### Compute Infrastructure
|
160 |
+
|
161 |
+
[More Information Needed]
|
162 |
+
|
163 |
+
#### Hardware
|
164 |
+
|
165 |
+
[More Information Needed]
|
166 |
+
|
167 |
+
#### Software
|
168 |
+
|
169 |
+
[More Information Needed]
|
170 |
+
|
171 |
+
## Citation [optional]
|
172 |
+
|
173 |
+
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
174 |
+
|
175 |
+
**BibTeX:**
|
176 |
+
|
177 |
+
[More Information Needed]
|
178 |
+
|
179 |
+
**APA:**
|
180 |
+
|
181 |
+
[More Information Needed]
|
182 |
+
|
183 |
+
## Glossary [optional]
|
184 |
+
|
185 |
+
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
186 |
+
|
187 |
+
[More Information Needed]
|
188 |
+
|
189 |
+
## More Information [optional]
|
190 |
+
|
191 |
+
[More Information Needed]
|
192 |
+
|
193 |
+
## Model Card Authors [optional]
|
194 |
+
|
195 |
+
[More Information Needed]
|
196 |
+
|
197 |
+
## Model Card Contact
|
198 |
+
|
199 |
+
[More Information Needed]
|
200 |
+
### Framework versions
|
201 |
+
|
202 |
+
- PEFT 0.14.0
|
app.py
ADDED
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# app.py
|
2 |
+
import os
|
3 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
4 |
+
from fastapi import FastAPI
|
5 |
+
from pydantic import BaseModel
|
6 |
+
|
7 |
+
hf_token = os.getenv("HF_TOKEN")
|
8 |
+
model_path = "./capybara-finetuned" # локальная модель или путь на Hugging Face
|
9 |
+
|
10 |
+
tokenizer = AutoTokenizer.from_pretrained(
|
11 |
+
model_path,
|
12 |
+
token=hf_token,
|
13 |
+
trust_remote_code=True,
|
14 |
+
use_fast=False,
|
15 |
+
)
|
16 |
+
|
17 |
+
model = AutoModelForCausalLM.from_pretrained(
|
18 |
+
model_path,
|
19 |
+
token=hf_token,
|
20 |
+
device_map="auto",
|
21 |
+
torch_dtype="auto",
|
22 |
+
trust_remote_code=True
|
23 |
+
)
|
24 |
+
|
25 |
+
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
26 |
+
|
27 |
+
app = FastAPI()
|
28 |
+
|
29 |
+
class Input(BaseModel):
|
30 |
+
text: str
|
31 |
+
|
32 |
+
@app.post("/classify")
|
33 |
+
async def classify(input: Input):
|
34 |
+
prompt = f"### Вопрос:\n{input.text}\n\n### Класс:"
|
35 |
+
output = pipe(prompt, max_new_tokens=10, do_sample=False)[0]["generated_text"]
|
36 |
+
label = output.split("### Класс:")[-1].strip().split()[0].lower()
|
37 |
+
return {"label": label}
|
capybara-finetuned/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
capybara-finetuned/README.md
ADDED
@@ -0,0 +1,202 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: NousResearch/Nous-Capybara-3B-V1.9
|
3 |
+
library_name: peft
|
4 |
+
---
|
5 |
+
|
6 |
+
# Model Card for Model ID
|
7 |
+
|
8 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Model Description
|
15 |
+
|
16 |
+
<!-- Provide a longer summary of what this model is. -->
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
- **Developed by:** [More Information Needed]
|
21 |
+
- **Funded by [optional]:** [More Information Needed]
|
22 |
+
- **Shared by [optional]:** [More Information Needed]
|
23 |
+
- **Model type:** [More Information Needed]
|
24 |
+
- **Language(s) (NLP):** [More Information Needed]
|
25 |
+
- **License:** [More Information Needed]
|
26 |
+
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
+
|
28 |
+
### Model Sources [optional]
|
29 |
+
|
30 |
+
<!-- Provide the basic links for the model. -->
|
31 |
+
|
32 |
+
- **Repository:** [More Information Needed]
|
33 |
+
- **Paper [optional]:** [More Information Needed]
|
34 |
+
- **Demo [optional]:** [More Information Needed]
|
35 |
+
|
36 |
+
## Uses
|
37 |
+
|
38 |
+
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
39 |
+
|
40 |
+
### Direct Use
|
41 |
+
|
42 |
+
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
43 |
+
|
44 |
+
[More Information Needed]
|
45 |
+
|
46 |
+
### Downstream Use [optional]
|
47 |
+
|
48 |
+
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
49 |
+
|
50 |
+
[More Information Needed]
|
51 |
+
|
52 |
+
### Out-of-Scope Use
|
53 |
+
|
54 |
+
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
55 |
+
|
56 |
+
[More Information Needed]
|
57 |
+
|
58 |
+
## Bias, Risks, and Limitations
|
59 |
+
|
60 |
+
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
61 |
+
|
62 |
+
[More Information Needed]
|
63 |
+
|
64 |
+
### Recommendations
|
65 |
+
|
66 |
+
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
67 |
+
|
68 |
+
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
69 |
+
|
70 |
+
## How to Get Started with the Model
|
71 |
+
|
72 |
+
Use the code below to get started with the model.
|
73 |
+
|
74 |
+
[More Information Needed]
|
75 |
+
|
76 |
+
## Training Details
|
77 |
+
|
78 |
+
### Training Data
|
79 |
+
|
80 |
+
<!-- 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. -->
|
81 |
+
|
82 |
+
[More Information Needed]
|
83 |
+
|
84 |
+
### Training Procedure
|
85 |
+
|
86 |
+
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
87 |
+
|
88 |
+
#### Preprocessing [optional]
|
89 |
+
|
90 |
+
[More Information Needed]
|
91 |
+
|
92 |
+
|
93 |
+
#### Training Hyperparameters
|
94 |
+
|
95 |
+
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
96 |
+
|
97 |
+
#### Speeds, Sizes, Times [optional]
|
98 |
+
|
99 |
+
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
|
100 |
+
|
101 |
+
[More Information Needed]
|
102 |
+
|
103 |
+
## Evaluation
|
104 |
+
|
105 |
+
<!-- This section describes the evaluation protocols and provides the results. -->
|
106 |
+
|
107 |
+
### Testing Data, Factors & Metrics
|
108 |
+
|
109 |
+
#### Testing Data
|
110 |
+
|
111 |
+
<!-- This should link to a Dataset Card if possible. -->
|
112 |
+
|
113 |
+
[More Information Needed]
|
114 |
+
|
115 |
+
#### Factors
|
116 |
+
|
117 |
+
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
118 |
+
|
119 |
+
[More Information Needed]
|
120 |
+
|
121 |
+
#### Metrics
|
122 |
+
|
123 |
+
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
124 |
+
|
125 |
+
[More Information Needed]
|
126 |
+
|
127 |
+
### Results
|
128 |
+
|
129 |
+
[More Information Needed]
|
130 |
+
|
131 |
+
#### Summary
|
132 |
+
|
133 |
+
|
134 |
+
|
135 |
+
## Model Examination [optional]
|
136 |
+
|
137 |
+
<!-- Relevant interpretability work for the model goes here -->
|
138 |
+
|
139 |
+
[More Information Needed]
|
140 |
+
|
141 |
+
## Environmental Impact
|
142 |
+
|
143 |
+
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
144 |
+
|
145 |
+
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).
|
146 |
+
|
147 |
+
- **Hardware Type:** [More Information Needed]
|
148 |
+
- **Hours used:** [More Information Needed]
|
149 |
+
- **Cloud Provider:** [More Information Needed]
|
150 |
+
- **Compute Region:** [More Information Needed]
|
151 |
+
- **Carbon Emitted:** [More Information Needed]
|
152 |
+
|
153 |
+
## Technical Specifications [optional]
|
154 |
+
|
155 |
+
### Model Architecture and Objective
|
156 |
+
|
157 |
+
[More Information Needed]
|
158 |
+
|
159 |
+
### Compute Infrastructure
|
160 |
+
|
161 |
+
[More Information Needed]
|
162 |
+
|
163 |
+
#### Hardware
|
164 |
+
|
165 |
+
[More Information Needed]
|
166 |
+
|
167 |
+
#### Software
|
168 |
+
|
169 |
+
[More Information Needed]
|
170 |
+
|
171 |
+
## Citation [optional]
|
172 |
+
|
173 |
+
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
174 |
+
|
175 |
+
**BibTeX:**
|
176 |
+
|
177 |
+
[More Information Needed]
|
178 |
+
|
179 |
+
**APA:**
|
180 |
+
|
181 |
+
[More Information Needed]
|
182 |
+
|
183 |
+
## Glossary [optional]
|
184 |
+
|
185 |
+
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
186 |
+
|
187 |
+
[More Information Needed]
|
188 |
+
|
189 |
+
## More Information [optional]
|
190 |
+
|
191 |
+
[More Information Needed]
|
192 |
+
|
193 |
+
## Model Card Authors [optional]
|
194 |
+
|
195 |
+
[More Information Needed]
|
196 |
+
|
197 |
+
## Model Card Contact
|
198 |
+
|
199 |
+
[More Information Needed]
|
200 |
+
### Framework versions
|
201 |
+
|
202 |
+
- PEFT 0.14.0
|
capybara-finetuned/adapter_config.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "NousResearch/Nous-Capybara-3B-V1.9",
|
5 |
+
"bias": "none",
|
6 |
+
"eva_config": null,
|
7 |
+
"exclude_modules": null,
|
8 |
+
"fan_in_fan_out": false,
|
9 |
+
"inference_mode": true,
|
10 |
+
"init_lora_weights": true,
|
11 |
+
"layer_replication": null,
|
12 |
+
"layers_pattern": null,
|
13 |
+
"layers_to_transform": null,
|
14 |
+
"loftq_config": {},
|
15 |
+
"lora_alpha": 32,
|
16 |
+
"lora_bias": false,
|
17 |
+
"lora_dropout": 0.05,
|
18 |
+
"megatron_config": null,
|
19 |
+
"megatron_core": "megatron.core",
|
20 |
+
"modules_to_save": null,
|
21 |
+
"peft_type": "LORA",
|
22 |
+
"r": 16,
|
23 |
+
"rank_pattern": {},
|
24 |
+
"revision": null,
|
25 |
+
"target_modules": [
|
26 |
+
"k_proj",
|
27 |
+
"o_proj",
|
28 |
+
"q_proj",
|
29 |
+
"v_proj"
|
30 |
+
],
|
31 |
+
"task_type": "CAUSAL_LM",
|
32 |
+
"use_dora": false,
|
33 |
+
"use_rslora": false
|
34 |
+
}
|
capybara-finetuned/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9c3e58810019f8c2423473124d01849395572ed748ab106aff696e68fe511594
|
3 |
+
size 41977424
|
capybara-finetuned/checkpoint-193/README.md
ADDED
@@ -0,0 +1,202 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: NousResearch/Nous-Capybara-3B-V1.9
|
3 |
+
library_name: peft
|
4 |
+
---
|
5 |
+
|
6 |
+
# Model Card for Model ID
|
7 |
+
|
8 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Model Description
|
15 |
+
|
16 |
+
<!-- Provide a longer summary of what this model is. -->
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
- **Developed by:** [More Information Needed]
|
21 |
+
- **Funded by [optional]:** [More Information Needed]
|
22 |
+
- **Shared by [optional]:** [More Information Needed]
|
23 |
+
- **Model type:** [More Information Needed]
|
24 |
+
- **Language(s) (NLP):** [More Information Needed]
|
25 |
+
- **License:** [More Information Needed]
|
26 |
+
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
+
|
28 |
+
### Model Sources [optional]
|
29 |
+
|
30 |
+
<!-- Provide the basic links for the model. -->
|
31 |
+
|
32 |
+
- **Repository:** [More Information Needed]
|
33 |
+
- **Paper [optional]:** [More Information Needed]
|
34 |
+
- **Demo [optional]:** [More Information Needed]
|
35 |
+
|
36 |
+
## Uses
|
37 |
+
|
38 |
+
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
39 |
+
|
40 |
+
### Direct Use
|
41 |
+
|
42 |
+
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
43 |
+
|
44 |
+
[More Information Needed]
|
45 |
+
|
46 |
+
### Downstream Use [optional]
|
47 |
+
|
48 |
+
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
49 |
+
|
50 |
+
[More Information Needed]
|
51 |
+
|
52 |
+
### Out-of-Scope Use
|
53 |
+
|
54 |
+
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
55 |
+
|
56 |
+
[More Information Needed]
|
57 |
+
|
58 |
+
## Bias, Risks, and Limitations
|
59 |
+
|
60 |
+
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
61 |
+
|
62 |
+
[More Information Needed]
|
63 |
+
|
64 |
+
### Recommendations
|
65 |
+
|
66 |
+
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
67 |
+
|
68 |
+
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
69 |
+
|
70 |
+
## How to Get Started with the Model
|
71 |
+
|
72 |
+
Use the code below to get started with the model.
|
73 |
+
|
74 |
+
[More Information Needed]
|
75 |
+
|
76 |
+
## Training Details
|
77 |
+
|
78 |
+
### Training Data
|
79 |
+
|
80 |
+
<!-- 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. -->
|
81 |
+
|
82 |
+
[More Information Needed]
|
83 |
+
|
84 |
+
### Training Procedure
|
85 |
+
|
86 |
+
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
87 |
+
|
88 |
+
#### Preprocessing [optional]
|
89 |
+
|
90 |
+
[More Information Needed]
|
91 |
+
|
92 |
+
|
93 |
+
#### Training Hyperparameters
|
94 |
+
|
95 |
+
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
96 |
+
|
97 |
+
#### Speeds, Sizes, Times [optional]
|
98 |
+
|
99 |
+
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
|
100 |
+
|
101 |
+
[More Information Needed]
|
102 |
+
|
103 |
+
## Evaluation
|
104 |
+
|
105 |
+
<!-- This section describes the evaluation protocols and provides the results. -->
|
106 |
+
|
107 |
+
### Testing Data, Factors & Metrics
|
108 |
+
|
109 |
+
#### Testing Data
|
110 |
+
|
111 |
+
<!-- This should link to a Dataset Card if possible. -->
|
112 |
+
|
113 |
+
[More Information Needed]
|
114 |
+
|
115 |
+
#### Factors
|
116 |
+
|
117 |
+
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
118 |
+
|
119 |
+
[More Information Needed]
|
120 |
+
|
121 |
+
#### Metrics
|
122 |
+
|
123 |
+
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
124 |
+
|
125 |
+
[More Information Needed]
|
126 |
+
|
127 |
+
### Results
|
128 |
+
|
129 |
+
[More Information Needed]
|
130 |
+
|
131 |
+
#### Summary
|
132 |
+
|
133 |
+
|
134 |
+
|
135 |
+
## Model Examination [optional]
|
136 |
+
|
137 |
+
<!-- Relevant interpretability work for the model goes here -->
|
138 |
+
|
139 |
+
[More Information Needed]
|
140 |
+
|
141 |
+
## Environmental Impact
|
142 |
+
|
143 |
+
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
144 |
+
|
145 |
+
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).
|
146 |
+
|
147 |
+
- **Hardware Type:** [More Information Needed]
|
148 |
+
- **Hours used:** [More Information Needed]
|
149 |
+
- **Cloud Provider:** [More Information Needed]
|
150 |
+
- **Compute Region:** [More Information Needed]
|
151 |
+
- **Carbon Emitted:** [More Information Needed]
|
152 |
+
|
153 |
+
## Technical Specifications [optional]
|
154 |
+
|
155 |
+
### Model Architecture and Objective
|
156 |
+
|
157 |
+
[More Information Needed]
|
158 |
+
|
159 |
+
### Compute Infrastructure
|
160 |
+
|
161 |
+
[More Information Needed]
|
162 |
+
|
163 |
+
#### Hardware
|
164 |
+
|
165 |
+
[More Information Needed]
|
166 |
+
|
167 |
+
#### Software
|
168 |
+
|
169 |
+
[More Information Needed]
|
170 |
+
|
171 |
+
## Citation [optional]
|
172 |
+
|
173 |
+
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
174 |
+
|
175 |
+
**BibTeX:**
|
176 |
+
|
177 |
+
[More Information Needed]
|
178 |
+
|
179 |
+
**APA:**
|
180 |
+
|
181 |
+
[More Information Needed]
|
182 |
+
|
183 |
+
## Glossary [optional]
|
184 |
+
|
185 |
+
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
186 |
+
|
187 |
+
[More Information Needed]
|
188 |
+
|
189 |
+
## More Information [optional]
|
190 |
+
|
191 |
+
[More Information Needed]
|
192 |
+
|
193 |
+
## Model Card Authors [optional]
|
194 |
+
|
195 |
+
[More Information Needed]
|
196 |
+
|
197 |
+
## Model Card Contact
|
198 |
+
|
199 |
+
[More Information Needed]
|
200 |
+
### Framework versions
|
201 |
+
|
202 |
+
- PEFT 0.14.0
|
capybara-finetuned/checkpoint-193/adapter_config.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "NousResearch/Nous-Capybara-3B-V1.9",
|
5 |
+
"bias": "none",
|
6 |
+
"eva_config": null,
|
7 |
+
"exclude_modules": null,
|
8 |
+
"fan_in_fan_out": false,
|
9 |
+
"inference_mode": true,
|
10 |
+
"init_lora_weights": true,
|
11 |
+
"layer_replication": null,
|
12 |
+
"layers_pattern": null,
|
13 |
+
"layers_to_transform": null,
|
14 |
+
"loftq_config": {},
|
15 |
+
"lora_alpha": 32,
|
16 |
+
"lora_bias": false,
|
17 |
+
"lora_dropout": 0.05,
|
18 |
+
"megatron_config": null,
|
19 |
+
"megatron_core": "megatron.core",
|
20 |
+
"modules_to_save": null,
|
21 |
+
"peft_type": "LORA",
|
22 |
+
"r": 16,
|
23 |
+
"rank_pattern": {},
|
24 |
+
"revision": null,
|
25 |
+
"target_modules": [
|
26 |
+
"k_proj",
|
27 |
+
"o_proj",
|
28 |
+
"q_proj",
|
29 |
+
"v_proj"
|
30 |
+
],
|
31 |
+
"task_type": "CAUSAL_LM",
|
32 |
+
"use_dora": false,
|
33 |
+
"use_rslora": false
|
34 |
+
}
|
capybara-finetuned/checkpoint-193/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2c1cffa05984bb9ec4cccf6fe24b4fa1e2f160a679abffee220de9914bf0b4f4
|
3 |
+
size 41977424
|
capybara-finetuned/checkpoint-193/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:de5346d830853d3585860a55a744ab7368789f3985ea18232bd4fc69de13e474
|
3 |
+
size 84100858
|
capybara-finetuned/checkpoint-193/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5875d8cc785fd3e6c6b01f2be69f2dcb4892851d4e48f50fde2b99e161c32ac5
|
3 |
+
size 14244
|
capybara-finetuned/checkpoint-193/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d83b6a26d96339cdb88224ae8d6de87f5142e8f652dddb69b9d2ad537cd43ce1
|
3 |
+
size 988
|
capybara-finetuned/checkpoint-193/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1df4562b25749970667ca99578444d9238c0f6469c2aeca8d78f2e9cb3094c36
|
3 |
+
size 1064
|
capybara-finetuned/checkpoint-193/trainer_state.json
ADDED
@@ -0,0 +1,167 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": null,
|
3 |
+
"best_metric": null,
|
4 |
+
"best_model_checkpoint": null,
|
5 |
+
"epoch": 1.0,
|
6 |
+
"eval_steps": 500,
|
7 |
+
"global_step": 193,
|
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.05181347150259067,
|
14 |
+
"grad_norm": 0.8095124363899231,
|
15 |
+
"learning_rate": 0.00019792746113989637,
|
16 |
+
"loss": 12.3719,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.10362694300518134,
|
21 |
+
"grad_norm": 0.2720714509487152,
|
22 |
+
"learning_rate": 0.00019447322970639035,
|
23 |
+
"loss": 0.2877,
|
24 |
+
"step": 20
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"epoch": 0.15544041450777202,
|
28 |
+
"grad_norm": 0.3335513770580292,
|
29 |
+
"learning_rate": 0.0001910189982728843,
|
30 |
+
"loss": 0.3723,
|
31 |
+
"step": 30
|
32 |
+
},
|
33 |
+
{
|
34 |
+
"epoch": 0.20725388601036268,
|
35 |
+
"grad_norm": 0.21196776628494263,
|
36 |
+
"learning_rate": 0.00018756476683937825,
|
37 |
+
"loss": 0.1598,
|
38 |
+
"step": 40
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"epoch": 0.25906735751295334,
|
42 |
+
"grad_norm": 0.41250184178352356,
|
43 |
+
"learning_rate": 0.0001841105354058722,
|
44 |
+
"loss": 0.1899,
|
45 |
+
"step": 50
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 0.31088082901554404,
|
49 |
+
"grad_norm": 0.2743557095527649,
|
50 |
+
"learning_rate": 0.00018065630397236614,
|
51 |
+
"loss": 0.3541,
|
52 |
+
"step": 60
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 0.3626943005181347,
|
56 |
+
"grad_norm": 0.28188714385032654,
|
57 |
+
"learning_rate": 0.00017720207253886012,
|
58 |
+
"loss": 0.1689,
|
59 |
+
"step": 70
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 0.41450777202072536,
|
63 |
+
"grad_norm": 0.3063858151435852,
|
64 |
+
"learning_rate": 0.00017374784110535407,
|
65 |
+
"loss": 0.2957,
|
66 |
+
"step": 80
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 0.46632124352331605,
|
70 |
+
"grad_norm": 0.24155083298683167,
|
71 |
+
"learning_rate": 0.00017029360967184802,
|
72 |
+
"loss": 0.2411,
|
73 |
+
"step": 90
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.5181347150259067,
|
77 |
+
"grad_norm": 0.3580751121044159,
|
78 |
+
"learning_rate": 0.00016683937823834197,
|
79 |
+
"loss": 0.2035,
|
80 |
+
"step": 100
|
81 |
+
},
|
82 |
+
{
|
83 |
+
"epoch": 0.5699481865284974,
|
84 |
+
"grad_norm": 0.32368239760398865,
|
85 |
+
"learning_rate": 0.00016338514680483594,
|
86 |
+
"loss": 0.2863,
|
87 |
+
"step": 110
|
88 |
+
},
|
89 |
+
{
|
90 |
+
"epoch": 0.6217616580310881,
|
91 |
+
"grad_norm": 0.31771329045295715,
|
92 |
+
"learning_rate": 0.0001599309153713299,
|
93 |
+
"loss": 0.2815,
|
94 |
+
"step": 120
|
95 |
+
},
|
96 |
+
{
|
97 |
+
"epoch": 0.6735751295336787,
|
98 |
+
"grad_norm": 0.1532900333404541,
|
99 |
+
"learning_rate": 0.00015647668393782384,
|
100 |
+
"loss": 0.1949,
|
101 |
+
"step": 130
|
102 |
+
},
|
103 |
+
{
|
104 |
+
"epoch": 0.7253886010362695,
|
105 |
+
"grad_norm": 0.41243669390678406,
|
106 |
+
"learning_rate": 0.0001530224525043178,
|
107 |
+
"loss": 0.1595,
|
108 |
+
"step": 140
|
109 |
+
},
|
110 |
+
{
|
111 |
+
"epoch": 0.7772020725388601,
|
112 |
+
"grad_norm": 0.10424460470676422,
|
113 |
+
"learning_rate": 0.00014956822107081174,
|
114 |
+
"loss": 0.1694,
|
115 |
+
"step": 150
|
116 |
+
},
|
117 |
+
{
|
118 |
+
"epoch": 0.8290155440414507,
|
119 |
+
"grad_norm": 0.2025623470544815,
|
120 |
+
"learning_rate": 0.0001461139896373057,
|
121 |
+
"loss": 0.2369,
|
122 |
+
"step": 160
|
123 |
+
},
|
124 |
+
{
|
125 |
+
"epoch": 0.8808290155440415,
|
126 |
+
"grad_norm": 0.1760733723640442,
|
127 |
+
"learning_rate": 0.00014265975820379966,
|
128 |
+
"loss": 0.3239,
|
129 |
+
"step": 170
|
130 |
+
},
|
131 |
+
{
|
132 |
+
"epoch": 0.9326424870466321,
|
133 |
+
"grad_norm": 0.10223771631717682,
|
134 |
+
"learning_rate": 0.0001392055267702936,
|
135 |
+
"loss": 0.1242,
|
136 |
+
"step": 180
|
137 |
+
},
|
138 |
+
{
|
139 |
+
"epoch": 0.9844559585492227,
|
140 |
+
"grad_norm": 0.4485471844673157,
|
141 |
+
"learning_rate": 0.00013575129533678756,
|
142 |
+
"loss": 0.3755,
|
143 |
+
"step": 190
|
144 |
+
}
|
145 |
+
],
|
146 |
+
"logging_steps": 10,
|
147 |
+
"max_steps": 579,
|
148 |
+
"num_input_tokens_seen": 0,
|
149 |
+
"num_train_epochs": 3,
|
150 |
+
"save_steps": 500,
|
151 |
+
"stateful_callbacks": {
|
152 |
+
"TrainerControl": {
|
153 |
+
"args": {
|
154 |
+
"should_epoch_stop": false,
|
155 |
+
"should_evaluate": false,
|
156 |
+
"should_log": false,
|
157 |
+
"should_save": true,
|
158 |
+
"should_training_stop": false
|
159 |
+
},
|
160 |
+
"attributes": {}
|
161 |
+
}
|
162 |
+
},
|
163 |
+
"total_flos": 3174543906570240.0,
|
164 |
+
"train_batch_size": 1,
|
165 |
+
"trial_name": null,
|
166 |
+
"trial_params": null
|
167 |
+
}
|
capybara-finetuned/checkpoint-193/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5cb5dfa95d8bedc071d0ca429ace88a97d08b4e6c0549993636c23bd1392d3f9
|
3 |
+
size 5304
|
capybara-finetuned/checkpoint-386/README.md
ADDED
@@ -0,0 +1,202 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: NousResearch/Nous-Capybara-3B-V1.9
|
3 |
+
library_name: peft
|
4 |
+
---
|
5 |
+
|
6 |
+
# Model Card for Model ID
|
7 |
+
|
8 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Model Description
|
15 |
+
|
16 |
+
<!-- Provide a longer summary of what this model is. -->
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
- **Developed by:** [More Information Needed]
|
21 |
+
- **Funded by [optional]:** [More Information Needed]
|
22 |
+
- **Shared by [optional]:** [More Information Needed]
|
23 |
+
- **Model type:** [More Information Needed]
|
24 |
+
- **Language(s) (NLP):** [More Information Needed]
|
25 |
+
- **License:** [More Information Needed]
|
26 |
+
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
+
|
28 |
+
### Model Sources [optional]
|
29 |
+
|
30 |
+
<!-- Provide the basic links for the model. -->
|
31 |
+
|
32 |
+
- **Repository:** [More Information Needed]
|
33 |
+
- **Paper [optional]:** [More Information Needed]
|
34 |
+
- **Demo [optional]:** [More Information Needed]
|
35 |
+
|
36 |
+
## Uses
|
37 |
+
|
38 |
+
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
39 |
+
|
40 |
+
### Direct Use
|
41 |
+
|
42 |
+
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
43 |
+
|
44 |
+
[More Information Needed]
|
45 |
+
|
46 |
+
### Downstream Use [optional]
|
47 |
+
|
48 |
+
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
49 |
+
|
50 |
+
[More Information Needed]
|
51 |
+
|
52 |
+
### Out-of-Scope Use
|
53 |
+
|
54 |
+
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
55 |
+
|
56 |
+
[More Information Needed]
|
57 |
+
|
58 |
+
## Bias, Risks, and Limitations
|
59 |
+
|
60 |
+
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
61 |
+
|
62 |
+
[More Information Needed]
|
63 |
+
|
64 |
+
### Recommendations
|
65 |
+
|
66 |
+
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
67 |
+
|
68 |
+
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
69 |
+
|
70 |
+
## How to Get Started with the Model
|
71 |
+
|
72 |
+
Use the code below to get started with the model.
|
73 |
+
|
74 |
+
[More Information Needed]
|
75 |
+
|
76 |
+
## Training Details
|
77 |
+
|
78 |
+
### Training Data
|
79 |
+
|
80 |
+
<!-- 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. -->
|
81 |
+
|
82 |
+
[More Information Needed]
|
83 |
+
|
84 |
+
### Training Procedure
|
85 |
+
|
86 |
+
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
87 |
+
|
88 |
+
#### Preprocessing [optional]
|
89 |
+
|
90 |
+
[More Information Needed]
|
91 |
+
|
92 |
+
|
93 |
+
#### Training Hyperparameters
|
94 |
+
|
95 |
+
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
96 |
+
|
97 |
+
#### Speeds, Sizes, Times [optional]
|
98 |
+
|
99 |
+
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
|
100 |
+
|
101 |
+
[More Information Needed]
|
102 |
+
|
103 |
+
## Evaluation
|
104 |
+
|
105 |
+
<!-- This section describes the evaluation protocols and provides the results. -->
|
106 |
+
|
107 |
+
### Testing Data, Factors & Metrics
|
108 |
+
|
109 |
+
#### Testing Data
|
110 |
+
|
111 |
+
<!-- This should link to a Dataset Card if possible. -->
|
112 |
+
|
113 |
+
[More Information Needed]
|
114 |
+
|
115 |
+
#### Factors
|
116 |
+
|
117 |
+
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
118 |
+
|
119 |
+
[More Information Needed]
|
120 |
+
|
121 |
+
#### Metrics
|
122 |
+
|
123 |
+
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
124 |
+
|
125 |
+
[More Information Needed]
|
126 |
+
|
127 |
+
### Results
|
128 |
+
|
129 |
+
[More Information Needed]
|
130 |
+
|
131 |
+
#### Summary
|
132 |
+
|
133 |
+
|
134 |
+
|
135 |
+
## Model Examination [optional]
|
136 |
+
|
137 |
+
<!-- Relevant interpretability work for the model goes here -->
|
138 |
+
|
139 |
+
[More Information Needed]
|
140 |
+
|
141 |
+
## Environmental Impact
|
142 |
+
|
143 |
+
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
144 |
+
|
145 |
+
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).
|
146 |
+
|
147 |
+
- **Hardware Type:** [More Information Needed]
|
148 |
+
- **Hours used:** [More Information Needed]
|
149 |
+
- **Cloud Provider:** [More Information Needed]
|
150 |
+
- **Compute Region:** [More Information Needed]
|
151 |
+
- **Carbon Emitted:** [More Information Needed]
|
152 |
+
|
153 |
+
## Technical Specifications [optional]
|
154 |
+
|
155 |
+
### Model Architecture and Objective
|
156 |
+
|
157 |
+
[More Information Needed]
|
158 |
+
|
159 |
+
### Compute Infrastructure
|
160 |
+
|
161 |
+
[More Information Needed]
|
162 |
+
|
163 |
+
#### Hardware
|
164 |
+
|
165 |
+
[More Information Needed]
|
166 |
+
|
167 |
+
#### Software
|
168 |
+
|
169 |
+
[More Information Needed]
|
170 |
+
|
171 |
+
## Citation [optional]
|
172 |
+
|
173 |
+
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
174 |
+
|
175 |
+
**BibTeX:**
|
176 |
+
|
177 |
+
[More Information Needed]
|
178 |
+
|
179 |
+
**APA:**
|
180 |
+
|
181 |
+
[More Information Needed]
|
182 |
+
|
183 |
+
## Glossary [optional]
|
184 |
+
|
185 |
+
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
186 |
+
|
187 |
+
[More Information Needed]
|
188 |
+
|
189 |
+
## More Information [optional]
|
190 |
+
|
191 |
+
[More Information Needed]
|
192 |
+
|
193 |
+
## Model Card Authors [optional]
|
194 |
+
|
195 |
+
[More Information Needed]
|
196 |
+
|
197 |
+
## Model Card Contact
|
198 |
+
|
199 |
+
[More Information Needed]
|
200 |
+
### Framework versions
|
201 |
+
|
202 |
+
- PEFT 0.14.0
|
capybara-finetuned/checkpoint-386/adapter_config.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "NousResearch/Nous-Capybara-3B-V1.9",
|
5 |
+
"bias": "none",
|
6 |
+
"eva_config": null,
|
7 |
+
"exclude_modules": null,
|
8 |
+
"fan_in_fan_out": false,
|
9 |
+
"inference_mode": true,
|
10 |
+
"init_lora_weights": true,
|
11 |
+
"layer_replication": null,
|
12 |
+
"layers_pattern": null,
|
13 |
+
"layers_to_transform": null,
|
14 |
+
"loftq_config": {},
|
15 |
+
"lora_alpha": 32,
|
16 |
+
"lora_bias": false,
|
17 |
+
"lora_dropout": 0.05,
|
18 |
+
"megatron_config": null,
|
19 |
+
"megatron_core": "megatron.core",
|
20 |
+
"modules_to_save": null,
|
21 |
+
"peft_type": "LORA",
|
22 |
+
"r": 16,
|
23 |
+
"rank_pattern": {},
|
24 |
+
"revision": null,
|
25 |
+
"target_modules": [
|
26 |
+
"k_proj",
|
27 |
+
"o_proj",
|
28 |
+
"q_proj",
|
29 |
+
"v_proj"
|
30 |
+
],
|
31 |
+
"task_type": "CAUSAL_LM",
|
32 |
+
"use_dora": false,
|
33 |
+
"use_rslora": false
|
34 |
+
}
|
capybara-finetuned/checkpoint-386/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3c48baf1611005f53c51a539c229cfaf642e73bf8e0bd31de3a6fdc5137c2bf4
|
3 |
+
size 41977424
|
capybara-finetuned/checkpoint-386/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0a7ae2f56bbb3d9d1eddcce3fe03c531162565f3bf8fade6f62694e71b727cdd
|
3 |
+
size 84100858
|
capybara-finetuned/checkpoint-386/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1fe550639dad04133d303c68a9bec2bc782de93b44e018edd8ecc0eb84989821
|
3 |
+
size 14244
|
capybara-finetuned/checkpoint-386/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:280ca9c9355adf4466a41f2ee36bbca8aa427ac79a7ddbfe9392315c33de41ad
|
3 |
+
size 988
|
capybara-finetuned/checkpoint-386/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5b59b3a2a0fed8bd1d0e2c1ff88569235401e51bccb1720c6a789060b38d11fd
|
3 |
+
size 1064
|
capybara-finetuned/checkpoint-386/trainer_state.json
ADDED
@@ -0,0 +1,300 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": null,
|
3 |
+
"best_metric": null,
|
4 |
+
"best_model_checkpoint": null,
|
5 |
+
"epoch": 2.0,
|
6 |
+
"eval_steps": 500,
|
7 |
+
"global_step": 386,
|
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.05181347150259067,
|
14 |
+
"grad_norm": 0.8095124363899231,
|
15 |
+
"learning_rate": 0.00019792746113989637,
|
16 |
+
"loss": 12.3719,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.10362694300518134,
|
21 |
+
"grad_norm": 0.2720714509487152,
|
22 |
+
"learning_rate": 0.00019447322970639035,
|
23 |
+
"loss": 0.2877,
|
24 |
+
"step": 20
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"epoch": 0.15544041450777202,
|
28 |
+
"grad_norm": 0.3335513770580292,
|
29 |
+
"learning_rate": 0.0001910189982728843,
|
30 |
+
"loss": 0.3723,
|
31 |
+
"step": 30
|
32 |
+
},
|
33 |
+
{
|
34 |
+
"epoch": 0.20725388601036268,
|
35 |
+
"grad_norm": 0.21196776628494263,
|
36 |
+
"learning_rate": 0.00018756476683937825,
|
37 |
+
"loss": 0.1598,
|
38 |
+
"step": 40
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"epoch": 0.25906735751295334,
|
42 |
+
"grad_norm": 0.41250184178352356,
|
43 |
+
"learning_rate": 0.0001841105354058722,
|
44 |
+
"loss": 0.1899,
|
45 |
+
"step": 50
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 0.31088082901554404,
|
49 |
+
"grad_norm": 0.2743557095527649,
|
50 |
+
"learning_rate": 0.00018065630397236614,
|
51 |
+
"loss": 0.3541,
|
52 |
+
"step": 60
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 0.3626943005181347,
|
56 |
+
"grad_norm": 0.28188714385032654,
|
57 |
+
"learning_rate": 0.00017720207253886012,
|
58 |
+
"loss": 0.1689,
|
59 |
+
"step": 70
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 0.41450777202072536,
|
63 |
+
"grad_norm": 0.3063858151435852,
|
64 |
+
"learning_rate": 0.00017374784110535407,
|
65 |
+
"loss": 0.2957,
|
66 |
+
"step": 80
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 0.46632124352331605,
|
70 |
+
"grad_norm": 0.24155083298683167,
|
71 |
+
"learning_rate": 0.00017029360967184802,
|
72 |
+
"loss": 0.2411,
|
73 |
+
"step": 90
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.5181347150259067,
|
77 |
+
"grad_norm": 0.3580751121044159,
|
78 |
+
"learning_rate": 0.00016683937823834197,
|
79 |
+
"loss": 0.2035,
|
80 |
+
"step": 100
|
81 |
+
},
|
82 |
+
{
|
83 |
+
"epoch": 0.5699481865284974,
|
84 |
+
"grad_norm": 0.32368239760398865,
|
85 |
+
"learning_rate": 0.00016338514680483594,
|
86 |
+
"loss": 0.2863,
|
87 |
+
"step": 110
|
88 |
+
},
|
89 |
+
{
|
90 |
+
"epoch": 0.6217616580310881,
|
91 |
+
"grad_norm": 0.31771329045295715,
|
92 |
+
"learning_rate": 0.0001599309153713299,
|
93 |
+
"loss": 0.2815,
|
94 |
+
"step": 120
|
95 |
+
},
|
96 |
+
{
|
97 |
+
"epoch": 0.6735751295336787,
|
98 |
+
"grad_norm": 0.1532900333404541,
|
99 |
+
"learning_rate": 0.00015647668393782384,
|
100 |
+
"loss": 0.1949,
|
101 |
+
"step": 130
|
102 |
+
},
|
103 |
+
{
|
104 |
+
"epoch": 0.7253886010362695,
|
105 |
+
"grad_norm": 0.41243669390678406,
|
106 |
+
"learning_rate": 0.0001530224525043178,
|
107 |
+
"loss": 0.1595,
|
108 |
+
"step": 140
|
109 |
+
},
|
110 |
+
{
|
111 |
+
"epoch": 0.7772020725388601,
|
112 |
+
"grad_norm": 0.10424460470676422,
|
113 |
+
"learning_rate": 0.00014956822107081174,
|
114 |
+
"loss": 0.1694,
|
115 |
+
"step": 150
|
116 |
+
},
|
117 |
+
{
|
118 |
+
"epoch": 0.8290155440414507,
|
119 |
+
"grad_norm": 0.2025623470544815,
|
120 |
+
"learning_rate": 0.0001461139896373057,
|
121 |
+
"loss": 0.2369,
|
122 |
+
"step": 160
|
123 |
+
},
|
124 |
+
{
|
125 |
+
"epoch": 0.8808290155440415,
|
126 |
+
"grad_norm": 0.1760733723640442,
|
127 |
+
"learning_rate": 0.00014265975820379966,
|
128 |
+
"loss": 0.3239,
|
129 |
+
"step": 170
|
130 |
+
},
|
131 |
+
{
|
132 |
+
"epoch": 0.9326424870466321,
|
133 |
+
"grad_norm": 0.10223771631717682,
|
134 |
+
"learning_rate": 0.0001392055267702936,
|
135 |
+
"loss": 0.1242,
|
136 |
+
"step": 180
|
137 |
+
},
|
138 |
+
{
|
139 |
+
"epoch": 0.9844559585492227,
|
140 |
+
"grad_norm": 0.4485471844673157,
|
141 |
+
"learning_rate": 0.00013575129533678756,
|
142 |
+
"loss": 0.3755,
|
143 |
+
"step": 190
|
144 |
+
},
|
145 |
+
{
|
146 |
+
"epoch": 1.0362694300518134,
|
147 |
+
"grad_norm": 0.39652231335639954,
|
148 |
+
"learning_rate": 0.0001322970639032815,
|
149 |
+
"loss": 0.2445,
|
150 |
+
"step": 200
|
151 |
+
},
|
152 |
+
{
|
153 |
+
"epoch": 1.0880829015544042,
|
154 |
+
"grad_norm": 0.36412283778190613,
|
155 |
+
"learning_rate": 0.00012884283246977548,
|
156 |
+
"loss": 0.2351,
|
157 |
+
"step": 210
|
158 |
+
},
|
159 |
+
{
|
160 |
+
"epoch": 1.1398963730569949,
|
161 |
+
"grad_norm": 0.21277184784412384,
|
162 |
+
"learning_rate": 0.00012538860103626943,
|
163 |
+
"loss": 0.1115,
|
164 |
+
"step": 220
|
165 |
+
},
|
166 |
+
{
|
167 |
+
"epoch": 1.1917098445595855,
|
168 |
+
"grad_norm": 0.4270087480545044,
|
169 |
+
"learning_rate": 0.00012193436960276338,
|
170 |
+
"loss": 0.1656,
|
171 |
+
"step": 230
|
172 |
+
},
|
173 |
+
{
|
174 |
+
"epoch": 1.2435233160621761,
|
175 |
+
"grad_norm": 0.35429275035858154,
|
176 |
+
"learning_rate": 0.00011848013816925734,
|
177 |
+
"loss": 0.2019,
|
178 |
+
"step": 240
|
179 |
+
},
|
180 |
+
{
|
181 |
+
"epoch": 1.2953367875647668,
|
182 |
+
"grad_norm": 0.5731873512268066,
|
183 |
+
"learning_rate": 0.0001150259067357513,
|
184 |
+
"loss": 0.193,
|
185 |
+
"step": 250
|
186 |
+
},
|
187 |
+
{
|
188 |
+
"epoch": 1.3471502590673574,
|
189 |
+
"grad_norm": 0.18436703085899353,
|
190 |
+
"learning_rate": 0.00011157167530224525,
|
191 |
+
"loss": 0.2204,
|
192 |
+
"step": 260
|
193 |
+
},
|
194 |
+
{
|
195 |
+
"epoch": 1.3989637305699483,
|
196 |
+
"grad_norm": 0.6231918334960938,
|
197 |
+
"learning_rate": 0.0001081174438687392,
|
198 |
+
"loss": 0.2204,
|
199 |
+
"step": 270
|
200 |
+
},
|
201 |
+
{
|
202 |
+
"epoch": 1.450777202072539,
|
203 |
+
"grad_norm": 0.27956509590148926,
|
204 |
+
"learning_rate": 0.00010466321243523317,
|
205 |
+
"loss": 0.1864,
|
206 |
+
"step": 280
|
207 |
+
},
|
208 |
+
{
|
209 |
+
"epoch": 1.5025906735751295,
|
210 |
+
"grad_norm": 0.24223612248897552,
|
211 |
+
"learning_rate": 0.00010120898100172711,
|
212 |
+
"loss": 0.1985,
|
213 |
+
"step": 290
|
214 |
+
},
|
215 |
+
{
|
216 |
+
"epoch": 1.5544041450777202,
|
217 |
+
"grad_norm": 0.28999006748199463,
|
218 |
+
"learning_rate": 9.775474956822108e-05,
|
219 |
+
"loss": 0.1322,
|
220 |
+
"step": 300
|
221 |
+
},
|
222 |
+
{
|
223 |
+
"epoch": 1.6062176165803108,
|
224 |
+
"grad_norm": 0.505622923374176,
|
225 |
+
"learning_rate": 9.430051813471503e-05,
|
226 |
+
"loss": 0.1745,
|
227 |
+
"step": 310
|
228 |
+
},
|
229 |
+
{
|
230 |
+
"epoch": 1.6580310880829017,
|
231 |
+
"grad_norm": 0.17805594205856323,
|
232 |
+
"learning_rate": 9.084628670120899e-05,
|
233 |
+
"loss": 0.229,
|
234 |
+
"step": 320
|
235 |
+
},
|
236 |
+
{
|
237 |
+
"epoch": 1.709844559585492,
|
238 |
+
"grad_norm": 0.8911346197128296,
|
239 |
+
"learning_rate": 8.739205526770295e-05,
|
240 |
+
"loss": 0.2056,
|
241 |
+
"step": 330
|
242 |
+
},
|
243 |
+
{
|
244 |
+
"epoch": 1.761658031088083,
|
245 |
+
"grad_norm": 0.2521348297595978,
|
246 |
+
"learning_rate": 8.39378238341969e-05,
|
247 |
+
"loss": 0.2274,
|
248 |
+
"step": 340
|
249 |
+
},
|
250 |
+
{
|
251 |
+
"epoch": 1.8134715025906736,
|
252 |
+
"grad_norm": 0.1958165019750595,
|
253 |
+
"learning_rate": 8.048359240069086e-05,
|
254 |
+
"loss": 0.1346,
|
255 |
+
"step": 350
|
256 |
+
},
|
257 |
+
{
|
258 |
+
"epoch": 1.8652849740932642,
|
259 |
+
"grad_norm": 0.43222665786743164,
|
260 |
+
"learning_rate": 7.702936096718481e-05,
|
261 |
+
"loss": 0.1926,
|
262 |
+
"step": 360
|
263 |
+
},
|
264 |
+
{
|
265 |
+
"epoch": 1.917098445595855,
|
266 |
+
"grad_norm": 0.28354278206825256,
|
267 |
+
"learning_rate": 7.357512953367876e-05,
|
268 |
+
"loss": 0.1491,
|
269 |
+
"step": 370
|
270 |
+
},
|
271 |
+
{
|
272 |
+
"epoch": 1.9689119170984455,
|
273 |
+
"grad_norm": 0.30970773100852966,
|
274 |
+
"learning_rate": 7.012089810017272e-05,
|
275 |
+
"loss": 0.1939,
|
276 |
+
"step": 380
|
277 |
+
}
|
278 |
+
],
|
279 |
+
"logging_steps": 10,
|
280 |
+
"max_steps": 579,
|
281 |
+
"num_input_tokens_seen": 0,
|
282 |
+
"num_train_epochs": 3,
|
283 |
+
"save_steps": 500,
|
284 |
+
"stateful_callbacks": {
|
285 |
+
"TrainerControl": {
|
286 |
+
"args": {
|
287 |
+
"should_epoch_stop": false,
|
288 |
+
"should_evaluate": false,
|
289 |
+
"should_log": false,
|
290 |
+
"should_save": true,
|
291 |
+
"should_training_stop": false
|
292 |
+
},
|
293 |
+
"attributes": {}
|
294 |
+
}
|
295 |
+
},
|
296 |
+
"total_flos": 6349087813140480.0,
|
297 |
+
"train_batch_size": 1,
|
298 |
+
"trial_name": null,
|
299 |
+
"trial_params": null
|
300 |
+
}
|
capybara-finetuned/checkpoint-386/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5cb5dfa95d8bedc071d0ca429ace88a97d08b4e6c0549993636c23bd1392d3f9
|
3 |
+
size 5304
|
capybara-finetuned/checkpoint-579/README.md
ADDED
@@ -0,0 +1,202 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
base_model: NousResearch/Nous-Capybara-3B-V1.9
|
3 |
+
library_name: peft
|
4 |
+
---
|
5 |
+
|
6 |
+
# Model Card for Model ID
|
7 |
+
|
8 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
9 |
+
|
10 |
+
|
11 |
+
|
12 |
+
## Model Details
|
13 |
+
|
14 |
+
### Model Description
|
15 |
+
|
16 |
+
<!-- Provide a longer summary of what this model is. -->
|
17 |
+
|
18 |
+
|
19 |
+
|
20 |
+
- **Developed by:** [More Information Needed]
|
21 |
+
- **Funded by [optional]:** [More Information Needed]
|
22 |
+
- **Shared by [optional]:** [More Information Needed]
|
23 |
+
- **Model type:** [More Information Needed]
|
24 |
+
- **Language(s) (NLP):** [More Information Needed]
|
25 |
+
- **License:** [More Information Needed]
|
26 |
+
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
+
|
28 |
+
### Model Sources [optional]
|
29 |
+
|
30 |
+
<!-- Provide the basic links for the model. -->
|
31 |
+
|
32 |
+
- **Repository:** [More Information Needed]
|
33 |
+
- **Paper [optional]:** [More Information Needed]
|
34 |
+
- **Demo [optional]:** [More Information Needed]
|
35 |
+
|
36 |
+
## Uses
|
37 |
+
|
38 |
+
<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
|
39 |
+
|
40 |
+
### Direct Use
|
41 |
+
|
42 |
+
<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
|
43 |
+
|
44 |
+
[More Information Needed]
|
45 |
+
|
46 |
+
### Downstream Use [optional]
|
47 |
+
|
48 |
+
<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
|
49 |
+
|
50 |
+
[More Information Needed]
|
51 |
+
|
52 |
+
### Out-of-Scope Use
|
53 |
+
|
54 |
+
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
|
55 |
+
|
56 |
+
[More Information Needed]
|
57 |
+
|
58 |
+
## Bias, Risks, and Limitations
|
59 |
+
|
60 |
+
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
61 |
+
|
62 |
+
[More Information Needed]
|
63 |
+
|
64 |
+
### Recommendations
|
65 |
+
|
66 |
+
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
67 |
+
|
68 |
+
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
69 |
+
|
70 |
+
## How to Get Started with the Model
|
71 |
+
|
72 |
+
Use the code below to get started with the model.
|
73 |
+
|
74 |
+
[More Information Needed]
|
75 |
+
|
76 |
+
## Training Details
|
77 |
+
|
78 |
+
### Training Data
|
79 |
+
|
80 |
+
<!-- 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. -->
|
81 |
+
|
82 |
+
[More Information Needed]
|
83 |
+
|
84 |
+
### Training Procedure
|
85 |
+
|
86 |
+
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
|
87 |
+
|
88 |
+
#### Preprocessing [optional]
|
89 |
+
|
90 |
+
[More Information Needed]
|
91 |
+
|
92 |
+
|
93 |
+
#### Training Hyperparameters
|
94 |
+
|
95 |
+
- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
|
96 |
+
|
97 |
+
#### Speeds, Sizes, Times [optional]
|
98 |
+
|
99 |
+
<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
|
100 |
+
|
101 |
+
[More Information Needed]
|
102 |
+
|
103 |
+
## Evaluation
|
104 |
+
|
105 |
+
<!-- This section describes the evaluation protocols and provides the results. -->
|
106 |
+
|
107 |
+
### Testing Data, Factors & Metrics
|
108 |
+
|
109 |
+
#### Testing Data
|
110 |
+
|
111 |
+
<!-- This should link to a Dataset Card if possible. -->
|
112 |
+
|
113 |
+
[More Information Needed]
|
114 |
+
|
115 |
+
#### Factors
|
116 |
+
|
117 |
+
<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
|
118 |
+
|
119 |
+
[More Information Needed]
|
120 |
+
|
121 |
+
#### Metrics
|
122 |
+
|
123 |
+
<!-- These are the evaluation metrics being used, ideally with a description of why. -->
|
124 |
+
|
125 |
+
[More Information Needed]
|
126 |
+
|
127 |
+
### Results
|
128 |
+
|
129 |
+
[More Information Needed]
|
130 |
+
|
131 |
+
#### Summary
|
132 |
+
|
133 |
+
|
134 |
+
|
135 |
+
## Model Examination [optional]
|
136 |
+
|
137 |
+
<!-- Relevant interpretability work for the model goes here -->
|
138 |
+
|
139 |
+
[More Information Needed]
|
140 |
+
|
141 |
+
## Environmental Impact
|
142 |
+
|
143 |
+
<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
|
144 |
+
|
145 |
+
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).
|
146 |
+
|
147 |
+
- **Hardware Type:** [More Information Needed]
|
148 |
+
- **Hours used:** [More Information Needed]
|
149 |
+
- **Cloud Provider:** [More Information Needed]
|
150 |
+
- **Compute Region:** [More Information Needed]
|
151 |
+
- **Carbon Emitted:** [More Information Needed]
|
152 |
+
|
153 |
+
## Technical Specifications [optional]
|
154 |
+
|
155 |
+
### Model Architecture and Objective
|
156 |
+
|
157 |
+
[More Information Needed]
|
158 |
+
|
159 |
+
### Compute Infrastructure
|
160 |
+
|
161 |
+
[More Information Needed]
|
162 |
+
|
163 |
+
#### Hardware
|
164 |
+
|
165 |
+
[More Information Needed]
|
166 |
+
|
167 |
+
#### Software
|
168 |
+
|
169 |
+
[More Information Needed]
|
170 |
+
|
171 |
+
## Citation [optional]
|
172 |
+
|
173 |
+
<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
|
174 |
+
|
175 |
+
**BibTeX:**
|
176 |
+
|
177 |
+
[More Information Needed]
|
178 |
+
|
179 |
+
**APA:**
|
180 |
+
|
181 |
+
[More Information Needed]
|
182 |
+
|
183 |
+
## Glossary [optional]
|
184 |
+
|
185 |
+
<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
|
186 |
+
|
187 |
+
[More Information Needed]
|
188 |
+
|
189 |
+
## More Information [optional]
|
190 |
+
|
191 |
+
[More Information Needed]
|
192 |
+
|
193 |
+
## Model Card Authors [optional]
|
194 |
+
|
195 |
+
[More Information Needed]
|
196 |
+
|
197 |
+
## Model Card Contact
|
198 |
+
|
199 |
+
[More Information Needed]
|
200 |
+
### Framework versions
|
201 |
+
|
202 |
+
- PEFT 0.14.0
|
capybara-finetuned/checkpoint-579/adapter_config.json
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"alpha_pattern": {},
|
3 |
+
"auto_mapping": null,
|
4 |
+
"base_model_name_or_path": "NousResearch/Nous-Capybara-3B-V1.9",
|
5 |
+
"bias": "none",
|
6 |
+
"eva_config": null,
|
7 |
+
"exclude_modules": null,
|
8 |
+
"fan_in_fan_out": false,
|
9 |
+
"inference_mode": true,
|
10 |
+
"init_lora_weights": true,
|
11 |
+
"layer_replication": null,
|
12 |
+
"layers_pattern": null,
|
13 |
+
"layers_to_transform": null,
|
14 |
+
"loftq_config": {},
|
15 |
+
"lora_alpha": 32,
|
16 |
+
"lora_bias": false,
|
17 |
+
"lora_dropout": 0.05,
|
18 |
+
"megatron_config": null,
|
19 |
+
"megatron_core": "megatron.core",
|
20 |
+
"modules_to_save": null,
|
21 |
+
"peft_type": "LORA",
|
22 |
+
"r": 16,
|
23 |
+
"rank_pattern": {},
|
24 |
+
"revision": null,
|
25 |
+
"target_modules": [
|
26 |
+
"k_proj",
|
27 |
+
"o_proj",
|
28 |
+
"q_proj",
|
29 |
+
"v_proj"
|
30 |
+
],
|
31 |
+
"task_type": "CAUSAL_LM",
|
32 |
+
"use_dora": false,
|
33 |
+
"use_rslora": false
|
34 |
+
}
|
capybara-finetuned/checkpoint-579/adapter_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9c3e58810019f8c2423473124d01849395572ed748ab106aff696e68fe511594
|
3 |
+
size 41977424
|
capybara-finetuned/checkpoint-579/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1ae81316b6c06fe5011bed6387e38d2e9e6d59b269947e0e98a135f250e53431
|
3 |
+
size 84100858
|
capybara-finetuned/checkpoint-579/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9be5bd0f0585baf47114de4868010f1d69cccda30c0451f6ffa7a6656210a61c
|
3 |
+
size 14244
|
capybara-finetuned/checkpoint-579/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b70d513d90f98dc0b17bacf317603627627da3ea2748eb84b7ab7f4d8ec7a878
|
3 |
+
size 988
|
capybara-finetuned/checkpoint-579/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:180e243f8de76fb954da6fdb3fad5a9f4cc7a2aa5ac46cf1ccaa53c9bb3946ef
|
3 |
+
size 1064
|
capybara-finetuned/checkpoint-579/trainer_state.json
ADDED
@@ -0,0 +1,433 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_global_step": null,
|
3 |
+
"best_metric": null,
|
4 |
+
"best_model_checkpoint": null,
|
5 |
+
"epoch": 3.0,
|
6 |
+
"eval_steps": 500,
|
7 |
+
"global_step": 579,
|
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.05181347150259067,
|
14 |
+
"grad_norm": 0.8095124363899231,
|
15 |
+
"learning_rate": 0.00019792746113989637,
|
16 |
+
"loss": 12.3719,
|
17 |
+
"step": 10
|
18 |
+
},
|
19 |
+
{
|
20 |
+
"epoch": 0.10362694300518134,
|
21 |
+
"grad_norm": 0.2720714509487152,
|
22 |
+
"learning_rate": 0.00019447322970639035,
|
23 |
+
"loss": 0.2877,
|
24 |
+
"step": 20
|
25 |
+
},
|
26 |
+
{
|
27 |
+
"epoch": 0.15544041450777202,
|
28 |
+
"grad_norm": 0.3335513770580292,
|
29 |
+
"learning_rate": 0.0001910189982728843,
|
30 |
+
"loss": 0.3723,
|
31 |
+
"step": 30
|
32 |
+
},
|
33 |
+
{
|
34 |
+
"epoch": 0.20725388601036268,
|
35 |
+
"grad_norm": 0.21196776628494263,
|
36 |
+
"learning_rate": 0.00018756476683937825,
|
37 |
+
"loss": 0.1598,
|
38 |
+
"step": 40
|
39 |
+
},
|
40 |
+
{
|
41 |
+
"epoch": 0.25906735751295334,
|
42 |
+
"grad_norm": 0.41250184178352356,
|
43 |
+
"learning_rate": 0.0001841105354058722,
|
44 |
+
"loss": 0.1899,
|
45 |
+
"step": 50
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 0.31088082901554404,
|
49 |
+
"grad_norm": 0.2743557095527649,
|
50 |
+
"learning_rate": 0.00018065630397236614,
|
51 |
+
"loss": 0.3541,
|
52 |
+
"step": 60
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 0.3626943005181347,
|
56 |
+
"grad_norm": 0.28188714385032654,
|
57 |
+
"learning_rate": 0.00017720207253886012,
|
58 |
+
"loss": 0.1689,
|
59 |
+
"step": 70
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 0.41450777202072536,
|
63 |
+
"grad_norm": 0.3063858151435852,
|
64 |
+
"learning_rate": 0.00017374784110535407,
|
65 |
+
"loss": 0.2957,
|
66 |
+
"step": 80
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 0.46632124352331605,
|
70 |
+
"grad_norm": 0.24155083298683167,
|
71 |
+
"learning_rate": 0.00017029360967184802,
|
72 |
+
"loss": 0.2411,
|
73 |
+
"step": 90
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 0.5181347150259067,
|
77 |
+
"grad_norm": 0.3580751121044159,
|
78 |
+
"learning_rate": 0.00016683937823834197,
|
79 |
+
"loss": 0.2035,
|
80 |
+
"step": 100
|
81 |
+
},
|
82 |
+
{
|
83 |
+
"epoch": 0.5699481865284974,
|
84 |
+
"grad_norm": 0.32368239760398865,
|
85 |
+
"learning_rate": 0.00016338514680483594,
|
86 |
+
"loss": 0.2863,
|
87 |
+
"step": 110
|
88 |
+
},
|
89 |
+
{
|
90 |
+
"epoch": 0.6217616580310881,
|
91 |
+
"grad_norm": 0.31771329045295715,
|
92 |
+
"learning_rate": 0.0001599309153713299,
|
93 |
+
"loss": 0.2815,
|
94 |
+
"step": 120
|
95 |
+
},
|
96 |
+
{
|
97 |
+
"epoch": 0.6735751295336787,
|
98 |
+
"grad_norm": 0.1532900333404541,
|
99 |
+
"learning_rate": 0.00015647668393782384,
|
100 |
+
"loss": 0.1949,
|
101 |
+
"step": 130
|
102 |
+
},
|
103 |
+
{
|
104 |
+
"epoch": 0.7253886010362695,
|
105 |
+
"grad_norm": 0.41243669390678406,
|
106 |
+
"learning_rate": 0.0001530224525043178,
|
107 |
+
"loss": 0.1595,
|
108 |
+
"step": 140
|
109 |
+
},
|
110 |
+
{
|
111 |
+
"epoch": 0.7772020725388601,
|
112 |
+
"grad_norm": 0.10424460470676422,
|
113 |
+
"learning_rate": 0.00014956822107081174,
|
114 |
+
"loss": 0.1694,
|
115 |
+
"step": 150
|
116 |
+
},
|
117 |
+
{
|
118 |
+
"epoch": 0.8290155440414507,
|
119 |
+
"grad_norm": 0.2025623470544815,
|
120 |
+
"learning_rate": 0.0001461139896373057,
|
121 |
+
"loss": 0.2369,
|
122 |
+
"step": 160
|
123 |
+
},
|
124 |
+
{
|
125 |
+
"epoch": 0.8808290155440415,
|
126 |
+
"grad_norm": 0.1760733723640442,
|
127 |
+
"learning_rate": 0.00014265975820379966,
|
128 |
+
"loss": 0.3239,
|
129 |
+
"step": 170
|
130 |
+
},
|
131 |
+
{
|
132 |
+
"epoch": 0.9326424870466321,
|
133 |
+
"grad_norm": 0.10223771631717682,
|
134 |
+
"learning_rate": 0.0001392055267702936,
|
135 |
+
"loss": 0.1242,
|
136 |
+
"step": 180
|
137 |
+
},
|
138 |
+
{
|
139 |
+
"epoch": 0.9844559585492227,
|
140 |
+
"grad_norm": 0.4485471844673157,
|
141 |
+
"learning_rate": 0.00013575129533678756,
|
142 |
+
"loss": 0.3755,
|
143 |
+
"step": 190
|
144 |
+
},
|
145 |
+
{
|
146 |
+
"epoch": 1.0362694300518134,
|
147 |
+
"grad_norm": 0.39652231335639954,
|
148 |
+
"learning_rate": 0.0001322970639032815,
|
149 |
+
"loss": 0.2445,
|
150 |
+
"step": 200
|
151 |
+
},
|
152 |
+
{
|
153 |
+
"epoch": 1.0880829015544042,
|
154 |
+
"grad_norm": 0.36412283778190613,
|
155 |
+
"learning_rate": 0.00012884283246977548,
|
156 |
+
"loss": 0.2351,
|
157 |
+
"step": 210
|
158 |
+
},
|
159 |
+
{
|
160 |
+
"epoch": 1.1398963730569949,
|
161 |
+
"grad_norm": 0.21277184784412384,
|
162 |
+
"learning_rate": 0.00012538860103626943,
|
163 |
+
"loss": 0.1115,
|
164 |
+
"step": 220
|
165 |
+
},
|
166 |
+
{
|
167 |
+
"epoch": 1.1917098445595855,
|
168 |
+
"grad_norm": 0.4270087480545044,
|
169 |
+
"learning_rate": 0.00012193436960276338,
|
170 |
+
"loss": 0.1656,
|
171 |
+
"step": 230
|
172 |
+
},
|
173 |
+
{
|
174 |
+
"epoch": 1.2435233160621761,
|
175 |
+
"grad_norm": 0.35429275035858154,
|
176 |
+
"learning_rate": 0.00011848013816925734,
|
177 |
+
"loss": 0.2019,
|
178 |
+
"step": 240
|
179 |
+
},
|
180 |
+
{
|
181 |
+
"epoch": 1.2953367875647668,
|
182 |
+
"grad_norm": 0.5731873512268066,
|
183 |
+
"learning_rate": 0.0001150259067357513,
|
184 |
+
"loss": 0.193,
|
185 |
+
"step": 250
|
186 |
+
},
|
187 |
+
{
|
188 |
+
"epoch": 1.3471502590673574,
|
189 |
+
"grad_norm": 0.18436703085899353,
|
190 |
+
"learning_rate": 0.00011157167530224525,
|
191 |
+
"loss": 0.2204,
|
192 |
+
"step": 260
|
193 |
+
},
|
194 |
+
{
|
195 |
+
"epoch": 1.3989637305699483,
|
196 |
+
"grad_norm": 0.6231918334960938,
|
197 |
+
"learning_rate": 0.0001081174438687392,
|
198 |
+
"loss": 0.2204,
|
199 |
+
"step": 270
|
200 |
+
},
|
201 |
+
{
|
202 |
+
"epoch": 1.450777202072539,
|
203 |
+
"grad_norm": 0.27956509590148926,
|
204 |
+
"learning_rate": 0.00010466321243523317,
|
205 |
+
"loss": 0.1864,
|
206 |
+
"step": 280
|
207 |
+
},
|
208 |
+
{
|
209 |
+
"epoch": 1.5025906735751295,
|
210 |
+
"grad_norm": 0.24223612248897552,
|
211 |
+
"learning_rate": 0.00010120898100172711,
|
212 |
+
"loss": 0.1985,
|
213 |
+
"step": 290
|
214 |
+
},
|
215 |
+
{
|
216 |
+
"epoch": 1.5544041450777202,
|
217 |
+
"grad_norm": 0.28999006748199463,
|
218 |
+
"learning_rate": 9.775474956822108e-05,
|
219 |
+
"loss": 0.1322,
|
220 |
+
"step": 300
|
221 |
+
},
|
222 |
+
{
|
223 |
+
"epoch": 1.6062176165803108,
|
224 |
+
"grad_norm": 0.505622923374176,
|
225 |
+
"learning_rate": 9.430051813471503e-05,
|
226 |
+
"loss": 0.1745,
|
227 |
+
"step": 310
|
228 |
+
},
|
229 |
+
{
|
230 |
+
"epoch": 1.6580310880829017,
|
231 |
+
"grad_norm": 0.17805594205856323,
|
232 |
+
"learning_rate": 9.084628670120899e-05,
|
233 |
+
"loss": 0.229,
|
234 |
+
"step": 320
|
235 |
+
},
|
236 |
+
{
|
237 |
+
"epoch": 1.709844559585492,
|
238 |
+
"grad_norm": 0.8911346197128296,
|
239 |
+
"learning_rate": 8.739205526770295e-05,
|
240 |
+
"loss": 0.2056,
|
241 |
+
"step": 330
|
242 |
+
},
|
243 |
+
{
|
244 |
+
"epoch": 1.761658031088083,
|
245 |
+
"grad_norm": 0.2521348297595978,
|
246 |
+
"learning_rate": 8.39378238341969e-05,
|
247 |
+
"loss": 0.2274,
|
248 |
+
"step": 340
|
249 |
+
},
|
250 |
+
{
|
251 |
+
"epoch": 1.8134715025906736,
|
252 |
+
"grad_norm": 0.1958165019750595,
|
253 |
+
"learning_rate": 8.048359240069086e-05,
|
254 |
+
"loss": 0.1346,
|
255 |
+
"step": 350
|
256 |
+
},
|
257 |
+
{
|
258 |
+
"epoch": 1.8652849740932642,
|
259 |
+
"grad_norm": 0.43222665786743164,
|
260 |
+
"learning_rate": 7.702936096718481e-05,
|
261 |
+
"loss": 0.1926,
|
262 |
+
"step": 360
|
263 |
+
},
|
264 |
+
{
|
265 |
+
"epoch": 1.917098445595855,
|
266 |
+
"grad_norm": 0.28354278206825256,
|
267 |
+
"learning_rate": 7.357512953367876e-05,
|
268 |
+
"loss": 0.1491,
|
269 |
+
"step": 370
|
270 |
+
},
|
271 |
+
{
|
272 |
+
"epoch": 1.9689119170984455,
|
273 |
+
"grad_norm": 0.30970773100852966,
|
274 |
+
"learning_rate": 7.012089810017272e-05,
|
275 |
+
"loss": 0.1939,
|
276 |
+
"step": 380
|
277 |
+
},
|
278 |
+
{
|
279 |
+
"epoch": 2.0207253886010363,
|
280 |
+
"grad_norm": 0.23131532967090607,
|
281 |
+
"learning_rate": 6.666666666666667e-05,
|
282 |
+
"loss": 0.1619,
|
283 |
+
"step": 390
|
284 |
+
},
|
285 |
+
{
|
286 |
+
"epoch": 2.0725388601036268,
|
287 |
+
"grad_norm": 0.6155455708503723,
|
288 |
+
"learning_rate": 6.321243523316063e-05,
|
289 |
+
"loss": 0.1383,
|
290 |
+
"step": 400
|
291 |
+
},
|
292 |
+
{
|
293 |
+
"epoch": 2.1243523316062176,
|
294 |
+
"grad_norm": 0.4131735563278198,
|
295 |
+
"learning_rate": 5.9758203799654574e-05,
|
296 |
+
"loss": 0.1434,
|
297 |
+
"step": 410
|
298 |
+
},
|
299 |
+
{
|
300 |
+
"epoch": 2.1761658031088085,
|
301 |
+
"grad_norm": 0.3871082365512848,
|
302 |
+
"learning_rate": 5.6303972366148536e-05,
|
303 |
+
"loss": 0.0827,
|
304 |
+
"step": 420
|
305 |
+
},
|
306 |
+
{
|
307 |
+
"epoch": 2.227979274611399,
|
308 |
+
"grad_norm": 0.8619069457054138,
|
309 |
+
"learning_rate": 5.284974093264249e-05,
|
310 |
+
"loss": 0.161,
|
311 |
+
"step": 430
|
312 |
+
},
|
313 |
+
{
|
314 |
+
"epoch": 2.2797927461139897,
|
315 |
+
"grad_norm": 0.3584190905094147,
|
316 |
+
"learning_rate": 4.939550949913645e-05,
|
317 |
+
"loss": 0.1341,
|
318 |
+
"step": 440
|
319 |
+
},
|
320 |
+
{
|
321 |
+
"epoch": 2.33160621761658,
|
322 |
+
"grad_norm": 0.2826884090900421,
|
323 |
+
"learning_rate": 4.59412780656304e-05,
|
324 |
+
"loss": 0.1257,
|
325 |
+
"step": 450
|
326 |
+
},
|
327 |
+
{
|
328 |
+
"epoch": 2.383419689119171,
|
329 |
+
"grad_norm": 0.4198673367500305,
|
330 |
+
"learning_rate": 4.248704663212436e-05,
|
331 |
+
"loss": 0.1216,
|
332 |
+
"step": 460
|
333 |
+
},
|
334 |
+
{
|
335 |
+
"epoch": 2.4352331606217614,
|
336 |
+
"grad_norm": 0.582490086555481,
|
337 |
+
"learning_rate": 3.9032815198618314e-05,
|
338 |
+
"loss": 0.1325,
|
339 |
+
"step": 470
|
340 |
+
},
|
341 |
+
{
|
342 |
+
"epoch": 2.4870466321243523,
|
343 |
+
"grad_norm": 0.7895885109901428,
|
344 |
+
"learning_rate": 3.557858376511227e-05,
|
345 |
+
"loss": 0.128,
|
346 |
+
"step": 480
|
347 |
+
},
|
348 |
+
{
|
349 |
+
"epoch": 2.538860103626943,
|
350 |
+
"grad_norm": 0.24237169325351715,
|
351 |
+
"learning_rate": 3.212435233160622e-05,
|
352 |
+
"loss": 0.1391,
|
353 |
+
"step": 490
|
354 |
+
},
|
355 |
+
{
|
356 |
+
"epoch": 2.5906735751295336,
|
357 |
+
"grad_norm": 0.46788838505744934,
|
358 |
+
"learning_rate": 2.8670120898100174e-05,
|
359 |
+
"loss": 0.1712,
|
360 |
+
"step": 500
|
361 |
+
},
|
362 |
+
{
|
363 |
+
"epoch": 2.6424870466321244,
|
364 |
+
"grad_norm": 0.2091088891029358,
|
365 |
+
"learning_rate": 2.521588946459413e-05,
|
366 |
+
"loss": 0.1146,
|
367 |
+
"step": 510
|
368 |
+
},
|
369 |
+
{
|
370 |
+
"epoch": 2.694300518134715,
|
371 |
+
"grad_norm": 0.3197329342365265,
|
372 |
+
"learning_rate": 2.1761658031088085e-05,
|
373 |
+
"loss": 0.1483,
|
374 |
+
"step": 520
|
375 |
+
},
|
376 |
+
{
|
377 |
+
"epoch": 2.7461139896373057,
|
378 |
+
"grad_norm": 0.5090299844741821,
|
379 |
+
"learning_rate": 1.830742659758204e-05,
|
380 |
+
"loss": 0.1418,
|
381 |
+
"step": 530
|
382 |
+
},
|
383 |
+
{
|
384 |
+
"epoch": 2.7979274611398965,
|
385 |
+
"grad_norm": 0.1630585491657257,
|
386 |
+
"learning_rate": 1.4853195164075994e-05,
|
387 |
+
"loss": 0.1563,
|
388 |
+
"step": 540
|
389 |
+
},
|
390 |
+
{
|
391 |
+
"epoch": 2.849740932642487,
|
392 |
+
"grad_norm": 0.9052726626396179,
|
393 |
+
"learning_rate": 1.139896373056995e-05,
|
394 |
+
"loss": 0.1299,
|
395 |
+
"step": 550
|
396 |
+
},
|
397 |
+
{
|
398 |
+
"epoch": 2.901554404145078,
|
399 |
+
"grad_norm": 0.21230316162109375,
|
400 |
+
"learning_rate": 7.944732297063904e-06,
|
401 |
+
"loss": 0.1496,
|
402 |
+
"step": 560
|
403 |
+
},
|
404 |
+
{
|
405 |
+
"epoch": 2.9533678756476682,
|
406 |
+
"grad_norm": 0.4288373291492462,
|
407 |
+
"learning_rate": 4.490500863557859e-06,
|
408 |
+
"loss": 0.1735,
|
409 |
+
"step": 570
|
410 |
+
}
|
411 |
+
],
|
412 |
+
"logging_steps": 10,
|
413 |
+
"max_steps": 579,
|
414 |
+
"num_input_tokens_seen": 0,
|
415 |
+
"num_train_epochs": 3,
|
416 |
+
"save_steps": 500,
|
417 |
+
"stateful_callbacks": {
|
418 |
+
"TrainerControl": {
|
419 |
+
"args": {
|
420 |
+
"should_epoch_stop": false,
|
421 |
+
"should_evaluate": false,
|
422 |
+
"should_log": false,
|
423 |
+
"should_save": true,
|
424 |
+
"should_training_stop": true
|
425 |
+
},
|
426 |
+
"attributes": {}
|
427 |
+
}
|
428 |
+
},
|
429 |
+
"total_flos": 9523631719710720.0,
|
430 |
+
"train_batch_size": 1,
|
431 |
+
"trial_name": null,
|
432 |
+
"trial_params": null
|
433 |
+
}
|
capybara-finetuned/checkpoint-579/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5cb5dfa95d8bedc071d0ca429ace88a97d08b4e6c0549993636c23bd1392d3f9
|
3 |
+
size 5304
|
capybara-finetuned/special_tokens_map.json
ADDED
@@ -0,0 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bos_token": {
|
3 |
+
"content": "<|endoftext|>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"eos_token": {
|
10 |
+
"content": "<|im_end|>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"pad_token": {
|
17 |
+
"content": "[PAD]",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
},
|
23 |
+
"unk_token": {
|
24 |
+
"content": "<|endoftext|>",
|
25 |
+
"lstrip": false,
|
26 |
+
"normalized": false,
|
27 |
+
"rstrip": false,
|
28 |
+
"single_word": false
|
29 |
+
}
|
30 |
+
}
|
capybara-finetuned/tokenizer.json
ADDED
The diff for this file is too large to render.
See raw diff
|
|
capybara-finetuned/tokenizer_config.json
ADDED
@@ -0,0 +1,239 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_bos_token": false,
|
3 |
+
"add_eos_token": false,
|
4 |
+
"add_prefix_space": false,
|
5 |
+
"added_tokens_decoder": {
|
6 |
+
"0": {
|
7 |
+
"content": "<|endoftext|>",
|
8 |
+
"lstrip": false,
|
9 |
+
"normalized": false,
|
10 |
+
"rstrip": false,
|
11 |
+
"single_word": false,
|
12 |
+
"special": true
|
13 |
+
},
|
14 |
+
"1": {
|
15 |
+
"content": "<|padding|>",
|
16 |
+
"lstrip": false,
|
17 |
+
"normalized": false,
|
18 |
+
"rstrip": false,
|
19 |
+
"single_word": false,
|
20 |
+
"special": true
|
21 |
+
},
|
22 |
+
"50254": {
|
23 |
+
"content": " ",
|
24 |
+
"lstrip": false,
|
25 |
+
"normalized": true,
|
26 |
+
"rstrip": false,
|
27 |
+
"single_word": false,
|
28 |
+
"special": false
|
29 |
+
},
|
30 |
+
"50255": {
|
31 |
+
"content": " ",
|
32 |
+
"lstrip": false,
|
33 |
+
"normalized": true,
|
34 |
+
"rstrip": false,
|
35 |
+
"single_word": false,
|
36 |
+
"special": false
|
37 |
+
},
|
38 |
+
"50256": {
|
39 |
+
"content": " ",
|
40 |
+
"lstrip": false,
|
41 |
+
"normalized": true,
|
42 |
+
"rstrip": false,
|
43 |
+
"single_word": false,
|
44 |
+
"special": false
|
45 |
+
},
|
46 |
+
"50257": {
|
47 |
+
"content": " ",
|
48 |
+
"lstrip": false,
|
49 |
+
"normalized": true,
|
50 |
+
"rstrip": false,
|
51 |
+
"single_word": false,
|
52 |
+
"special": false
|
53 |
+
},
|
54 |
+
"50258": {
|
55 |
+
"content": " ",
|
56 |
+
"lstrip": false,
|
57 |
+
"normalized": true,
|
58 |
+
"rstrip": false,
|
59 |
+
"single_word": false,
|
60 |
+
"special": false
|
61 |
+
},
|
62 |
+
"50259": {
|
63 |
+
"content": " ",
|
64 |
+
"lstrip": false,
|
65 |
+
"normalized": true,
|
66 |
+
"rstrip": false,
|
67 |
+
"single_word": false,
|
68 |
+
"special": false
|
69 |
+
},
|
70 |
+
"50260": {
|
71 |
+
"content": " ",
|
72 |
+
"lstrip": false,
|
73 |
+
"normalized": true,
|
74 |
+
"rstrip": false,
|
75 |
+
"single_word": false,
|
76 |
+
"special": false
|
77 |
+
},
|
78 |
+
"50261": {
|
79 |
+
"content": " ",
|
80 |
+
"lstrip": false,
|
81 |
+
"normalized": true,
|
82 |
+
"rstrip": false,
|
83 |
+
"single_word": false,
|
84 |
+
"special": false
|
85 |
+
},
|
86 |
+
"50262": {
|
87 |
+
"content": " ",
|
88 |
+
"lstrip": false,
|
89 |
+
"normalized": true,
|
90 |
+
"rstrip": false,
|
91 |
+
"single_word": false,
|
92 |
+
"special": false
|
93 |
+
},
|
94 |
+
"50263": {
|
95 |
+
"content": " ",
|
96 |
+
"lstrip": false,
|
97 |
+
"normalized": true,
|
98 |
+
"rstrip": false,
|
99 |
+
"single_word": false,
|
100 |
+
"special": false
|
101 |
+
},
|
102 |
+
"50264": {
|
103 |
+
"content": " ",
|
104 |
+
"lstrip": false,
|
105 |
+
"normalized": true,
|
106 |
+
"rstrip": false,
|
107 |
+
"single_word": false,
|
108 |
+
"special": false
|
109 |
+
},
|
110 |
+
"50265": {
|
111 |
+
"content": " ",
|
112 |
+
"lstrip": false,
|
113 |
+
"normalized": true,
|
114 |
+
"rstrip": false,
|
115 |
+
"single_word": false,
|
116 |
+
"special": false
|
117 |
+
},
|
118 |
+
"50266": {
|
119 |
+
"content": " ",
|
120 |
+
"lstrip": false,
|
121 |
+
"normalized": true,
|
122 |
+
"rstrip": false,
|
123 |
+
"single_word": false,
|
124 |
+
"special": false
|
125 |
+
},
|
126 |
+
"50267": {
|
127 |
+
"content": " ",
|
128 |
+
"lstrip": false,
|
129 |
+
"normalized": true,
|
130 |
+
"rstrip": false,
|
131 |
+
"single_word": false,
|
132 |
+
"special": false
|
133 |
+
},
|
134 |
+
"50268": {
|
135 |
+
"content": " ",
|
136 |
+
"lstrip": false,
|
137 |
+
"normalized": true,
|
138 |
+
"rstrip": false,
|
139 |
+
"single_word": false,
|
140 |
+
"special": false
|
141 |
+
},
|
142 |
+
"50269": {
|
143 |
+
"content": " ",
|
144 |
+
"lstrip": false,
|
145 |
+
"normalized": true,
|
146 |
+
"rstrip": false,
|
147 |
+
"single_word": false,
|
148 |
+
"special": false
|
149 |
+
},
|
150 |
+
"50270": {
|
151 |
+
"content": " ",
|
152 |
+
"lstrip": false,
|
153 |
+
"normalized": true,
|
154 |
+
"rstrip": false,
|
155 |
+
"single_word": false,
|
156 |
+
"special": false
|
157 |
+
},
|
158 |
+
"50271": {
|
159 |
+
"content": " ",
|
160 |
+
"lstrip": false,
|
161 |
+
"normalized": true,
|
162 |
+
"rstrip": false,
|
163 |
+
"single_word": false,
|
164 |
+
"special": false
|
165 |
+
},
|
166 |
+
"50272": {
|
167 |
+
"content": " ",
|
168 |
+
"lstrip": false,
|
169 |
+
"normalized": true,
|
170 |
+
"rstrip": false,
|
171 |
+
"single_word": false,
|
172 |
+
"special": false
|
173 |
+
},
|
174 |
+
"50273": {
|
175 |
+
"content": " ",
|
176 |
+
"lstrip": false,
|
177 |
+
"normalized": true,
|
178 |
+
"rstrip": false,
|
179 |
+
"single_word": false,
|
180 |
+
"special": false
|
181 |
+
},
|
182 |
+
"50274": {
|
183 |
+
"content": " ",
|
184 |
+
"lstrip": false,
|
185 |
+
"normalized": true,
|
186 |
+
"rstrip": false,
|
187 |
+
"single_word": false,
|
188 |
+
"special": false
|
189 |
+
},
|
190 |
+
"50275": {
|
191 |
+
"content": " ",
|
192 |
+
"lstrip": false,
|
193 |
+
"normalized": true,
|
194 |
+
"rstrip": false,
|
195 |
+
"single_word": false,
|
196 |
+
"special": false
|
197 |
+
},
|
198 |
+
"50276": {
|
199 |
+
"content": " ",
|
200 |
+
"lstrip": false,
|
201 |
+
"normalized": true,
|
202 |
+
"rstrip": false,
|
203 |
+
"single_word": false,
|
204 |
+
"special": false
|
205 |
+
},
|
206 |
+
"50277": {
|
207 |
+
"content": "[PAD]",
|
208 |
+
"lstrip": false,
|
209 |
+
"normalized": false,
|
210 |
+
"rstrip": false,
|
211 |
+
"single_word": false,
|
212 |
+
"special": true
|
213 |
+
},
|
214 |
+
"50278": {
|
215 |
+
"content": "<|im_end|>",
|
216 |
+
"lstrip": false,
|
217 |
+
"normalized": false,
|
218 |
+
"rstrip": false,
|
219 |
+
"single_word": false,
|
220 |
+
"special": true
|
221 |
+
},
|
222 |
+
"50279": {
|
223 |
+
"content": "<|im_start|>",
|
224 |
+
"lstrip": false,
|
225 |
+
"normalized": false,
|
226 |
+
"rstrip": false,
|
227 |
+
"single_word": false,
|
228 |
+
"special": false
|
229 |
+
}
|
230 |
+
},
|
231 |
+
"bos_token": "<|endoftext|>",
|
232 |
+
"clean_up_tokenization_spaces": true,
|
233 |
+
"eos_token": "<|im_end|>",
|
234 |
+
"extra_special_tokens": {},
|
235 |
+
"model_max_length": 1000000000000000019884624838656,
|
236 |
+
"pad_token": "[PAD]",
|
237 |
+
"tokenizer_class": "GPTNeoXTokenizer",
|
238 |
+
"unk_token": "<|endoftext|>"
|
239 |
+
}
|
capybara-finetuned/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5cb5dfa95d8bedc071d0ca429ace88a97d08b4e6c0549993636c23bd1392d3f9
|
3 |
+
size 5304
|
requirements.txt
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
transformers==4.37.2
|
2 |
+
accelerate==0.27.2
|
3 |
+
torch>=2.1.0
|
4 |
+
peft==0.8.2
|
5 |
+
bitsandbytes==0.43.1
|
6 |
+
uvicorn
|
7 |
+
fastapi
|