Spaces:
Runtime error
Runtime error
acc
Browse files
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,
|