Skip to content

Commit 82c9707

Browse files
authored
[Minor] remove unused variables in GRPO scripts (#3038)
1 parent fda1b8c commit 82c9707

File tree

2 files changed

+0
-10
lines changed

2 files changed

+0
-10
lines changed

sota-implementations/grpo/grpo-async.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -85,11 +85,6 @@ def make_env(cfg: DictConfig, devices: list[int] | None = None):
8585
# For the collector actor, we want inference_model devices first, then ref_model devices
8686
train_tokenizer = get_tokenizer(cfg)
8787

88-
# Get device information
89-
num_inf_devices = cfg.inference_model.num_devices
90-
num_ref_devices = cfg.ref_model.num_devices
91-
num_inf_devices + num_ref_devices
92-
9388
# Create a new config with adjusted device assignments
9489
ref_cfg = DictConfig(dict(cfg))
9590
ref_model = get_ref_model(ref_cfg, train_tokenizer, devices=devices)

sota-implementations/grpo/grpo-sync.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -86,11 +86,6 @@ def make_env(cfg: DictConfig, devices: list[int] | None = None):
8686
# For the collector actor, we want inference_model devices first, then ref_model devices
8787
train_tokenizer = get_tokenizer(cfg)
8888

89-
# Get device information
90-
num_inf_devices = cfg.inference_model.num_devices
91-
num_ref_devices = cfg.ref_model.num_devices
92-
num_inf_devices + num_ref_devices
93-
9489
# Create a new config with adjusted device assignments
9590
ref_cfg = DictConfig(dict(cfg))
9691
ref_model = get_ref_model(ref_cfg, train_tokenizer, devices=devices)

0 commit comments

Comments
 (0)