diffsynth-engine 0.6.1.dev14__py3-none-any.whl → 0.6.1.dev25__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (45) hide show
  1. diffsynth_engine/__init__.py +6 -2
  2. diffsynth_engine/conf/models/wan/dit/wan_dit_keymap.json +41 -0
  3. diffsynth_engine/configs/__init__.py +10 -6
  4. diffsynth_engine/configs/pipeline.py +17 -10
  5. diffsynth_engine/models/base.py +1 -1
  6. diffsynth_engine/models/basic/attention.py +59 -20
  7. diffsynth_engine/models/basic/transformer_helper.py +36 -2
  8. diffsynth_engine/models/basic/video_sparse_attention.py +238 -0
  9. diffsynth_engine/models/flux/flux_controlnet.py +7 -19
  10. diffsynth_engine/models/flux/flux_dit.py +27 -38
  11. diffsynth_engine/models/flux/flux_dit_fbcache.py +9 -7
  12. diffsynth_engine/models/flux/flux_ipadapter.py +5 -5
  13. diffsynth_engine/models/qwen_image/qwen2_5_vl.py +5 -0
  14. diffsynth_engine/models/qwen_image/qwen_image_dit.py +28 -34
  15. diffsynth_engine/models/qwen_image/qwen_image_dit_fbcache.py +14 -6
  16. diffsynth_engine/models/wan/wan_audio_encoder.py +0 -1
  17. diffsynth_engine/models/wan/wan_dit.py +64 -27
  18. diffsynth_engine/pipelines/base.py +36 -4
  19. diffsynth_engine/pipelines/flux_image.py +19 -17
  20. diffsynth_engine/pipelines/qwen_image.py +45 -36
  21. diffsynth_engine/pipelines/sdxl_image.py +1 -1
  22. diffsynth_engine/pipelines/utils.py +52 -0
  23. diffsynth_engine/pipelines/wan_s2v.py +4 -9
  24. diffsynth_engine/pipelines/wan_video.py +43 -19
  25. diffsynth_engine/tokenizers/base.py +6 -0
  26. diffsynth_engine/tokenizers/qwen2.py +12 -4
  27. diffsynth_engine/utils/constants.py +13 -12
  28. diffsynth_engine/utils/flag.py +6 -0
  29. diffsynth_engine/utils/parallel.py +62 -29
  30. {diffsynth_engine-0.6.1.dev14.dist-info → diffsynth_engine-0.6.1.dev25.dist-info}/METADATA +1 -1
  31. {diffsynth_engine-0.6.1.dev14.dist-info → diffsynth_engine-0.6.1.dev25.dist-info}/RECORD +45 -43
  32. /diffsynth_engine/conf/models/wan/dit/{wan2.1-flf2v-14b.json → wan2.1_flf2v_14b.json} +0 -0
  33. /diffsynth_engine/conf/models/wan/dit/{wan2.1-i2v-14b.json → wan2.1_i2v_14b.json} +0 -0
  34. /diffsynth_engine/conf/models/wan/dit/{wan2.1-t2v-1.3b.json → wan2.1_t2v_1.3b.json} +0 -0
  35. /diffsynth_engine/conf/models/wan/dit/{wan2.1-t2v-14b.json → wan2.1_t2v_14b.json} +0 -0
  36. /diffsynth_engine/conf/models/wan/dit/{wan2.2-i2v-a14b.json → wan2.2_i2v_a14b.json} +0 -0
  37. /diffsynth_engine/conf/models/wan/dit/{wan2.2-s2v-14b.json → wan2.2_s2v_14b.json} +0 -0
  38. /diffsynth_engine/conf/models/wan/dit/{wan2.2-t2v-a14b.json → wan2.2_t2v_a14b.json} +0 -0
  39. /diffsynth_engine/conf/models/wan/dit/{wan2.2-ti2v-5b.json → wan2.2_ti2v_5b.json} +0 -0
  40. /diffsynth_engine/conf/models/wan/vae/{wan2.1-vae.json → wan2.1_vae.json} +0 -0
  41. /diffsynth_engine/conf/models/wan/vae/{wan2.2-vae.json → wan2.2_vae.json} +0 -0
  42. /diffsynth_engine/conf/models/wan/vae/{wan-vae-keymap.json → wan_vae_keymap.json} +0 -0
  43. {diffsynth_engine-0.6.1.dev14.dist-info → diffsynth_engine-0.6.1.dev25.dist-info}/WHEEL +0 -0
  44. {diffsynth_engine-0.6.1.dev14.dist-info → diffsynth_engine-0.6.1.dev25.dist-info}/licenses/LICENSE +0 -0
  45. {diffsynth_engine-0.6.1.dev14.dist-info → diffsynth_engine-0.6.1.dev25.dist-info}/top_level.txt +0 -0