fix: resolve unexpected 'out of memory' issue in multi-GPU setup (#2373)

BienBoy

2025-02-01 c1e365fea09aafda387cac12fdff43d28c598979

1
2
3
4
5
6
repos:
  - repo: https://github.com/psf/black
    rev: 24.4.0
    hooks:
      - id: black
        args: ['--line-length=100']  # 示例参数，black默认使用4个空格缩进