Code

[Feat] Add implementations for USP#446

Code

Merged

feifeibear merged 28 commits intoxdit-project:mainxdit-project/xDiT:mainfrom xibosun:text_slicexibosun/xDiT:text_sliceCopy head branch name to clipboardFeb 10, 2025

+93-20

Commits

Commits on Nov 22, 2024

• split text in MM-DiT

  xibosun

  committed
  d631586

  Copy full SHA for d631586

• check the use_parallel_vae flag for CogVideo

  xibosun

  committed
  d48b78b

  Copy full SHA for d48b78b

Commits on Nov 27, 2024

• fix dimensions in all_gather

  xibosun

  committed
  d35e495

  Copy full SHA for d35e495

• optimizations on H100

  xibosun

  committed
  faafcd1

  Copy full SHA for faafcd1

Commits on Nov 28, 2024

• support optimized USP in Flux

  xibosun

  committed
  a41b7c1

  Copy full SHA for a41b7c1

• do not split text if undivisible by sp_degree

  xibosun

  committed
  55b8711

  Copy full SHA for 55b8711

• polish optimized USP

  xibosun

  committed
  726f402

  Copy full SHA for 726f402

• update diffusers versio in setup.py

  xibosun

  committed
  61b4b90

  Copy full SHA for 61b4b90

• merge to main

  xibosun

  committed
  b8b0b10

  Copy full SHA for b8b0b10

Commits on Nov 29, 2024

• fix bugs

  xibosun

  committed
  d43176d

  Copy full SHA for d43176d

• unify USP interface

  xibosun

  committed
  5d7c886

  Copy full SHA for 5d7c886

• optimized USP in CogVideo

  xibosun

  committed
  3c17dea

  Copy full SHA for 3c17dea

Commits on Dec 3, 2024

• use optimized USP in cogvideo

  xibosun

  committed
  cc1f2da

  Copy full SHA for cc1f2da

• add CogVideoX1.5-5B performance on H20 and L20

  xibosun

  committed
  73a071b

  Copy full SHA for 73a071b

• merge upstream main

  xibosun

  committed
  7a687df

  Copy full SHA for 7a687df

• rename files and update docs

  xibosun

  committed
  8d6de01

  Copy full SHA for 8d6de01

Commits on Dec 5, 2024

• Merge remote-tracking branch 'upstream/main' into text_slice

  xibosun

  committed
  c2fdea2

  Copy full SHA for c2fdea2

• decouple retime state from USP

  xibosun

  committed
  0172a69

  Copy full SHA for 0172a69

Commits on Dec 17, 2024

• add doc for adding new models

  xibosun

  committed
  d45155f

  Copy full SHA for d45155f

• Merge remote-tracking branch 'upstream/main' into text_slice

  xibosun

  committed
  e5e21c7

  Copy full SHA for e5e21c7

• fix typos

  xibosun

  committed
  3fd6809

  Copy full SHA for 3fd6809

Commits on Dec 18, 2024

• add docs for adding models

  xibosun

  committed
  a5318fb

  Copy full SHA for a5318fb

• fix docs for adding models

  xibosun

  committed
  a88cf5b

  Copy full SHA for a88cf5b

• fix docs for adding new models

  xibosun

  committed
  57db9de

  Copy full SHA for 57db9de

Commits on Dec 19, 2024

• fix docs for adding models

  xibosun

  committed
  564a483

  Copy full SHA for 564a483

• add figure to illustrate USP

  xibosun

  committed
  01a68a2

  Copy full SHA for 01a68a2

Commits on Feb 10, 2025

• Merge remote-tracking branch 'upstream/main' into text_slice

  xibosun

  committed
  b495702

  Copy full SHA for b495702

• feat: add usp implementations

  xibosun

  committed
  4bbbd1b

  Copy full SHA for 4bbbd1b