Update video_model.py

video_model.py

@@ -13,9 +13,11 @@ transformer = LTXVideoTransformer3DModel.from_single_file(
     quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
     torch_dtype=torch.bfloat16,
 )
+vae = AutoencoderKLLTXVideo.from_single_file("https://huggingface.co/city96/LTX-Video-gguf/blob/main/LTX-Video-VAE-BF16.safetensors", torch_dtype=torch.bfloat16)
 t2v_pipe = LTXPipeline.from_pretrained(
     "Lightricks/LTX-Video",
     transformer=transformer,
+    vae=vae,
     torch_dtype=torch.bfloat16,
 )
 t2v_pipe.to(device)