fixed read

allenai · Feb 15, 2024 · 3951feb · 3951feb
1 parent 9b41f4e
commit 3951feb
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/analysis/per_token_reward.py b/analysis/per_token_reward.py
@@ -101,9 +101,6 @@ def main():
     custom_dialogue = False
 
     models = args.model
-    # print (models)
-    # exit(1)
-    models = ["OpenAssistant/reward-model-deberta-v3-large-v2", "PKU-Alignment/beaver-7b-v1.0-cost", "IDEA-CCNL/Ziya-LLaMA-7B-Reward"]
 
     ###############
     # Setup logging