nvidia
/

OpenCodeReasoning-Nemotron-1.1-7B

Text Generation

text-generation-inference

Model card Files Files and versions Community

wasiuddina commited on Jun 12

Commit

0a4f558

·

verified ·

1 Parent(s): e515163

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -75,7 +75,7 @@ To run inference on coding problems:
 import transformers
 import torch
-model_id = "nvidia/OpenCodeReasoning-Nemotron-7B"
 pipeline = transformers.pipeline(
     "text-generation",
@@ -104,7 +104,7 @@ messages = [
 outputs = pipeline(
     messages,
-    max_new_tokens=32768,
 )
 print(outputs[0]["generated_text"][-1]['content'])

 import transformers
 import torch
+model_id = "nvidia/OpenCodeReasoning-Nemotron-7B-v1.1"
 pipeline = transformers.pipeline(
     "text-generation",
 outputs = pipeline(
     messages,
+    max_new_tokens=49152,
 )
 print(outputs[0]["generated_text"][-1]['content'])