Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix inpaint device problem in --always-gpu mode. (#1420)

Browse Source 
2.1.841
This commit is contained in:
lllyasviel
2023-12-14 19:15:16 -08:00
committed by GitHub
parent a39c1d1349
commit 179bcb2c4e
5 changed files with 37 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/core.py
+1 -1
View File
@@ -191,7 +191,7 @@ def encode_vae_inpaint(vae, pixels, mask):
latent_mask = mask[:, None, :, :]
latent_mask = torch.nn.functional.interpolate(latent_mask, size=(H * 8, W * 8), mode="bilinear").round()
latent_mask = torch.nn.functional.max_pool2d(latent_mask, (8, 8)).round()
latent_mask = torch.nn.functional.max_pool2d(latent_mask, (8, 8)).round().to(latent)
return latent, latent_mask