Commit
·
752d2cf
1
Parent(s):
3d973c3
Update README.md
Browse files
README.md
CHANGED
|
@@ -72,8 +72,8 @@ This is an [OpenLlama model](https://huggingface.co/openlm-research/open_llama_3
|
|
| 72 |
```python
|
| 73 |
from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
|
| 74 |
|
| 75 |
-
tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/
|
| 76 |
-
model = AutoModelForCausalLM.from_pretrained("mwitiderrick/
|
| 77 |
query = "Write a quick sort algorithm in Python"
|
| 78 |
text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
|
| 79 |
output = text_gen(f"### Instruction:\n{query}\n### Response:\n")
|
|
|
|
| 72 |
```python
|
| 73 |
from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
|
| 74 |
|
| 75 |
+
tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_code_instruct_0.1")
|
| 76 |
+
model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_code_instruct_0.1")
|
| 77 |
query = "Write a quick sort algorithm in Python"
|
| 78 |
text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
|
| 79 |
output = text_gen(f"### Instruction:\n{query}\n### Response:\n")
|