tatsu-lab/alpaca
Viewer โข Updated โข 52k โข 81.7k โข 1.08k
์ด ๋ชจ๋ธ์ Qwen/Qwen3-0.6B์ ๊ธฐ๋ฐ์ผ๋ก LoRA(Low-Rank Adaptation) ๊ธฐ๋ฒ์ ์ฌ์ฉํด ํ์ธํ๋ํ ํ, ๊ธฐ๋ณธ ๋ชจ๋ธ๊ณผ ๋ณํฉ๋ ์์ ํ ๋ชจ๋ธ์ ๋๋ค.
from transformers import AutoTokenizer, AutoModelForCausalLM
# ๋ณํฉ๋ ๋ชจ๋ธ๊ณผ ํ ํฌ๋์ด์ ๋ก๋ (๋ณ๋์ ์ด๋ํฐ ๋ก๋ ๋ถํ์)
tokenizer = AutoTokenizer.from_pretrained("Whitewinter/model-merged", trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
"Whitewinter/model-merged",
trust_remote_code=True,
torch_dtype=torch.bfloat16,
device_map="auto"
)
# ์ถ๋ก
prompt = "### Instruction:\nExplain what machine learning is.\n\n### Response:\n"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=200, temperature=0.7)
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(response)