eeuuia commited on
Commit
b54d196
·
verified ·
1 Parent(s): d27849f

Update api/ltx/ltx_aduc_manager.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_aduc_manager.py +7 -0
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
  # ==============================================================================