Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_pipeline.py
Browse files
api/ltx/ltx_aduc_pipeline.py
CHANGED
|
@@ -28,13 +28,14 @@ from managers.gpu_manager import gpu_manager
|
|
| 28 |
from transformers import T5EncoderModel, T5Tokenizer
|
| 29 |
from safetensors import safe_open
|
| 30 |
from managers.gpu_manager import gpu_manager
|
| 31 |
-
|
| 32 |
|
| 33 |
|
| 34 |
# ==============================================================================
|
| 35 |
# --- INITIAL SETUP & CONFIGURATION ---
|
| 36 |
# ==============================================================================
|
| 37 |
|
|
|
|
| 38 |
warnings.filterwarnings("ignore")
|
| 39 |
logging.getLogger("huggingface_hub").setLevel(logging.ERROR)
|
| 40 |
logging.basicConfig(level=logging.INFO, format='[%(levelname)s] %(message)s')
|
|
|
|
| 28 |
from transformers import T5EncoderModel, T5Tokenizer
|
| 29 |
from safetensors import safe_open
|
| 30 |
from managers.gpu_manager import gpu_manager
|
| 31 |
+
|
| 32 |
|
| 33 |
|
| 34 |
# ==============================================================================
|
| 35 |
# --- INITIAL SETUP & CONFIGURATION ---
|
| 36 |
# ==============================================================================
|
| 37 |
|
| 38 |
+
from utils.debug_utils import log_function_io
|
| 39 |
warnings.filterwarnings("ignore")
|
| 40 |
logging.getLogger("huggingface_hub").setLevel(logging.ERROR)
|
| 41 |
logging.basicConfig(level=logging.INFO, format='[%(levelname)s] %(message)s')
|