Spaces:

1inkusFace
/

StableDiffusion-3.5-Large-lora

Running on Zero

1inkusFace commited on 24 days ago

Commit

390af65

verified ·

1 Parent(s): f863db1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -89,7 +89,7 @@ pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safe
 pipe.to(device=device, dtype=torch.bfloat16)
 #pipe.to(device)
-pipe.vae=vaeX #.to('cpu')
 upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
 MAX_SEED = np.iinfo(np.int32).max

 pipe.to(device=device, dtype=torch.bfloat16)
 #pipe.to(device)
+pipe.vae=vaeX.to('cpu')
 upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
 MAX_SEED = np.iinfo(np.int32).max