Update train_prompts.py

applications/Chat/examples/train_prompts.py

@@ -74,7 +74,6 @@ def main(args):
 
         if args.model == "gpt2":
             actor = GPTActor(pretrained=args.pretrain, lora_rank=args.lora_rank)
-            # actor = GPTActor.from_pretrained(args.pretrain)
         elif args.model == "bloom":
             actor = BLOOMActor(pretrained=args.pretrain, lora_rank=args.lora_rank)
         elif args.model == "opt":
@@ -166,9 +165,6 @@ def main(args):
         (actor, actor_optim), (critic, critic_optim), reward_model, initial_model
     )
 
-    # strategy.load_model(initial_model, args.pretrain)
-    # strategy.load_model(actor, args.pretrain)
-
     # configure trainer
     trainer = PPOTrainer(
         strategy,