Update app.py

app.py

@@ -1,7 +1,7 @@
 import spaces
 import torch
 import os
-from diffusers import AutoencoderKLLTXVideo, LTXImageToVideoPipeline, LTXVideoTransformer3DModel
+from diffusers import AutoencoderKLLTXVideo, LTXImageToVideoPipeline, LTXVideoTransformer3DModel, T5EncoderModel
 from diffusers.utils import export_to_video, load_image #, PIL_INTERPOLATION
 
 import gradio as gr
@@ -16,7 +16,7 @@ torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
 torch.backends.cudnn.allow_tf32 = False
 torch.backends.cudnn.deterministic = False
 torch.backends.cudnn.benchmark = False
-torch.backends.cuda.preferred_blas_library="cublas"
+#torch.backends.cuda.preferred_blas_library="cublas"
 #torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
 os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
@@ -30,8 +30,17 @@ single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9
 transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN)
 
 #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",token=HF_TOKEN, transformer=transformer).to(torch.device("cuda"),torch.bfloat16)
+pipe = LTXImageToVideoPipeline.from_pretrained(
+    "Lightricks/LTX-Video",
+    token=HF_TOKEN,
+    transformer=transformer,
+    text_encoder=None,
+    token=True
+).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)
 
 @spaces.GPU(duration=80)
 def generate_video(
@@ -46,6 +55,7 @@ def generate_video(
     fps,
     progress=gr.Progress(track_tqdm=True)
 ):
+    pipe.text_encoder=text_encoder
     seed=random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
     image = Image.open(image_url).convert("RGB")