tatsu-lab/alpaca
Viewer β’ Updated β’ 52k β’ 83.7k β’ 1.39k
μ΄ λͺ¨λΈμ 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)