From d2db3afd26703ecc9229e8807cb33e4ea04def66 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ma=C3=ABlys=20Jusseaux?= Date: Wed, 7 Aug 2024 18:33:40 +0200 Subject: [PATCH] changed pipeline.to to pipeline=pipeline.to --- diffusers_in_comfy.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/diffusers_in_comfy.py b/diffusers_in_comfy.py index 5b256a2..49fb3cd 100644 --- a/diffusers_in_comfy.py +++ b/diffusers_in_comfy.py @@ -95,7 +95,7 @@ class Text2ImgStableDiffusionPipeline: pipeline.enable_model_cpu_offload() - pipeline.to(device) + pipeline = pipeline.to(device) print(f'Low VRAM options is {low_vram}, sending pipeline to {device}') return (pipeline,) @@ -373,6 +373,7 @@ class LoRALoader: pipeline.set_adapters([lora_name], adapter_weights=[lora_scale]) + return (pipeline,)