Update stableDiffusion.py

soten355 · Aug 5, 2023 · 767f8a3 · 767f8a3
1 parent 8e1b752
commit 767f8a3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/stableDiffusionDiffusers/stableDiffusion.py b/stableDiffusionDiffusers/stableDiffusion.py
@@ -256,8 +256,9 @@ def compileModels(
                         torch_dtype = self.dtype,
                         safety_checker = None
                     )
-
-                    self.pipeline.enable_model_cpu_offload(gpu_id = self.torchDevice)
+
+                    ## To be fully implemented with HuggingFace updates their code for my request
+                    # self.pipeline.enable_model_cpu_offload(gpu_id = self.torchDevice)
 
                     if self.VAE != None:
                         print("...using",self.VAE,"as VAE...")
@@ -662,4 +663,4 @@ def clearMemory():
     for image in images:
         location = "stableDiffusionDiffusers/final0000"+str(seed)+".png"
         image.save(location)
-        i += 1
+        i += 1