ford442 commited on
Commit
395f049
·
verified ·
1 Parent(s): 8bf85b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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