transformers
Tensor Parallelism and `mps` device
#44506
Merged

Commits
  • fix: fail when using `mps` with TP
    michaelbenayoun committed 130 days ago
  • Merge branch 'main' into mps_tp
    3outeille committed 127 days ago
Loading