mwitiderrick commited on
Commit
06092d0
1 Parent(s): fe655e2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -78,7 +78,7 @@ tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_instruct_v
78
  model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_instruct_v_0.2")
79
  query = "Provide step-by-step instructions for making banana bread"
80
  text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
81
- output = text_gen(f"### Instruction:\n{query}### Response:\n")
82
  print(output[0]['generated_text'])
83
  """
84
  ### Instruction:
 
78
  model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_instruct_v_0.2")
79
  query = "Provide step-by-step instructions for making banana bread"
80
  text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
81
+ output = text_gen(f"### Instruction:\n{query}\n### Response:\n")
82
  print(output[0]['generated_text'])
83
  """
84
  ### Instruction: