Views
No views yet
nakotsuko13/dpo-dataset-step5-cleansed to suppress invalid actions (e.g., hallucinated "Task failed" commands) and enforce strict environment rule adherence.peft. You can load it directly:1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "your_hf_username/your_repo_name" # HFに上げた場合
5
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto"
10)
11tokenizer = AutoTokenizer.from_pretrained(model_id)
12