# Специфікації родини моделей LTX
ltx_video_original = {
    "resolution": "768x512",  # Базова модель
    "max_duration": 5,  # секунди
    "fps": range(24, 31),  # 24-30 FPS
    "diffusion_steps": 20,
    "h100_time": "4 секунди для 5-секундного відео",
    "rtx4090_time": "11 секунд для 5-секундного відео"
}
 
ltx2_capabilities = {
    "resolution": "до 3840x2160",  # Нативний 4K
    "max_duration": 10,  # секунди підтверджено, 60s експериментально
    "fps": "до 50",
    "synchronized_audio": True,
    "rtx4090_4k_time": "9-12 хвилин для 10 секунд"
}

# Використання LTX-Video з бібліотекою Diffusers
from diffusers import LTXVideoPipeline
import torch
 
# Ініціалізація з оптимізацією пам'яті
pipe = LTXVideoPipeline.from_pretrained(
    "Lightricks/LTX-Video",
    torch_dtype=torch.float16
).to("cuda")
 
# Генерація з налаштовуваними кроками
video = pipe(
    prompt="Аерозйомка гірського ландшафту на світанку",
    num_inference_steps=8,  # Режим швидкої чернетки
    height=704,
    width=1216,
    num_frames=121,  # ~4 секунди при 30fps
    guidance_scale=1.0  # CFG не потрібен
).frames

# Виробничий конвеєр для подовження відео
from ltx_video import LTXPipeline
 
pipeline = LTXPipeline(model="ltx-2", device="cuda")
 
# Генерація початкового сегмента
initial = pipeline.generate(
    prompt="Робот досліджує стародавні руїни",
    resolution=(1920, 1080),
    duration=5
)
 
# Подовження з керівництвом ключовими кадрами
extended = pipeline.extend_video(
    video=initial,
    direction="forward",
    keyframes=[
        {"frame": 150, "prompt": "Робот знаходить артефакт"},
        {"frame": 300, "prompt": "Артефакт активується"}
    ]
)