Update upscaler_specialist.py

upscaler_specialist.py

@@ -73,8 +73,13 @@ class UpscalerSpecialist:
         try:
             logger.info(f"[Upscaler] Recebido shape {latents.shape}. Executando upscale em {self.device}...")
             result = self.pipe_upsample(latents=latents, output_type="latent")
-            logger.info(f"[Upscaler] Upscale concluído. Novo shape: {result.latents.shape}")
-            return result.latents
+            
+            # [CORREÇÃO APLICADA] O resultado está no atributo .images
+            output_tensor = result.images
+            
+            logger.info(f"[Upscaler] Upscale concluído. Novo shape: {output_tensor.shape}")
+            return output_tensor
+        
         except Exception as e:
             logger.error(f"[Upscaler] Erro durante upscale: {e}", exc_info=True)
             return latents
@@ -83,4 +88,4 @@ class UpscalerSpecialist:
 # ---------------------------
 # Singleton global
 # ---------------------------
-upscaler_specialist_singleton = UpscalerSpecialist()
+upscaler_specialist_singleton = UpscalerSpecialist()