LucasAI111 commited on
Commit
af88f08
·
verified ·
1 Parent(s): 56cc5e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -121,13 +121,13 @@ import torch
121
 
122
  # Load the model and tokenizer
123
  model = AutoModelForCausalLM.from_pretrained(
124
- "zai-org/GLM-4.5-Air",
125
  torch_dtype=torch.float16,
126
  device_map="auto",
127
  trust_remote_code=True
128
  )
129
  tokenizer = AutoTokenizer.from_pretrained(
130
- "zai-org/GLM-4.5-Air",
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