fujio48694062
commited on
Commit
•
708d87f
1
Parent(s):
91aa530
Update README.md
Browse files
README.md
CHANGED
@@ -24,7 +24,6 @@ This llama model was trained 2x faster with [Unsloth](https://github.com/unsloth
|
|
24 |
----------
|
25 |
以下は、elyza-tasks-100-TV_0の回答のためのコードです。
|
26 |
|
27 |
-
①Fine-tuning用コード
|
28 |
```python
|
29 |
# -*- coding: utf-8 -*-
|
30 |
"""LoRA_template_unsloth_20241127.ipynb
|
@@ -341,103 +340,4 @@ model.push_to_hub_merged(
|
|
341 |
token=HF_TOKEN,
|
342 |
private=True
|
343 |
)
|
344 |
-
```
|
345 |
-
|
346 |
-
②推論用コード
|
347 |
-
```python
|
348 |
-
# -*- coding: utf-8 -*-
|
349 |
-
"""Model_Inference_Template_unsloth_20241127.ipynb
|
350 |
-
|
351 |
-
Automatically generated by Colab.
|
352 |
-
|
353 |
-
Original file is located at
|
354 |
-
https://colab.research.google.com/drive/1qRChXRT-qm94rl5roqgrjSxlK5_A9MlI
|
355 |
-
|
356 |
-
# 推論用コード
|
357 |
-
本コードはunslothで学習したqLoRAのアダプタを用いてELYZA-tasks-100-TVの出力を得るためのコードです。
|
358 |
-
Hugging Faceにアダプタをアップロードしてあることが前提となります。
|
359 |
-
このコードはunslothライブラリを用いてモデルを読み込み、推論するためのコードとなります。
|
360 |
-
このコードで生成されたjsonlファイルは課題の成果として提出可能なフォーマットになっております。
|
361 |
-
|
362 |
-
※本コードはGoogle Colabでの動作を想定しており、Omnicampusでの動作を想定しておりません。
|
363 |
-
Omnicampus向けのコードは別途用意しております。
|
364 |
-
"""
|
365 |
-
|
366 |
-
# Commented out IPython magic to ensure Python compatibility.
|
367 |
-
# # 必要なライブラリをインストール
|
368 |
-
# %%capture
|
369 |
-
# !pip install unsloth
|
370 |
-
# !pip uninstall unsloth -y && pip install --upgrade --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
|
371 |
-
# !pip install -U torch
|
372 |
-
# !pip install -U peft
|
373 |
-
|
374 |
-
# 必要なライブラリを読み込み
|
375 |
-
from unsloth import FastLanguageModel
|
376 |
-
from peft import PeftModel
|
377 |
-
import torch
|
378 |
-
import json
|
379 |
-
from tqdm import tqdm
|
380 |
-
import re
|
381 |
-
|
382 |
-
# ベースとなるモデルと学習したLoRAのアダプタ(Hugging FaceのIDを指定)。
|
383 |
-
model_id = "llm-jp/llm-jp-3-13b"
|
384 |
-
adapter_id = ""
|
385 |
-
|
386 |
-
# Hugging Face Token を指定。
|
387 |
-
# 下記の URL から Hugging Face Token を取得できますので下記の HF_TOKEN に入れてください。
|
388 |
-
# https://huggingface.co/settings/tokens
|
389 |
-
HF_TOKEN = "" #@param {type:"string"}
|
390 |
-
|
391 |
-
# unslothのFastLanguageModelで元のモデルをロード。
|
392 |
-
dtype = None # Noneにしておけば自動で設定
|
393 |
-
load_in_4bit = True # 今回は13Bモデルを扱うためTrue
|
394 |
-
|
395 |
-
model, tokenizer = FastLanguageModel.from_pretrained(
|
396 |
-
model_name=model_id,
|
397 |
-
dtype=dtype,
|
398 |
-
load_in_4bit=load_in_4bit,
|
399 |
-
trust_remote_code=True,
|
400 |
-
)
|
401 |
-
|
402 |
-
# 元のモデルにLoRAのアダプタを統合。
|
403 |
-
model = PeftModel.from_pretrained(model, adapter_id, token = HF_TOKEN)
|
404 |
-
|
405 |
-
# タスクとなるデータの読み込み。
|
406 |
-
# 事前にデータをアップロードしてください。
|
407 |
-
datasets = []
|
408 |
-
with open("./elyza-tasks-100-TV_0.jsonl", "r") as f:
|
409 |
-
item = ""
|
410 |
-
for line in f:
|
411 |
-
line = line.strip()
|
412 |
-
item += line
|
413 |
-
if item.endswith("}"):
|
414 |
-
datasets.append(json.loads(item))
|
415 |
-
item = ""
|
416 |
-
|
417 |
-
# モデルを用いてタスクの推論。
|
418 |
-
|
419 |
-
# 推論するためにモデルのモードを変更
|
420 |
-
FastLanguageModel.for_inference(model)
|
421 |
-
|
422 |
-
results = []
|
423 |
-
for dt in tqdm(datasets):
|
424 |
-
input = dt["input"]
|
425 |
-
|
426 |
-
prompt = f"""### 指示\n{input}\n### 回答\n"""
|
427 |
-
|
428 |
-
inputs = tokenizer([prompt], return_tensors = "pt").to(model.device)
|
429 |
-
|
430 |
-
outputs = model.generate(**inputs, max_new_tokens = 512, use_cache = True, do_sample=False, repetition_penalty=1.2)
|
431 |
-
prediction = tokenizer.decode(outputs[0], skip_special_tokens=True).split('\n### 回答')[-1]
|
432 |
-
|
433 |
-
results.append({"task_id": dt["task_id"], "input": input, "output": prediction})
|
434 |
-
|
435 |
-
# 結果をjsonlで保存。
|
436 |
-
|
437 |
-
# ここではadapter_idを元にファイル名を決定しているが、ファイル名は任意で問題なし。
|
438 |
-
json_file_id = re.sub(".*/", "", adapter_id)
|
439 |
-
with open(f"/content/{json_file_id}_output.jsonl", 'w', encoding='utf-8') as f:
|
440 |
-
for result in results:
|
441 |
-
json.dump(result, f, ensure_ascii=False)
|
442 |
-
f.write('\n')
|
443 |
```
|
|
|
24 |
----------
|
25 |
以下は、elyza-tasks-100-TV_0の回答のためのコードです。
|
26 |
|
|
|
27 |
```python
|
28 |
# -*- coding: utf-8 -*-
|
29 |
"""LoRA_template_unsloth_20241127.ipynb
|
|
|
340 |
token=HF_TOKEN,
|
341 |
private=True
|
342 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
343 |
```
|