mwitiderrick commited on
Commit
75630d7
·
1 Parent(s): fd4ab10

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -76,7 +76,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
76
  tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/shearedplats-2.7b-v2-instruct-v0.1")
77
  model = AutoModelForCausalLM.from_pretrained("mwitiderrick/shearedplats-2.7b-v2-instruct-v0.1")
78
  query = "Provide step-by-step instructions for making a sweet chicken bugger"
79
- text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=450)
80
  output = text_gen(f"### Instruction:\n{query}\n### Response:\n")
81
  print(output[0]['generated_text'])
82
  """
 
76
  tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/shearedplats-2.7b-v2-instruct-v0.1")
77
  model = AutoModelForCausalLM.from_pretrained("mwitiderrick/shearedplats-2.7b-v2-instruct-v0.1")
78
  query = "Provide step-by-step instructions for making a sweet chicken bugger"
79
+ text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=350)
80
  output = text_gen(f"### Instruction:\n{query}\n### Response:\n")
81
  print(output[0]['generated_text'])
82
  """