Update app.py

app.py

@@ -76,6 +76,8 @@ def load_t2v_pipeline():
             num_frames=MAX_FRAMES_MODEL,
         )
         t2v_pipe_cache[0].enable_model_cpu_offload()  # Enable CPU offload for memory optimization
+        print(f"Pipeline memory usage: {torch.cuda.max_memory_reserved() / 1024**3:.3f} GB")
+        
         clear_memory()
     return t2v_pipe_cache[0]
 
@@ -181,7 +183,7 @@ def generate_video(
 
     if mode == "Text-to-Video":
         unload_i2v_pipeline()  # Unload I2V to free memory
-        
+        print(t2v_pipe.hf_device_map)
         output_frames_list = t2v_pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,