Update app.py

app.py

@@ -41,12 +41,10 @@ import gc
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
 hf_hub_download(repo_id="wileewang/TransPixar", filename="cogvideox_rgba_lora.safetensors", local_dir="model_cogvideox_rgba_lora")
-# snapshot_download(repo_id="AlexWortega/RIFE", local_dir="model_rife")
 
 pipe = CogVideoXPipeline.from_pretrained("THUDM/CogVideoX-5B", torch_dtype=torch.bfloat16)
-pipe.enable_sequential_cpu_offload()
+# pipe.enable_sequential_cpu_offload()
 pipe.vae.enable_slicing()
 pipe.vae.enable_tiling()
 pipe.scheduler = CogVideoXDPMScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
@@ -76,14 +74,6 @@ os.makedirs("./gradio_tmp", exist_ok=True)
 # frame_interpolation_model = load_rife_model("model_rife")
 
 
-sys_prompt = """You are part of a team of bots that creates videos. You work with an assistant bot that will draw anything you say in square brackets.
-For example , outputting " a beautiful morning in the woods with the sun peaking through the trees " will trigger your partner bot to output an video of a forest morning , as described. You will be prompted by people looking to create detailed , amazing videos. The way to accomplish this is to take their short prompts and make them extremely detailed and descriptive.
-There are a few rules to follow:
-You will only ever output a single video description per user request.
-When modifications are requested , you should not simply make the description longer . You should refactor the entire description to integrate the suggestions.
-Other times the user will not want modifications , but instead want a new image . In this case , you should ignore your previous conversation with the user.
-Video descriptions must have the same num of words as examples below. Extra words will be ignored.
-"""
 def save_video(tensor: Union[List[np.ndarray], List[Image.Image]], fps: int = 8, prefix='rgb'):
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     video_path = f"./output/{prefix}_{timestamp}.mp4"