Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ from transformers import (
|
|
| 18 |
)
|
| 19 |
import torch
|
| 20 |
|
| 21 |
-
repo = "
|
| 22 |
model = AutoModelForCausalLM.from_pretrained(
|
| 23 |
repo,
|
| 24 |
torch_dtype=torch.float16,
|
|
|
|
| 18 |
)
|
| 19 |
import torch
|
| 20 |
|
| 21 |
+
repo = "heavytail/kullm-mistral"
|
| 22 |
model = AutoModelForCausalLM.from_pretrained(
|
| 23 |
repo,
|
| 24 |
torch_dtype=torch.float16,
|