mwitiderrick
commited on
Commit
·
75630d7
1
Parent(s):
fd4ab10
Update README.md
Browse files
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=
|
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 |
"""
|