Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("Scorched2/shader-v2", trust_remote_code=True)
4tokenizer = AutoTokenizer.from_pretrained("Scorched2/shader-v2")
5
6inputs = tokenizer("### Instruction:\nExplain AI.\n\n### Response:\n", return_tensors="pt")
7out = model.generate(**inputs, max_new_tokens=200)
8print(tokenizer.decode(out[0]))