Update README.md
Browse files
README.md
CHANGED
@@ -136,6 +136,19 @@ trainer = SFTTrainer(
|
|
136 |
|
137 |
#@title 学習実行
|
138 |
trainer_stats = trainer.train()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
139 |
```
|
140 |
### 推論
|
141 |
```
|
|
|
136 |
|
137 |
#@title 学習実行
|
138 |
trainer_stats = trainer.train()
|
139 |
+
|
140 |
+
# モデルとトークナイザーをHugging Faceにアップロード。
|
141 |
+
# 一旦privateでアップロードしてください。
|
142 |
+
# 最終成果物が決まったらpublicにするようお願いします。
|
143 |
+
new_model_name = "モデルの名前"
|
144 |
+
model.push_to_hub_merged(
|
145 |
+
new_model_name,
|
146 |
+
tokenizer=tokenizer,
|
147 |
+
save_method="lora",
|
148 |
+
token=HF_TOKEN,
|
149 |
+
private=True
|
150 |
+
) # Online saving
|
151 |
+
tokenizer.push_to_hub(new_model_name, token=HF_TOKEN) # Online saving
|
152 |
```
|
153 |
### 推論
|
154 |
```
|