Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,7 @@ Then, copy the snippet from the relevant section for your use case.
|
|
34 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
35 |
import transformers
|
36 |
import torch
|
37 |
-
model_id = "HODACHI/EZO-
|
38 |
dtype = torch.bfloat16
|
39 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
40 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
34 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
35 |
import transformers
|
36 |
import torch
|
37 |
+
model_id = "HODACHI/EZO-Common-9B-gemma-2-it"
|
38 |
dtype = torch.bfloat16
|
39 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
40 |
model = AutoModelForCausalLM.from_pretrained(
|