Update app.py
Browse files
app.py
CHANGED
|
@@ -56,6 +56,7 @@ unet.load_state_dict(load_file(hf_hub_download(REPO, CHECKPOINT), device="cuda")
|
|
| 56 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 57 |
BASE, unet=unet, torch_dtype=torch.float16, variant="fp16", safety_checker=False
|
| 58 |
).to("cuda")
|
|
|
|
| 59 |
|
| 60 |
if USE_TAESD:
|
| 61 |
pipe.vae = AutoencoderTiny.from_pretrained(
|
|
|
|
| 56 |
pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 57 |
BASE, unet=unet, torch_dtype=torch.float16, variant="fp16", safety_checker=False
|
| 58 |
).to("cuda")
|
| 59 |
+
unet = unet.to(dtype=torch.float16)
|
| 60 |
|
| 61 |
if USE_TAESD:
|
| 62 |
pipe.vae = AutoencoderTiny.from_pretrained(
|