nvidia
/

AceMath-72B-RM

Text Generation

Model card Files Files and versions Community

ychenNLP commited on 10 days ago

Commit

49271da

·

verified ·

1 Parent(s): 997bce2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -43,7 +43,7 @@ Greedy decoding (pass@1) results on a variety of math reasoning benchmarks. AceM
 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-model_name = "nvidia/AceMath-72B-RM" # Path to the model on cw-dfw-cs
 device = "auto" # the device to load the model onto
 model = AutoModelForSequenceClassification.from_pretrained(

 import torch
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+model_name = "nvidia/AceMath-72B-RM" # Path to the model
 device = "auto" # the device to load the model onto
 model = AutoModelForSequenceClassification.from_pretrained(