mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix (#521)

Browse Source 
* fix

* fix
This commit is contained in:
lllyasviel 2023-10-03 10:59:53 -07:00 committed by GitHub
parent bbae307ef2
commit 2f31d9e5a7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fooocus_version.py
View File

@ -1 +1 @@
version = '2.0.80' version = '2.0.81'

10
modules/core.py
View File

@ -127,13 +127,19 @@ def generate_empty_latent(width=1024, height=1024, batch_size=1):
@torch.no_grad() @torch.no_grad()
@torch.inference_mode() @torch.inference_mode()
def decode_vae(vae, latent_image, tiled=False): def decode_vae(vae, latent_image, tiled=False):
return (opVAEDecodeTiled if tiled else opVAEDecode).decode(samples=latent_image, vae=vae)[0] if tiled:
return opVAEDecodeTiled.decode(samples=latent_image, vae=vae, tile_size=512)[0]
else:
return opVAEDecode.decode(samples=latent_image, vae=vae)[0]
@torch.no_grad() @torch.no_grad()
@torch.inference_mode() @torch.inference_mode()
def encode_vae(vae, pixels, tiled=False): def encode_vae(vae, pixels, tiled=False):
return (opVAEEncodeTiled if tiled else opVAEEncode).encode(pixels=pixels, vae=vae)[0] if tiled:
return opVAEEncodeTiled.encode(pixels=pixels, vae=vae, tile_size=512)[0]
else:
return opVAEEncode.encode(pixels=pixels, vae=vae)[0]
@torch.no_grad() @torch.no_grad()