diff --git a/unsloth/models/rl.py b/unsloth/models/rl.py index f943afdcb4..33f5e0251e 100644 --- a/unsloth/models/rl.py +++ b/unsloth/models/rl.py @@ -481,9 +481,10 @@ def _patch_trl_rl_trainers(trainer_file = "grpo_trainer"): "include_tokens_per_second" : False, "include_num_input_tokens_seen" : False, "auto_find_batch_size" : True, # Auto /2 batch size - "dataloader_persistent_workers" : True, # Keeps dataloader in RAM "dataloader_pin_memory" : True, - # "dataloader_prefetch_factor" : 2, # Might fail so disable for now + # Might fail so disable for now + # "dataloader_persistent_workers" : True, # Keeps dataloader in RAM + # "dataloader_prefetch_factor" : 2, # "dataloader_num_workers" : 2, # Default is 0 means 1 } for k, v in replacements.items():