Optimize for 16GB GPU: set num_persistent_param_in_dit to 3B

app.py

@@ -74,7 +74,7 @@ def create_args(
     parser.add_argument("--num_inference_steps", type=int, default=inference_steps)
     parser.add_argument("--seed", type=int, default=seed)
     parser.add_argument("--fps", type=int, default=24)
-    parser.add_argument("--num_persistent_param_in_dit", type=int, default=7_000_000_000)
+    parser.add_argument("--num_persistent_param_in_dit", type=int, default=3_000_000_000)  # 16GB GPU优化
     
     return parser.parse_args([])
 

infer.py

@@ -114,7 +114,7 @@ def parse_args():
     parser.add_argument(
         "--num_persistent_param_in_dit",
         type=int,
-        default=7_000_000_000,
+        default=3_000_000_000,  # 16GB GPU优化
         help="DiT中持久参数数量，用于VRAM管理"
     )