Update app.py

app.py

@@ -144,12 +144,12 @@ except Exception as e:
 
 
 single_file_url = repo_id+"/ltx-video-2b-v0.9.1.safetensors"
-text_encoder = T5EncoderModel.from_pretrained(
+text_encoder = torch.compile(T5EncoderModel.from_pretrained(
   repo_id, subfolder="text_encoder", torch_dtype=torch.bfloat16
-)
+), mode="reduce-overhead", fullgraph=True)
 tokenizer = T5Tokenizer.from_pretrained(
   repo_id, subfolder="tokenizer", torch_dtype=torch.bfloat16
-)
+).to(memory_format=torch.channels_last)
 pipe = LTXPipeline.from_single_file(
     single_file_url, 
     text_encoder=text_encoder, 
@@ -162,11 +162,7 @@ pipe.to("cuda")
 # pipe.load_lora_weights("TODO/TODO", adapter_name="ltx-lora")
 # pipe.set_adapters(["lrx-lora"], adapter_weights=[1.0])
 
-# pipe.fuse_qkv_projections()
-
-pipe.unet.to(memory_format=torch.channels_last)
 pipe.vae.to(memory_format=torch.channels_last)
-pipe.unet = torch.compile(pipe.unet, mode="max-autotune", fullgraph=True)
 pipe.vae.decode = torch.compile(pipe.vae.decode, mode="max-autotune", fullgraph=True)
 
 @spaces.GPU(duration=120)