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