Remove device

Jack-Khuu · Jack-Khuu · commit 6c7e600bb4cd · 2025-10-14T10:37:54.000-07:00
diff --git a/apps/grpo/main.py b/apps/grpo/main.py
@@ -382,7 +382,6 @@ async def continuous_rollouts():
             input_ids = torch.ones(
                 (group_size, max_req_tokens + max_res_tokens),
                 dtype=torch.long,
-                device="cuda",
             )
             for i, response in enumerate(responses):
                 episode = Episode(

Original file line number	Diff line number	Diff line change
`@@ -382,7 +382,6 @@ async def continuous_rollouts():`
`382`	`382`	`input_ids = torch.ones(`
`383`	`383`	`(group_size, max_req_tokens + max_res_tokens),`
`384`	`384`	`dtype=torch.long,`
`385`		`- device="cuda",`
`386`	`385`	`)`
`387`	`386`	`for i, response in enumerate(responses):`
`388`	`387`	`episode = Episode(`