diff --git a/ocr-service/main.py b/ocr-service/main.py
index 8e4d2288..3a2c084e 100644
--- a/ocr-service/main.py
+++ b/ocr-service/main.py
@@ -366,12 +366,13 @@ async def train_model(
             os.makedirs(checkpoint_dir, exist_ok=True)
 
             cmd = [
-                "ketos", "--workers", "0",
+                "ketos", "--workers", "0", "--device", "cpu", "--threads", "2",
                 "train",
                 "-f", "path",
                 "-o", checkpoint_dir,
                 "-q", "fixed",
-                "-N", "50",
+                "-N", "10",
+                "-B", "1",
             ]
             if os.path.exists(KRAKEN_MODEL_PATH):
                 cmd += ["-i", KRAKEN_MODEL_PATH]
@@ -457,11 +458,11 @@ async def segtrain_model(
             os.makedirs(checkpoint_dir, exist_ok=True)
 
             cmd = [
-                "ketos", "--workers", "0",
+                "ketos", "--workers", "0", "--device", "cpu", "--threads", "2",
                 "segtrain",
                 "-o", checkpoint_dir,
                 "-q", "fixed",
-                "-N", "50",
+                "-N", "10",
             ]
             if os.path.exists(blla_model_path):
                 cmd += ["-i", blla_model_path]