test_video

Paused

user029182 commited on 22 days ago

Commit

aea3b85

verified ·

1 Parent(s): 59b5bd8

Update video_model.py

Files changed (1) hide show

video_model.py CHANGED Viewed

@@ -13,7 +13,7 @@ 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,

     quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
     torch_dtype=torch.bfloat16,
 )
+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,