Update app.py

app.py

@@ -26,23 +26,16 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 MAX_SEED = np.iinfo(np.int64).max
 
 single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"
-#vae_url = 'https://huggingface.co/spacepxl/ltx-video-0.9-vae-finetune/ltx-video-v0.9-vae_finetune_decoder_111k_smooth.safetensors'
-
-
-#vae = AutoencoderKLLTXVideo.from_single_file(vae_url,token=HF_TOKEN)
-#vaeX = AutoencoderKLLTXVideo.from_pretrained("Lightricks/LTX-Video",subfolder='vae',token=HF_TOKEN)
 
 pipe = LTXImageToVideoPipeline.from_pretrained(
-    #"Lightricks/LTX-Video",
-    'a-r-r-o-w/LTX-Video-0.9.1-diffusers',
+    "Lightricks/LTX-Video",
     token=HF_TOKEN,
-    #transformer=None,
+    transformer=None,
     text_encoder=None,
 ).to(torch.device("cuda"),torch.bfloat16)
 
-#text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
-text_encoder = T5EncoderModel.from_pretrained("a-r-r-o-w/LTX-Video-0.9.1-diffusers",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
-#transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN).to(torch.device("cuda"),torch.bfloat16)
+text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
+transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN).to(torch.device("cuda"),torch.bfloat16)
 
 @spaces.GPU(duration=80)
 def generate_video(
@@ -58,7 +51,7 @@ def generate_video(
     progress=gr.Progress(track_tqdm=True)
 ):
     pipe.text_encoder=text_encoder
-  #  pipe.transformer=transformer
+    pipe.transformer=transformer
     seed=random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
     image = Image.open(image_url).convert("RGB")