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