Spaces:
Runtime error
Runtime error
Update upscaler_specialist.py
Browse files- 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
|
| 78 |
-
|
|
|
|
| 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
|