Update app.py

app.py

@@ -28,7 +28,6 @@ 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'
 
-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)
@@ -36,11 +35,12 @@ transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=
 pipe = LTXImageToVideoPipeline.from_pretrained(
     "Lightricks/LTX-Video",
     token=HF_TOKEN,
-    transformer=transformer,
+    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)
+transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN).to(torch.device("cuda"),torch.bfloat16)
 
 @spaces.GPU(duration=80)
 def generate_video(
@@ -56,6 +56,7 @@ def generate_video(
     progress=gr.Progress(track_tqdm=True)
 ):
     pipe.text_encoder=text_encoder
+    pipe.transformer=transformer
     seed=random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
     image = Image.open(image_url).convert("RGB")