Update app.py

app.py

@@ -22,13 +22,19 @@ torch.set_grad_enabled(False)
 gc.collect()
 torch.cuda.empty_cache()
 
-ckpt_path = Path("a-r-r-o-w/LTX-Video-0.9.1-diffusers")
-single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"
-transformer = LTXVideoTransformer3DModel.from_single_file(
-    single_file_url, torch_dtype=torch.bfloat16
+ckpt_path = Path("Lightricks/LTX-Video-0.9.5")
+#single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"
+transformer = LTXVideoTransformer3DModel.from_pretrained(
+    ckpt_path,
+    subfolder="transformer",
+    torch_dtype=torch.bfloat16
+).to("cuda")
+
+vae = AutoencoderKLLTXVideo.from_pretrained(
+    ckpt_path,
+    subfolder="vae",
+    torch_dtype=torch.bfloat16
 )
-vae = AutoencoderKLLTXVideo.from_single_file(
-    single_file_url, torch_dtype=torch.bfloat16)
 vae.eval()
 vae = vae.to("cuda")