Update app.py
Browse files
app.py
CHANGED
|
@@ -83,7 +83,7 @@ pipe.load_lora_weights("ford442/sdxl-vae-bf16", weight_name="LoRA/UltraReal.safe
|
|
| 83 |
|
| 84 |
pipe.to(device=device, dtype=torch.bfloat16)
|
| 85 |
#pipe.to(device)
|
| 86 |
-
|
| 87 |
upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
|
| 88 |
|
| 89 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
|
| 83 |
|
| 84 |
pipe.to(device=device, dtype=torch.bfloat16)
|
| 85 |
#pipe.to(device)
|
| 86 |
+
pipe.vae=vaeX.to('cpu')
|
| 87 |
upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cpu'))
|
| 88 |
|
| 89 |
MAX_SEED = np.iinfo(np.int32).max
|