Views
No views yet

1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("cxllin/StableHermes-3b")
4model = AutoModelForCausalLM.from_pretrained(
5 "cxllin/StableHermes-3b",
6 trust_remote_code=True,
7 torch_dtype="auto",
8)
9model.cuda()
10inputs = tokenizer("Describe the potential implications of quantum computing on the future of cybersecurity.", return_tensors="pt").to("cuda")
11tokens = model.generate(
12 **inputs,
13 max_new_tokens=64,
14 temperature=0.75,
15 top_p=0.95,
16 do_sample=True,
17)
18print(tokenizer.decode(tokens[0], skip_special_tokens=True))