Views
No views yet
!pip install ctransformers1from ctransformers import AutoModelForCausalLM
2from transformers import AutoTokenizer
3import torch
4
5model = AutoModelForCausalLM.from_pretrained("alibidaran/llama-2-7b-virtual_doctor-gguf",hf=True)
6tokenizer = AutoTokenizer.from_pretrained("alibidaran/llama-2-7b-virtual_doctor")
7
8prompt = " Hi doctor, I have nose running, I have fever, I often feel tired, what should i do?"
9text=f"<s> ###Human: {prompt} ###Asistant: "
10inputs=tokenizer(text,return_tensors='pt').to('cpu')
11with torch.no_grad():
12 outputs=model.generate(**inputs,max_new_tokens=200,do_sample=True,top_p=0.92,top_k=10,temperature=0.7)
13print(tokenizer.decode(outputs[0], skip_special_tokens=True))
14