Update upscaler_specialist.py
Browse files- upscaler_specialist.py +8 -0
upscaler_specialist.py
CHANGED
|
@@ -56,3 +56,11 @@ class UpscalerSpecialist:
|
|
| 56 |
return latents
|
| 57 |
|
| 58 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
return latents
|
| 57 |
|
| 58 |
|
| 59 |
+
# Instanciação Singleton
|
| 60 |
+
# Depende do VAE do ltx_manager, então o obtemos de lá.
|
| 61 |
+
try:
|
| 62 |
+
base_vae_for_upscaler = ltx_manager_singleton.workers[0].pipeline.vae
|
| 63 |
+
upscaler_specialist_singleton = UpscalerSpecialist(base_vae=base_vae_for_upscaler)
|
| 64 |
+
except Exception as e:
|
| 65 |
+
logger.error(f"Não foi possível inicializar o UpscalerSpecialist Singleton: {e}")
|
| 66 |
+
upscaler_specialist_singleton = None
|