Update README.md
Browse files
README.md
CHANGED
@@ -66,11 +66,11 @@ Below results are the average of **64 evaluations** on each benchmark.
|
|
66 |
|
67 |
To run inference on coding problems:
|
68 |
|
69 |
-
|
70 |
import transformers
|
71 |
import torch
|
72 |
|
73 |
-
model_id = "nvidia/OpenCodeReasoning-Nemotron-
|
74 |
|
75 |
pipeline = transformers.pipeline(
|
76 |
"text-generation",
|
@@ -93,8 +93,8 @@ You must use ```python for just the final solution code block with the following
|
|
93 |
|
94 |
messages = [
|
95 |
{
|
96 |
-
"role": "user",
|
97 |
-
"content": prompt.format(user="Write a program to calculate the sum of the first
|
98 |
]
|
99 |
|
100 |
outputs = pipeline(
|
@@ -103,7 +103,9 @@ outputs = pipeline(
|
|
103 |
)
|
104 |
print(outputs[0]["generated_text"][-1]['content'])
|
105 |
|
106 |
-
|
|
|
|
|
107 |
|
108 |
|
109 |
## Citation
|
|
|
66 |
|
67 |
To run inference on coding problems:
|
68 |
|
69 |
+
````python
|
70 |
import transformers
|
71 |
import torch
|
72 |
|
73 |
+
model_id = "nvidia/OpenCodeReasoning-Nemotron-7B"
|
74 |
|
75 |
pipeline = transformers.pipeline(
|
76 |
"text-generation",
|
|
|
93 |
|
94 |
messages = [
|
95 |
{
|
96 |
+
"role": "user",
|
97 |
+
"content": prompt.format(user="Write a program to calculate the sum of the first <span class="math-inline">N</span> fibonacci numbers")},
|
98 |
]
|
99 |
|
100 |
outputs = pipeline(
|
|
|
103 |
)
|
104 |
print(outputs[0]["generated_text"][-1]['content'])
|
105 |
|
106 |
+
````
|
107 |
+
|
108 |
+
|
109 |
|
110 |
|
111 |
## Citation
|