# Thông số kỹ thuật họ mô hình LTX
ltx_video_original = {
    "resolution": "768x512",  # Mô hình cơ sở
    "max_duration": 5,  # giây
    "fps": range(24, 31),  # 24-30 FPS
    "diffusion_steps": 20,
    "h100_time": "4 giây cho video 5 giây",
    "rtx4090_time": "11 giây cho video 5 giây"
}
 
ltx2_capabilities = {
    "resolution": "lên đến 3840x2160",  # 4K tự nhiên
    "max_duration": 10,  # giây được xác nhận, 60s thử nghiệm
    "fps": "lên đến 50",
    "synchronized_audio": True,
    "rtx4090_4k_time": "9-12 phút cho 10 giây"
}

# Sử dụng LTX-Video với thư viện Diffusers
from diffusers import LTXVideoPipeline
import torch
 
# Khởi tạo với tối ưu hóa bộ nhớ
pipe = LTXVideoPipeline.from_pretrained(
    "Lightricks/LTX-Video",
    torch_dtype=torch.float16
).to("cuda")
 
# Tạo với các bước có thể cấu hình
video = pipe(
    prompt="Góc nhìn trên cao của cảnh quan núi lúc bình minh",
    num_inference_steps=8,  # Chế độ bản nháp nhanh
    height=704,
    width=1216,
    num_frames=121,  # ~4 giây ở 30fps
    guidance_scale=1.0  # Không cần CFG
).frames

# Quy trình sản xuất cho mở rộng video
from ltx_video import LTXPipeline
 
pipeline = LTXPipeline(model="ltx-2", device="cuda")
 
# Tạo phân đoạn ban đầu
initial = pipeline.generate(
    prompt="Robot khám phá tàn tích cổ đại",
    resolution=(1920, 1080),
    duration=5
)
 
# Mở rộng với hướng dẫn khung hình chính
extended = pipeline.extend_video(
    video=initial,
    direction="forward",
    keyframes=[
        {"frame": 150, "prompt": "Robot phát hiện đồ tạo tác"},
        {"frame": 300, "prompt": "Đồ tạo tác kích hoạt"}
    ]
)