ok

config/training_config_gemma3_1b.yaml

@@ -91,7 +91,7 @@ progressive_stages:
     dataset_config:
       # Mixture-of-Thoughts specific settings
       streaming: true  # Use streaming for large dataset
-      max_samples: 30000  # Limit samples for faster training
+      max_samples: 300000  # Limit samples for faster training
       split: "train"
 
 evaluation:

flake.nix

@@ -61,6 +61,12 @@
                     pytestCheckPhase = "echo 'Skipping tests'";
                     pythonImportsCheck = [];  # Disable import checks
                   });
+                  curl-cffi = python-super.curl-cffi.overrideAttrs (oldAttrs: {
+                    doCheck = false;
+                    doInstallCheck = false;
+                    pytestCheckPhase = "echo 'Skipping tests'";
+                    pythonImportsCheck = [];  # Disable import checks
+                  });
                 };
               };
             })
@@ -192,4 +198,4 @@
           LC_ALL = "en_US.UTF-8";
         };
       });
-}
+}