1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-24 20:59:27 +03:00
Files
DN6 9adb089d80 [CLI] Add --tensor-parallel to diffusers-cli run and fix per-rank device placement
- Shard the transformer with `TensorParallelConfig` on CUDA, TPU and Neuron, with
  remote-code transformers loaded via `--trust-remote-code`.
- Require `--seed`; reject `--device-map`, `--cpu-offload`, `--lora`,
  `--context-parallel` and modular pipelines.
- Wrap `--remote` runs in torchrun.
- Load standard pipelines without an indexed `device_map` (e.g. `cuda:<LOCAL_RANK>`)
  and place components after loading, fixing `--context-parallel` on ranks > 0.
2026-10-21 13:03:11 +05:30
..