Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2model = AutoModelForCausalLM.from_pretrained("Arushp1/llama3-medquad-qlora")
3tokenizer = AutoTokenizer.from_pretrained("Arushp1/llama3-medquad-qlora")
4
5pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
6
7query = "What are the symptoms of asthma?"
8print(pipe(query, max_new_tokens=100))