diff --git a/inference/gradio_composite_demo/app.py b/inference/gradio_composite_demo/app.py
index d0d0d57..b180e01 100644
--- a/inference/gradio_composite_demo/app.py
+++ b/inference/gradio_composite_demo/app.py
@@ -65,10 +65,10 @@ pipe_image = CogVideoXImageToVideoPipeline.from_pretrained(
 ).to(device)
 
 
-# pipe.transformer.to(memory_format=torch.channels_last)
-# pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True)
-# pipe_image.transformer.to(memory_format=torch.channels_last)
-# pipe_image.transformer = torch.compile(pipe_image.transformer, mode="max-autotune", fullgraph=True)
+pipe.transformer.to(memory_format=torch.channels_last)
+pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True)
+pipe_image.transformer.to(memory_format=torch.channels_last)
+pipe_image.transformer = torch.compile(pipe_image.transformer, mode="max-autotune", fullgraph=True)
 
 os.makedirs("./output", exist_ok=True)
 os.makedirs("./gradio_tmp", exist_ok=True)
diff --git a/requirements.txt b/requirements.txt
index ceecfe2..02c0987 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,13 +2,13 @@ diffusers>=0.30.3
 accelerate>=0.34.2
 transformers>=4.44.2
 numpy==1.26.0
-torch==2.4.0
-torchvision==0.19.0
-sentencepiece==0.2.0
+torch>=2.4.0
+torchvision>=0.19.0
+sentencepiece>=0.2.0
 SwissArmyTransformer>=0.4.12
 gradio>=4.44.0
 imageio>=2.35.1
 imageio-ffmpeg>=0.5.1
 openai>=1.45.0
-moviepy==1.0.3
+moviepy>=1.0.3
 pillow==9.5.0
\ No newline at end of file