linh5nb
/

Llama-2-7b-chat-finetune_covid

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

linh5nb commited on Apr 16

Commit

f58de22

•

1 Parent(s): 2c29a92

Update README.md

Files changed (1) hide show

README.md +3 -0

README.md CHANGED Viewed

@@ -11,8 +11,11 @@ model = AutoModelForCausalLM.from_pretrained("linh5nb/Llama-2-7b-chat-finetune_c
 tokenizer = AutoTokenizer.from_pretrained("linh5nb/Llama-2-7b-chat-finetune_covid")
 prompt = "What is the main cause of HIV-1 infection in children?"
 pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
 result = pipe(f"<s>[INST] {prompt} [/INST] ")
 print(result[0]['generated_text'])
 [More Information Needed]

 tokenizer = AutoTokenizer.from_pretrained("linh5nb/Llama-2-7b-chat-finetune_covid")
 prompt = "What is the main cause of HIV-1 infection in children?"
 pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
 result = pipe(f"<s>[INST] {prompt} [/INST] ")
 print(result[0]['generated_text'])
 [More Information Needed]