Update README.md
Browse files
README.md
CHANGED
@@ -47,9 +47,9 @@ print(assistant_response)
|
|
47 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
48 |
import torch
|
49 |
|
50 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
51 |
model = AutoModelForCausalLM.from_pretrained(
|
52 |
-
"
|
53 |
device_map="auto",
|
54 |
torch_dtype=torch.bfloat16,
|
55 |
)
|
|
|
47 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
48 |
import torch
|
49 |
|
50 |
+
tokenizer = AutoTokenizer.from_pretrained("korarishi/rishi-2-2b-it")
|
51 |
model = AutoModelForCausalLM.from_pretrained(
|
52 |
+
"korarishi/rishi-2-2b-it",
|
53 |
device_map="auto",
|
54 |
torch_dtype=torch.bfloat16,
|
55 |
)
|