github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix ernie unpatchify

Browse Source
This commit is contained in:
Jaret Burkett
2026-04-17 12:03:39 -06:00
parent 8cb9649382
commit 7c4f18ce51
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/diffusion_models/ernie_image/ernie_image.py
View File

@@ -251,7 +251,7 @@ class ErnieImageModel(BaseModel):
latents = latents * bn_std + bn_mean
# Unpatchify
latents = self._unpatchify_latents(latents)
latents = self.pipeline._unpatchify_latents(latents)
# Decode
images = self.vae.decode(latents, return_dict=False)[0]