From 767f8a3293c74aa69c6de30524a5902c99b9889a Mon Sep 17 00:00:00 2001 From: soten355 Date: Sat, 5 Aug 2023 14:01:55 -0700 Subject: [PATCH] Update stableDiffusion.py --- stableDiffusionDiffusers/stableDiffusion.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/stableDiffusionDiffusers/stableDiffusion.py b/stableDiffusionDiffusers/stableDiffusion.py index f06bb6d..c195496 100644 --- 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 \ No newline at end of file + i += 1