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 |
"""
|