smajumdar94 commited on
Commit
edac986
·
verified ·
1 Parent(s): 462f40b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -5
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
- ```python
70
  import transformers
71
  import torch
72
 
73
- model_id = "nvidia/OpenCodeReasoning-Nemotron-14B"
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 $N$ fibonacci numbers")},
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