Spaces:
Paused
Paused
Update api/seedvr/seedvr_aduc_pipeline.py
Browse files
api/seedvr/seedvr_aduc_pipeline.py
CHANGED
|
@@ -202,7 +202,7 @@ class SeedAducPipeline:
|
|
| 202 |
logging.info(f"✅ Video successfully saved to: {output_filepath}")
|
| 203 |
return str(output_filepath)
|
| 204 |
|
| 205 |
-
|
| 206 |
# --- CORREÇÃO IMPORTANTE ---
|
| 207 |
# Restaura o LTX para seus dispositivos corretos (main e vae)
|
| 208 |
#if gpu_manager.requires_memory_swap():
|
|
@@ -215,7 +215,7 @@ class SeedAducPipeline:
|
|
| 215 |
# main_device_str=str(ltx_main_device),
|
| 216 |
# vae_device_str=str(ltx_vae_device)
|
| 217 |
#)
|
| 218 |
-
|
| 219 |
|
| 220 |
# --- PONTO DE ENTRADA E INSTANCIAÇÃO ---
|
| 221 |
# A instância é criada na primeira importação.
|
|
|
|
| 202 |
logging.info(f"✅ Video successfully saved to: {output_filepath}")
|
| 203 |
return str(output_filepath)
|
| 204 |
|
| 205 |
+
finally:
|
| 206 |
# --- CORREÇÃO IMPORTANTE ---
|
| 207 |
# Restaura o LTX para seus dispositivos corretos (main e vae)
|
| 208 |
#if gpu_manager.requires_memory_swap():
|
|
|
|
| 215 |
# main_device_str=str(ltx_main_device),
|
| 216 |
# vae_device_str=str(ltx_vae_device)
|
| 217 |
#)
|
| 218 |
+
logging.info(f"[SWAP] LTX service restored to Main: {ltx_main_device}, VAE: {ltx_vae_device}.")
|
| 219 |
|
| 220 |
# --- PONTO DE ENTRADA E INSTANCIAÇÃO ---
|
| 221 |
# A instância é criada na primeira importação.
|