diffsynth-engine 0.4.3.dev5__tar.gz → 0.4.3.dev6__tar.gz

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 (188) hide show
  1. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/PKG-INFO +1 -1
  2. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/wan_video.py +13 -1
  3. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine.egg-info/PKG-INFO +1 -1
  4. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/.gitignore +0 -0
  5. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/.pre-commit-config.yaml +0 -0
  6. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/LICENSE +0 -0
  7. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/MANIFEST.in +0 -0
  8. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/README.md +0 -0
  9. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/assets/dingtalk.png +0 -0
  10. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/assets/showcase.jpeg +0 -0
  11. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/__init__.py +0 -0
  12. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/__init__.py +0 -0
  13. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/__init__.py +0 -0
  14. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/base_scheduler.py +0 -0
  15. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/flow_match/__init__.py +0 -0
  16. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_beta.py +0 -0
  17. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_ddim.py +0 -0
  18. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/flow_match/recifited_flow.py +0 -0
  19. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/__init__.py +0 -0
  20. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/beta.py +0 -0
  21. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/ddim.py +0 -0
  22. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/exponential.py +0 -0
  23. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/karras.py +0 -0
  24. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/linear.py +0 -0
  25. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/sgm_uniform.py +0 -0
  26. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/__init__.py +0 -0
  27. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/flow_match/__init__.py +0 -0
  28. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/flow_match/flow_match_euler.py +0 -0
  29. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/__init__.py +0 -0
  30. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/brownian_tree.py +0 -0
  31. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/ddpm.py +0 -0
  32. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/deis.py +0 -0
  33. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m.py +0 -0
  34. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m_sde.py +0 -0
  35. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_3m_sde.py +0 -0
  36. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/epsilon.py +0 -0
  37. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler.py +0 -0
  38. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler_ancestral.py +0 -0
  39. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/components/vae.json +0 -0
  40. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/flux/flux_dit.json +0 -0
  41. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/flux/flux_text_encoder.json +0 -0
  42. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/flux/flux_vae.json +0 -0
  43. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_config.json +0 -0
  44. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_vision_config.json +0 -0
  45. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae.json +0 -0
  46. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae_keymap.json +0 -0
  47. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/sd/sd_text_encoder.json +0 -0
  48. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/sd/sd_unet.json +0 -0
  49. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/sd3/sd3_dit.json +0 -0
  50. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/sd3/sd3_text_encoder.json +0 -0
  51. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/sdxl/sdxl_text_encoder.json +0 -0
  52. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/sdxl/sdxl_unet.json +0 -0
  53. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.1-flf2v-14b.json +0 -0
  54. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.1-i2v-14b.json +0 -0
  55. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-1.3b.json +0 -0
  56. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-14b.json +0 -0
  57. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.2-i2v-a14b.json +0 -0
  58. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.2-t2v-a14b.json +0 -0
  59. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/dit/wan2.2-ti2v-5b.json +0 -0
  60. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/vae/wan-vae-keymap.json +0 -0
  61. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/vae/wan2.1-vae.json +0 -0
  62. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/models/wan/vae/wan2.2-vae.json +0 -0
  63. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/merges.txt +0 -0
  64. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/special_tokens_map.json +0 -0
  65. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/tokenizer_config.json +0 -0
  66. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/vocab.json +0 -0
  67. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/special_tokens_map.json +0 -0
  68. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/spiece.model +0 -0
  69. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer.json +0 -0
  70. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer_config.json +0 -0
  71. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/added_tokens.json +0 -0
  72. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/merges.txt +0 -0
  73. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/special_tokens_map.json +0 -0
  74. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer.json +0 -0
  75. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer_config.json +0 -0
  76. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/vocab.json +0 -0
  77. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/merges.txt +0 -0
  78. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/special_tokens_map.json +0 -0
  79. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/tokenizer_config.json +0 -0
  80. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/vocab.json +0 -0
  81. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/merges.txt +0 -0
  82. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/special_tokens_map.json +0 -0
  83. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/tokenizer_config.json +0 -0
  84. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/vocab.json +0 -0
  85. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/special_tokens_map.json +0 -0
  86. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/spiece.model +0 -0
  87. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer.json +0 -0
  88. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer_config.json +0 -0
  89. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/configs/__init__.py +0 -0
  90. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/configs/controlnet.py +0 -0
  91. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/configs/pipeline.py +0 -0
  92. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/kernels/__init__.py +0 -0
  93. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/__init__.py +0 -0
  94. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/base.py +0 -0
  95. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/__init__.py +0 -0
  96. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/attention.py +0 -0
  97. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/lora.py +0 -0
  98. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/relative_position_emb.py +0 -0
  99. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/timestep.py +0 -0
  100. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/transformer_helper.py +0 -0
  101. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/basic/unet_helper.py +0 -0
  102. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/__init__.py +0 -0
  103. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_controlnet.py +0 -0
  104. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_dit.py +0 -0
  105. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_dit_fbcache.py +0 -0
  106. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_ipadapter.py +0 -0
  107. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_redux.py +0 -0
  108. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_text_encoder.py +0 -0
  109. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/flux/flux_vae.py +0 -0
  110. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/qwen_image/__init__.py +0 -0
  111. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/qwen_image/qwen2_5_vl.py +0 -0
  112. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/qwen_image/qwen_image_dit.py +0 -0
  113. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/qwen_image/qwen_image_dit_fbcache.py +0 -0
  114. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/qwen_image/qwen_image_vae.py +0 -0
  115. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd/__init__.py +0 -0
  116. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd/sd_controlnet.py +0 -0
  117. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd/sd_text_encoder.py +0 -0
  118. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd/sd_unet.py +0 -0
  119. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd/sd_vae.py +0 -0
  120. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd3/__init__.py +0 -0
  121. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd3/sd3_dit.py +0 -0
  122. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd3/sd3_text_encoder.py +0 -0
  123. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sd3/sd3_vae.py +0 -0
  124. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sdxl/__init__.py +0 -0
  125. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sdxl/sdxl_controlnet.py +0 -0
  126. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sdxl/sdxl_text_encoder.py +0 -0
  127. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sdxl/sdxl_unet.py +0 -0
  128. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/sdxl/sdxl_vae.py +0 -0
  129. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/text_encoder/__init__.py +0 -0
  130. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/text_encoder/clip.py +0 -0
  131. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/text_encoder/siglip.py +0 -0
  132. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/text_encoder/t5.py +0 -0
  133. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/utils.py +0 -0
  134. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/vae/__init__.py +0 -0
  135. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/vae/vae.py +0 -0
  136. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/wan/__init__.py +0 -0
  137. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/wan/wan_dit.py +0 -0
  138. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/wan/wan_image_encoder.py +0 -0
  139. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/wan/wan_text_encoder.py +0 -0
  140. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/models/wan/wan_vae.py +0 -0
  141. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/__init__.py +0 -0
  142. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/base.py +0 -0
  143. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/flux_image.py +0 -0
  144. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/qwen_image.py +0 -0
  145. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/sd_image.py +0 -0
  146. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/sdxl_image.py +0 -0
  147. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/utils.py +0 -0
  148. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/processor/__init__.py +0 -0
  149. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/processor/canny_processor.py +0 -0
  150. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/processor/depth_processor.py +0 -0
  151. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tokenizers/__init__.py +0 -0
  152. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tokenizers/base.py +0 -0
  153. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tokenizers/clip.py +0 -0
  154. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tokenizers/qwen2.py +0 -0
  155. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tokenizers/t5.py +0 -0
  156. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tokenizers/wan.py +0 -0
  157. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tools/__init__.py +0 -0
  158. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tools/flux_inpainting_tool.py +0 -0
  159. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tools/flux_outpainting_tool.py +0 -0
  160. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tools/flux_reference_tool.py +0 -0
  161. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/tools/flux_replace_tool.py +0 -0
  162. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/__init__.py +0 -0
  163. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/cache.py +0 -0
  164. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/constants.py +0 -0
  165. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/download.py +0 -0
  166. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/env.py +0 -0
  167. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/flag.py +0 -0
  168. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/fp8_linear.py +0 -0
  169. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/gguf.py +0 -0
  170. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/image.py +0 -0
  171. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/loader.py +0 -0
  172. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/lock.py +0 -0
  173. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/logging.py +0 -0
  174. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/offload.py +0 -0
  175. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/onnx.py +0 -0
  176. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/parallel.py +0 -0
  177. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/platform.py +0 -0
  178. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/prompt.py +0 -0
  179. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/utils/video.py +0 -0
  180. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine.egg-info/SOURCES.txt +0 -0
  181. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine.egg-info/dependency_links.txt +0 -0
  182. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine.egg-info/requires.txt +0 -0
  183. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine.egg-info/top_level.txt +0 -0
  184. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/docs/tutorial.md +0 -0
  185. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/docs/tutorial_zh.md +0 -0
  186. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/pyproject.toml +0 -0
  187. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/setup.cfg +0 -0
  188. {diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/setup.py +0 -0
{diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.4.3.dev5
3
+ Version: 0.4.3.dev6
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent
{diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine/pipelines/wan_video.py RENAMED
@@ -417,7 +417,7 @@ class WanVideoPipeline(BasePipeline):
417
417
  cfg_scale_ = cfg_scale if isinstance(cfg_scale, float) else cfg_scale[0]
418
418
 
419
419
  timestep = timestep * mask[:, :, :, ::2, ::2].flatten() # seq_len
420
- timestep = timestep.to(dtype=self.config.model_dtype, device=self.device)
420
+ timestep = timestep.to(dtype=self.dtype, device=self.device)
421
421
  # Classifier-free guidance
422
422
  noise_pred = self.predict_noise_with_cfg(
423
423
  model=model,
@@ -574,6 +574,18 @@ class WanVideoPipeline(BasePipeline):
574
574
  if config.offload_mode is not None:
575
575
  pipe.enable_cpu_offload(config.offload_mode)
576
576
 
577
+ if config.model_dtype == torch.float8_e4m3fn:
578
+ pipe.dtype = torch.bfloat16 # compute dtype
579
+ pipe.enable_fp8_autocast(
580
+ model_names=["dit"], compute_dtype=pipe.dtype, use_fp8_linear=config.use_fp8_linear
581
+ )
582
+
583
+ if config.t5_dtype == torch.float8_e4m3fn:
584
+ pipe.dtype = torch.bfloat16 # compute dtype
585
+ pipe.enable_fp8_autocast(
586
+ model_names=["text_encoder"], compute_dtype=pipe.dtype, use_fp8_linear=config.use_fp8_linear
587
+ )
588
+
577
589
  if config.parallelism > 1:
578
590
  return ParallelWrapper(
579
591
  pipe,
{diffsynth_engine-0.4.3.dev5 → diffsynth_engine-0.4.3.dev6}/diffsynth_engine.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.4.3.dev5
3
+ Version: 0.4.3.dev6
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent