DMindAI
/

DMind-2-4B

Model card Files Files and versions

LucasAI111 commited on Sep 8

Commit

880d642

·

verified ·

1 Parent(s): 142d4b2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -123,11 +123,11 @@ import torch
 # Load model (optimized for edge deployment)
 model = AutoModelForCausalLM.from_pretrained(
-    "Qwen/Qwen3-4B-Thinking-2507",
     torch_dtype=torch.float16,  # Use half precision to save VRAM
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-4B-Thinking-2507")
 # Investment analysis example
 prompt = """

 # Load model (optimized for edge deployment)
 model = AutoModelForCausalLM.from_pretrained(
+    "DMindAI/DMind-2-4B",
     torch_dtype=torch.float16,  # Use half precision to save VRAM
     device_map="auto"
 )
+tokenizer = AutoTokenizer.from_pretrained("DMindAI/DMind-2-4B")
 # Investment analysis example
 prompt = """