ponio-kabu-logo-creator

Running on Zero

bunarivenna commited on Jul 31, 2024

Commit

c725615

verified ·

1 Parent(s): 48e2519

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ if torch.cuda.is_available():
     pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
     pipe.set_adapters("dalle")
-    pipe.to("cuda")

     pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
     pipe.set_adapters("dalle")
+    pipe.enable_model_cpu_offload()