Update video_model.py

video_model.py

@@ -10,15 +10,15 @@ ckpt_path = (
 )
 transformer = LTXVideoTransformer3DModel.from_single_file(
     ckpt_path,
-    quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
-    torch_dtype=torch.bfloat16,
+    quantization_config=GGUFQuantizationConfig(compute_dtype=torch.float32),
+    torch_dtype=torch.float32,
 )
 vae = AutoencoderKLLTXVideo.from_single_file("https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.safetensors", torch_dtype=torch.float32)
 t2v_pipe = LTXPipeline.from_pretrained(
     "Lightricks/LTX-Video",
     transformer=transformer,
     vae=vae,
-    torch_dtype=torch.bfloat16,
+    torch_dtype=torch.float32,
 )
 t2v_pipe.to(device)