Update README.md
Browse files
README.md
CHANGED
|
@@ -121,13 +121,13 @@ import torch
|
|
| 121 |
|
| 122 |
# Load the model and tokenizer
|
| 123 |
model = AutoModelForCausalLM.from_pretrained(
|
| 124 |
-
"
|
| 125 |
torch_dtype=torch.float16,
|
| 126 |
device_map="auto",
|
| 127 |
trust_remote_code=True
|
| 128 |
)
|
| 129 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 130 |
-
"
|
| 131 |
trust_remote_code=True
|
| 132 |
)
|
| 133 |
|
|
|
|
| 121 |
|
| 122 |
# Load the model and tokenizer
|
| 123 |
model = AutoModelForCausalLM.from_pretrained(
|
| 124 |
+
"DMindAI/DMind-2-107B",
|
| 125 |
torch_dtype=torch.float16,
|
| 126 |
device_map="auto",
|
| 127 |
trust_remote_code=True
|
| 128 |
)
|
| 129 |
tokenizer = AutoTokenizer.from_pretrained(
|
| 130 |
+
"DMindAI/DMind-2-107B",
|
| 131 |
trust_remote_code=True
|
| 132 |
)
|
| 133 |
|