# Especificacions de la família de models LTX
ltx_video_original = {
    "resolution": "768x512",  # Model base
    "max_duration": 5,  # segons
    "fps": range(24, 31),  # 24-30 FPS
    "diffusion_steps": 20,
    "h100_time": "4 segons per vídeo de 5 segons",
    "rtx4090_time": "11 segons per vídeo de 5 segons"
}
 
ltx2_capabilities = {
    "resolution": "fins a 3840x2160",  # 4K natiu
    "max_duration": 10,  # segons confirmats, 60s experimental
    "fps": "fins a 50",
    "synchronized_audio": True,
    "rtx4090_4k_time": "9-12 minuts per 10 segons"
}

# Utilitzant LTX-Video amb la biblioteca Diffusers
from diffusers import LTXVideoPipeline
import torch
 
# Inicialitzar amb optimització de memòria
pipe = LTXVideoPipeline.from_pretrained(
    "Lightricks/LTX-Video",
    torch_dtype=torch.float16
).to("cuda")
 
# Generar amb passos configurables
video = pipe(
    prompt="Vista aèria de paisatge de muntanya a l'alba",
    num_inference_steps=8,  # Mode esborrany ràpid
    height=704,
    width=1216,
    num_frames=121,  # ~4 segons a 30fps
    guidance_scale=1.0  # No es necessita CFG
).frames

# Pipeline de producció per a extensió de vídeo
from ltx_video import LTXPipeline
 
pipeline = LTXPipeline(model="ltx-2", device="cuda")
 
# Generar segment inicial
initial = pipeline.generate(
    prompt="Robot explorant ruïnes antigues",
    resolution=(1920, 1080),
    duration=5
)
 
# Estendre amb guia de fotogrames clau
extended = pipeline.extend_video(
    video=initial,
    direction="forward",
    keyframes=[
        {"frame": 150, "prompt": "Robot descobreix artefacte"},
        {"frame": 300, "prompt": "L'artefacte s'activa"}
    ]
)