mwitiderrick
commited on
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")
|