Views
No views yet
transformers.1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4#v4 変更by tabi before model_id = "your_id/your-repo-name"
5model_id = "tabidance/qwen3-lora-dpo"
6
7tokenizer = AutoTokenizer.from_pretrained(model_id)
8model = AutoModelForCausalLM.from_pretrained(
9 model_id,
10 torch_dtype=torch.float16,
11 device_map="auto"
12)
13
14# Test inference
15#v4 再変更by tabi エラーを消すため、prompt, inputs, outputs, printの行に#をつけてコメントアウト
16
17