Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -223,7 +223,7 @@ def scheduler_swap_callback(pipeline, step_index, timestep, callback_kwargs):
|
|
223 |
torch.set_float32_matmul_precision("highest")
|
224 |
#callback_kwargs["latents"] = callback_kwargs["latents"].to(torch.bfloat16)
|
225 |
#pipe.unet.to(torch.float64)
|
226 |
-
pipeline.unet.set_default_attn_processor() ## custom ##
|
227 |
# pipe.vae = vae_a
|
228 |
# pipe.unet = unet_a
|
229 |
torch.backends.cudnn.deterministic = False
|
|
|
223 |
torch.set_float32_matmul_precision("highest")
|
224 |
#callback_kwargs["latents"] = callback_kwargs["latents"].to(torch.bfloat16)
|
225 |
#pipe.unet.to(torch.float64)
|
226 |
+
#pipeline.unet.set_default_attn_processor() ## custom ##
|
227 |
# pipe.vae = vae_a
|
228 |
# pipe.unet = unet_a
|
229 |
torch.backends.cudnn.deterministic = False
|