Views
No views yet
tinyrolls/Llama3_8B_dpo1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "tinyrolls/Llama3_8B_dpo"
4tok = AutoTokenizer.from_pretrained(model_id)
5model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto", device_map="auto")