Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ pip install transformers torch
|
|
| 58 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
| 59 |
|
| 60 |
# Load model and tokenizer
|
| 61 |
-
model_name = "
|
| 62 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 63 |
model = AutoModelForCausalLM.from_pretrained(
|
| 64 |
model_name,
|
|
|
|
| 58 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
| 59 |
|
| 60 |
# Load model and tokenizer
|
| 61 |
+
model_name = "barty/CodeLlama-13B-MORepair"
|
| 62 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 63 |
model = AutoModelForCausalLM.from_pretrained(
|
| 64 |
model_name,
|