Added support for training on primary gpu with low_vram flag. Updated example script to remove creepy horse sample at that seed

2024-08-11 09:54:30 -06:00
parent fa02e774b0
commit ec1ea7aa0e
4 changed files with 30 additions and 13 deletions
--- a/toolkit/config_modules.py
+++ b/toolkit/config_modules.py
@@ -411,6 +411,7 @@ class ModelConfig:

        # only for flux for now
        self.quantize = kwargs.get("quantize", False)
+        self.low_vram = kwargs.get("low_vram", False)
        pass