taoki's picture
Update README.md
239d561 verified
|
raw
history blame
1.88 kB
---
language:
- ja
license: other
tags:
- text-generation-inference
- transformers
- trl
- deepseek
datasets:
- nu-dialogue/jmultiwoz
- kunishou/amenokaku-code-instruct
- HachiML/alpaca_jp_python
license_name: fugaku-llm-tou
base_model: Fugaku-LLM/Fugaku-LLM-13B-instruct
---
# Uploaded model
- **Developed by:** taoki
- **License:** Fugaku-LLM Terms of Use
- **Finetuned from model :** Fugaku-LLM/Fugaku-LLM-13B-instruct
# Usage
```python
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
tokenizer = AutoTokenizer.from_pretrained(
"taoki/Fugaku-LLM-13B-instruct-qlora-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
)
model = AutoModelForCausalLM.from_pretrained(
"taoki/Fugaku-LLM-13B-instruct-qlora-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
)
if torch.cuda.is_available():
model = model.to("cuda")
prompt="""以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。
### 指示:
OpenCVを用いて定点カメラから画像を保存するコードを示してください。
### 応答:
"""
input_ids = tokenizer.encode(prompt,
add_special_tokens=False,
return_tensors="pt")
tokens = model.generate(
input_ids.to(device=model.device),
max_new_tokens=128,
do_sample=True,
temperature=0.1,
top_p=1.0,
repetition_penalty=1.0,
top_k=0
)
out = tokenizer.decode(tokens[0], skip_special_tokens=False)
print(out)
```
# Output
````
以下は、タスクを説明する指示です。要求を適切に満たす応答を書きなさい。
### 指示:
OpenCVを用いて定点カメラから画像を保存するコードを示してください。
### 応答:
```python
import cv2
# 画像を読み込む
img = cv2.imread('image.jpg')
# 画像を保存する
cv2.imwrite('output.jpg', img)
```<EOD|LLM-jp>
````