Bug fix

build_and_push_docker.yaml

@@ -0,0 +1,8 @@
+#!/usr/bin/env bash
+
+echo "Docker builds from the repo, not this dir. Make sure changes are pushed to the repo."
+# wait 2 seconds
+sleep 2
+docker build --build-arg CACHEBUST=$(date +%s) -t aitoolkit:latest -f docker/Dockerfile .
+docker tag aitoolkit:latest ostris/aitoolkit:latest
+docker push ostris/aitoolkit:latest

toolkit/stable_diffusion_model.py

@@ -468,8 +468,8 @@ class StableDiffusion:
                 transformer_path,
                 subfolder=subfolder,
                 torch_dtype=dtype,
-                low_cpu_mem_usage=False,
-                device_map=None
+                # low_cpu_mem_usage=False,
+                # device_map=None
             )
             transformer.to(self.device_torch, dtype=dtype)
             flush()