Spaces:

user029182
/

test_video

Paused

user029182 commited on Mar 22

Commit

f53286d

verified ·

1 Parent(s): 57da289

Update video_model.py

Files changed (1) hide show

video_model.py CHANGED Viewed

@@ -13,9 +13,11 @@ transformer = LTXVideoTransformer3DModel.from_single_file(
     quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
     torch_dtype=torch.bfloat16,
 )
 t2v_pipe = LTXPipeline.from_pretrained(
     "Lightricks/LTX-Video",
     transformer=transformer,
     torch_dtype=torch.bfloat16,
 )
 t2v_pipe.to(device)

     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)