Views
No views yet
Note: This is a derivative of the base model. Usage must comply with the base model's original terms.
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4repo = "your_id/your-merged-repo"
5
6tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 repo,
9 torch_dtype=torch.float16,
10 device_map="auto",
11 trust_remote_code=True,
12)
13
14# (optional) generation example
15inputs = tokenizer("Hello! What should I do next?", return_tensors="pt").to(model.device)
16out = model.generate(**inputs, max_new_tokens=256)
17print(tokenizer.decode(out[0], skip_special_tokens=True))