Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2
3tokenizer = AutoTokenizer.from_pretrained("yuvrajpant56/llama3-ultrachat-sft")
4model = AutoModelForCausalLM.from_pretrained("yuvrajpant56/llama3-ultrachat-sft", device_map="auto")
5
6chat = pipeline(
7 "text-generation",
8 model=model,
9 tokenizer=tokenizer,
10 max_new_tokens=128,
11 do_sample=True,
12 temperature=0.7,
13)
14
15prompt = "<|begin_of_sentence|>\nUser: Hello, how are you?\n<|end_of_sentence|>"
16print(chat(prompt))