Spaces:

habulaj
/

squaadai

Running on Zero

forplaytvplus commited on Jun 29

Commit

76340d8

•

1 Parent(s): 04c24b9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ def generate(
                 adapter_name2 = cached_loras[lora_key2]
             pipe.set_adapters([adapter_name1, adapter_name2], adapter_weights=[lora_scale, lora_scale2])
-    enable_model_cpu_offload()
     generator = torch.Generator().manual_seed(seed)
     if not use_negative_prompt:

                 adapter_name2 = cached_loras[lora_key2]
             pipe.set_adapters([adapter_name1, adapter_name2], adapter_weights=[lora_scale, lora_scale2])
+    pipe.enable_model_cpu_offload()
     generator = torch.Generator().manual_seed(seed)
     if not use_negative_prompt: