euiia commited on
Commit
b88a339
·
verified ·
1 Parent(s): ea16365

Update upscaler_specialist.py

Browse files
Files changed (1) hide show
  1. upscaler_specialist.py +3 -3
upscaler_specialist.py CHANGED
@@ -72,10 +72,10 @@ class UpscalerSpecialist:
72
 
73
  try:
74
  logger.info(f"[Upscaler] Recebido shape {latents.shape}. Executando upscale em {self.device}...")
75
- result = self.pipe_upsample(latents=latents, output_type="latent")
76
 
77
- # [CORREÇÃO APLICADA] O resultado está no atributo .images
78
- output_tensor = result.images
 
79
 
80
  logger.info(f"[Upscaler] Upscale concluído. Novo shape: {output_tensor.shape}")
81
  return output_tensor
 
72
 
73
  try:
74
  logger.info(f"[Upscaler] Recebido shape {latents.shape}. Executando upscale em {self.device}...")
 
75
 
76
+ # [CORREÇÃO FINAL] Conforme a documentação oficial, o resultado está em .frames
77
+ result = self.pipe_upsample(latents=latents, output_type="latent")
78
+ output_tensor = result.frames
79
 
80
  logger.info(f"[Upscaler] Upscale concluído. Novo shape: {output_tensor.shape}")
81
  return output_tensor