changed pipeline.to to pipeline=pipeline.to

2024-08-07 18:33:40 +02:00
parent a5d5331ddd
commit d2db3afd26
1 changed files with 2 additions and 1 deletions
@@ -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,)