nvidia
/

DLER-R1-1.5B-Research

Model card Files Files and versions

sliuau commited on Oct 22

Commit

0e1b6d9

·

verified ·

1 Parent(s): 757a3f2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -38,8 +38,8 @@ import torch
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = AutoModelForCausalLM.from_pretrained('nvidia/DLER-R1-1.5B').to(device)
-tokenizer = AutoTokenizer.from_pretrained('nvidia/DLER-R1-1.5B')
 messages = [

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = AutoModelForCausalLM.from_pretrained('nvidia/DLER-R1-1.5B-Research').to(device)
+tokenizer = AutoTokenizer.from_pretrained('nvidia/DLER-R1-1.5B-Research')
 messages = [