Update app.py

app.py

@@ -113,9 +113,9 @@ video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     torch_dtype=dtype
 ).to(device)
 
-#video_pipe.vae.enable_tiling()
-#video_pipe.vae.enable_slicing()
-#video_pipe.enable_model_cpu_offload()
+video_pipe.vae.enable_tiling()
+video_pipe.vae.enable_slicing()
+video_pipe.enable_model_cpu_offload()
 
 # functionality
 
@@ -181,7 +181,6 @@ def pipe_generate(img,p1,p2,time,title):
             num_images_per_prompt=1,
             num_inference_steps=image_steps,
             max_sequence_length=seq,
-            need_safetycheck=False,
             generator=torch.Generator(device).manual_seed(int(str(random.random()).split(".")[1]))
         ).images[0]