Update video_model.py

video_model.py

@@ -1,11 +1,23 @@
 import torch
 
-from diffusers import LTXPipeline, LTXImageToVideoPipeline
+from diffusers import LTXPipeline, LTXImageToVideoPipeline, LTXVideoTransformer3DModel, GGUFQuantizationConfig
 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-t2v_pipe = LTXPipeline.from_pretrained("Lightricks/LTX-Video", torch_dtype=torch.bfloat16)
+ckpt_path = (
+    "https://huggingface.co/city96/LTX-Video-gguf/blob/main/ltx-video-2b-v0.9-Q3_K_S.gguf"
+)
+transformer = LTXVideoTransformer3DModel.from_single_file(
+    ckpt_path,
+    quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
+    torch_dtype=torch.bfloat16,
+)
+pipe = LTXPipeline.from_pretrained(
+    "Lightricks/LTX-Video",
+    transformer=transformer,
+    torch_dtype=torch.bfloat16,
+)
 t2v_pipe.to(device)
 
 i2v_pipe = LTXImageToVideoPipeline.from_pipe(t2v_pipe)