KawasakiAkasei/Hunyuan3D_2.1_Low_VRAM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 为 api_server 加上 enable_flashvdm

Browse Source
This commit is contained in:
WncFht
2025-07-13 11:48:13 +08:00
parent 00fa3ac012
commit a6509b95fb
2 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api_server.py
View File

@@ -199,7 +199,10 @@ if __name__ == "__main__":
parser.add_argument("--model_path", type=str, default='tencent/Hunyuan3D-2.1')
parser.add_argument("--subfolder", type=str, default='hunyuan3d-dit-v2-1')
parser.add_argument("--device", type=str, default="cuda")
parser.add_argument('--mc_algo', type=str, default='mc')
parser.add_argument("--limit-model-concurrency", type=int, default=5)
parser.add_argument('--enable_flashvdm', action='store_true')
parser.add_argument('--compile', action='store_true')
parser.add_argument('--low_vram_mode', action='store_true')
parser.add_argument('--cache-path', type=str, default='./gradio_cache')
args = parser.parse_args()
@@ -219,6 +222,9 @@ if __name__ == "__main__":
low_vram_mode=args.low_vram_mode,
worker_id=worker_id,
model_semaphore=model_semaphore,
save_dir=SAVE_DIR
save_dir=SAVE_DIR,
mc_algo=args.mc_algo,
enable_flashvdm=args.enable_flashvdm,
compile=args.compile
)
uvicorn.run(app, host=args.host, port=args.port, log_level="info")