From 7c4f18ce51293f0fc17ae1ffab84cf2bbe646052 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Fri, 17 Apr 2026 12:03:39 -0600 Subject: [PATCH] Fix ernie unpatchify --- extensions_built_in/diffusion_models/ernie_image/ernie_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions_built_in/diffusion_models/ernie_image/ernie_image.py b/extensions_built_in/diffusion_models/ernie_image/ernie_image.py index d7177add..dba12046 100644 --- a/extensions_built_in/diffusion_models/ernie_image/ernie_image.py +++ b/extensions_built_in/diffusion_models/ernie_image/ernie_image.py @@ -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]