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