HunyuanDiT icon indicating copy to clipboard operation
HunyuanDiT copied to clipboard

Published 20 hours ago • verified publisher icon Tencent

caption_demo.py 支持并行打标

Open CanvaChen opened this issue 7 months ago • 0 comments

描述功能 支持多个进程对图片打标签

动机 提高显存利用率，加快打标速度

其他上下文

Used Dedicated Memory: 17036 MB (69%) 0%| | 15/6333 [01:02<7:09:55, 4.08s/it]

如上，仅6k张图就要用7小时，如果要打标一百万张图耗时将无法忍受，而且有30%的显存未利用

CanvaChen avatar Jul 24 '24 16:07 CanvaChen