diff --git a/scripts/chat_rl.py b/scripts/chat_rl.py index bc78e79..b1db7c7 100644 --- a/scripts/chat_rl.py +++ b/scripts/chat_rl.py @@ -18,7 +18,6 @@ torchrun --standalone --nproc_per_node=8 -m scripts.chat_rl -- --run=default import os import itertools -import re import wandb import torch import torch.distributed as dist