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
|