Views
No views yet
1
2model = AutoModelForCausalLM.from_pretrained(f"efromomr/llm-course-hw3-tinyllama-qlora", device_map="auto")
3tokenizer = AutoTokenizer.from_pretrained(f"efromomr/llm-course-hw3-tinyllama-qlora")
4tokenizer.pad_token = tokenizer.eos_token
5tokenizer.padding_side = "left"
6
7input_ids = tokenizer(text, return_tensors="pt").input_ids
8
9output_ids = model.generate(input_ids, max_new_tokens=16)
10generated_text = tokenizer.decode(output_ids[0][len(input_ids[0]) :], skip_special_tokens=True)
11print(generated_text)
12#positive