RobotsMali
/

reward-model

Generated from Trainer

Model card Files Files and versions

Panga-Azazia commited on Jan 9

Commit

c75dc47

·

verified ·

1 Parent(s): 62ddbbb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -165,7 +165,7 @@ feature_extractor : RewardFeatureExtractor = RewardFeatureExtractor(asr_model)
 processor : RewardModelProcessor = RewardModelProcessor(feature_extractor, tokenizer)
-model : RewardModel = RewardModel.from_pretrained("Panga-Azazia/reward-model")
 model.eval()
 model.to(device)

 processor : RewardModelProcessor = RewardModelProcessor(feature_extractor, tokenizer)
+model : RewardModel = RewardModel.from_pretrained("RobotsMali/reward-model")
 model.eval()
 model.to(device)