decode_latents typing

piercus · Jan 27, 2024 · cb2eee2 · cb2eee2
1 parent df3ccc8
commit cb2eee2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/refiners/foundationals/latent_diffusion/multi_diffusion.py b/src/refiners/foundationals/latent_diffusion/multi_diffusion.py
@@ -83,7 +83,7 @@ def device(self) -> Device:
     def dtype(self) -> DType:
         return self.ldm.dtype
 
-    def decode_latents(self, x: Tensor) -> Image.Image | list[Image.Image]:
+    def decode_latents(self, x: Tensor) -> Image.Image:
         return self.ldm.lda.decode_latents(x=x)
 
     @staticmethod