Setting it to total cpu_count // 4

2026-02-20 06:32:01 +00:00 · 2026-02-20 06:32:01 +00:00 · 798bfb8f6f
commit 798bfb8f6f
parent fdeccec259
1 changed files with 1 additions and 0 deletions
--- a/studio/backend/core/training/trainer.py
+++ b/studio/backend/core/training/trainer.py
@ -867,6 +867,7 @@ class UnslothTrainer:
                        self.trainer,
                        instruction_part=instruction_part,
                        response_part=response_part,
+                        num_proc=config_args.get("dataset_num_proc", max(1, os.cpu_count() // 4)),
                    )
                    print("Train on responses only configured successfully\n")
                except Exception as e: