Update README.md
Browse files
README.md
CHANGED
|
@@ -31,8 +31,8 @@ import torch
|
|
| 31 |
|
| 32 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 33 |
|
| 34 |
-
model = AutoModelForCausalLM.from_pretrained('nvidia/DLER-R1-7B').to(device)
|
| 35 |
-
tokenizer = AutoTokenizer.from_pretrained('nvidia/DLER-R1-7B')
|
| 36 |
|
| 37 |
messages = [
|
| 38 |
{"role": "user", "content": "Convert the point $(0,3)$ in rectangular coordinates to polar coordinates. Enter your answer in the form $(r,\\theta),$ where $r > 0$ and $0 \\le \\theta < 2 \\pi.$"+" Let's think step by step and output the final answer within \\boxed{}."},
|
|
|
|
| 31 |
|
| 32 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 33 |
|
| 34 |
+
model = AutoModelForCausalLM.from_pretrained('nvidia/DLER-R1-7B-Research').to(device)
|
| 35 |
+
tokenizer = AutoTokenizer.from_pretrained('nvidia/DLER-R1-7B-Research')
|
| 36 |
|
| 37 |
messages = [
|
| 38 |
{"role": "user", "content": "Convert the point $(0,3)$ in rectangular coordinates to polar coordinates. Enter your answer in the form $(r,\\theta),$ where $r > 0$ and $0 \\le \\theta < 2 \\pi.$"+" Let's think step by step and output the final answer within \\boxed{}."},
|