Spaces:

habulaj
/

squaadai

Running on Zero

forplaytvplus commited on Jun 26

Commit

3f45c7e

•

1 Parent(s): ddb9a25

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -135,6 +135,7 @@ def generate(
             pipe.load_lora_weights(lora2, adapter_name="2")
             pipe.set_adapters(["1", "2"], adapter_weights=[lora_scale, lora_scale2])
     pipe.enable_model_cpu_offload()
     generator = torch.Generator().manual_seed(seed)

             pipe.load_lora_weights(lora2, adapter_name="2")
             pipe.set_adapters(["1", "2"], adapter_weights=[lora_scale, lora_scale2])
+    reset_device_map()
     pipe.enable_model_cpu_offload()
     generator = torch.Generator().manual_seed(seed)