mwitiderrick
/

open_llama_3b_glaive_code_v0.1

Text Generation

text-generation-inference

Model card Files Files and versions

mwitiderrick commited on Dec 19, 2023

Commit

aab561f

·

1 Parent(s): 360bf72

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -72,7 +72,7 @@ tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_glaive_ass
 model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_glaive_assistant_v0.1")
 query = "Write a quick sort algorithm in Python"
 text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
-output = text_gen(f<s>[INST]{query}[/INST]")
 print(output[0]['generated_text'])
 """

 model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_glaive_assistant_v0.1")
 query = "Write a quick sort algorithm in Python"
 text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
+output = text_gen(f"<s>[INST]{query}[/INST]")
 print(output[0]['generated_text'])
 """