Update app.py

app.py

@@ -85,12 +85,12 @@ STYLE_ENHANCERS = {
     ]
 }
 
-# 视频参数配置
+# 视频参数配置 - 优化帧率和时长
 VIDEO_CONFIG = {
-    "default_duration": 4.0,  # 秒
-    "max_duration": 8.0,
-    "default_fps": 24,
-    "max_fps": 30,
+    "default_duration": 2.0,  # 降低到2秒，更稳定
+    "max_duration": 4.0,      # 最大4秒
+    "default_fps": 12,        # 降低到12fps，减少GPU占用和动画速度
+    "max_fps": 24,            # 最大24fps
     "default_width": 512,
     "default_height": 512,
     "max_resolution": 768
@@ -162,18 +162,20 @@ def initialize_model():
         
         pipeline = pipeline.to(device)
         
-        # GPU优化
+        # GPU优化 - CogVideoX专用
         if torch.cuda.is_available():
             try:
-                pipeline.enable_vae_slicing()
-                pipeline.enable_attention_slicing()
+                # CogVideoX特有的优化方法
+                if hasattr(pipeline, 'enable_vae_tiling'):
+                    pipeline.enable_vae_tiling()
+                if hasattr(pipeline, 'enable_model_cpu_offload'):
+                    pipeline.enable_model_cpu_offload()
+                # 通用内存优化
                 try:
                     pipeline.enable_xformers_memory_efficient_attention()
                 except:
                     pass
-                # 视频生成特有的内存优化
-                if hasattr(pipeline, 'enable_sequential_cpu_offload'):
-                    pipeline.enable_sequential_cpu_offload()
+                print("✅ CogVideoX memory optimizations applied")
             except Exception as mem_error:
                 print(f"⚠️ Memory optimization warning: {mem_error}")