Update api/ltx_server.py

api/ltx_server.py

@@ -770,7 +770,6 @@ class VideoService:
             print("".join(traceback.format_exception(type(e), e, e.__traceback__)))
             raise
 
-
 # ltx_server.py
 
     def generate(
@@ -837,6 +836,17 @@ class VideoService:
                 conditioning_items.append(ConditioningItem(end_tensor, last_frame_index, float(end_image_weight)))
             print(f"[DEBUG] Conditioning items: {len(conditioning_items)}")
 
+        # --- LÓGICA DE CONVERSÃO DO STG_MODE ---
+        stg_mode_str = self.config.get("stg_mode", "attention_values")
+        stg_mode_map = {
+            "attention_values": "AttentionValues",
+            "attention_skip": "AttentionSkip",
+            "residual": "Residual",
+            "transformer_block": "TransformerBlock"
+        }
+        stg_mode_enum_key = stg_mode_map.get(stg_mode_str.lower(), "AttentionValues")
+        # --- FIM DA LÓGICA DE CONVERSÃO ---
+
         call_kwargs = {
             "prompt": prompt,
             "negative_prompt": negative_prompt,
@@ -857,7 +867,7 @@ class VideoService:
             "mixed_precision": (self.config.get("precision") == "mixed_precision"),
             "offload_to_cpu": False,
             "enhance_prompt": False,
-            "skip_layer_strategy": SkipLayerStrategy[self.config.get("stg_mode", "AttentionValues")],
+            "skip_layer_strategy": SkipLayerStrategy[stg_mode_enum_key],
         }
         print(f"[DEBUG] output_type={call_kwargs['output_type']} skip_layer_strategy={call_kwargs['skip_layer_strategy']}")
 
@@ -998,7 +1008,7 @@ class VideoService:
             else:
                 lat_b1, lat_b2 = None, None
             
-            latents_parts = [p for p in [lat_a1, lat_a2, lat_b1, lat_b2] if p is not None]
+            latents_parts = [p for p in [lat_a1, lat_a2, lat_b1, lat_b2] if p is not None and p.shape[2] > 1]
             if not latents_parts:
                 latents_parts = [latents_cpu]
 
@@ -1075,6 +1085,5 @@ class VideoService:
             except Exception as e:
                 print(f"[DEBUG] finalize() no finally falhou: {e}")
 
-
 print("Criando instância do VideoService. O carregamento do modelo começará agora...")
 video_generation_service = VideoService()