Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_manager.py
Browse files
api/ltx/ltx_aduc_manager.py
CHANGED
|
@@ -11,6 +11,7 @@ from typing import List, Optional, Tuple, Union, Dict
|
|
| 11 |
import threading
|
| 12 |
import sys
|
| 13 |
import torch
|
|
|
|
| 14 |
|
| 15 |
# --- Importações da arquitetura ADUC-SDR ---
|
| 16 |
from managers.gpu_manager import gpu_manager
|
|
@@ -45,6 +46,12 @@ logging.basicConfig(level=logging.DEBUG)
|
|
| 45 |
logger.setLevel(logging.DEBUG)
|
| 46 |
|
| 47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
from ltx_video.pipelines.pipeline_ltx_video import ConditioningItem
|
| 49 |
|
| 50 |
# ==============================================================================
|
|
|
|
| 11 |
import threading
|
| 12 |
import sys
|
| 13 |
import torch
|
| 14 |
+
from dataclasses import dataclass
|
| 15 |
|
| 16 |
# --- Importações da arquitetura ADUC-SDR ---
|
| 17 |
from managers.gpu_manager import gpu_manager
|
|
|
|
| 46 |
logger.setLevel(logging.DEBUG)
|
| 47 |
|
| 48 |
|
| 49 |
+
@dataclass
|
| 50 |
+
class LatentConditioningItem:
|
| 51 |
+
latent_tensor: torch.Tensor
|
| 52 |
+
media_frame_number: int
|
| 53 |
+
conditioning_strength: float
|
| 54 |
+
|
| 55 |
from ltx_video.pipelines.pipeline_ltx_video import ConditioningItem
|
| 56 |
|
| 57 |
# ==============================================================================
|