aleafy commited on
Commit
5e2f182
·
1 Parent(s): c8013a6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -251,7 +251,7 @@ def dummy_process(input_fg, input_bg):
251
  text_uncond.to(dtype=torch.float16),
252
  cond_tensor.to(dtype=torch.float16)
253
  )
254
- inf_pipe.to(torch.float16)
255
  latent_pred = inf_pipe(
256
  latent=init_latent,
257
  text_cond=text_cond,
 
251
  text_uncond.to(dtype=torch.float16),
252
  cond_tensor.to(dtype=torch.float16)
253
  )
254
+ inf_pipe.unet.to(torch.float16)
255
  latent_pred = inf_pipe(
256
  latent=init_latent,
257
  text_cond=text_cond,