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]