diffsynth-engine 0.4.3.dev4__tar.gz → 0.4.3.dev5__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.dev4 → diffsynth_engine-0.4.3.dev5}/PKG-INFO +1 -1
  2. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/parallel.py +3 -8
  3. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine.egg-info/PKG-INFO +1 -1
  4. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/.gitignore +0 -0
  5. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/.pre-commit-config.yaml +0 -0
  6. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/LICENSE +0 -0
  7. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/MANIFEST.in +0 -0
  8. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/README.md +0 -0
  9. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/assets/dingtalk.png +0 -0
  10. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/assets/showcase.jpeg +0 -0
  11. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/__init__.py +0 -0
  12. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/__init__.py +0 -0
  13. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/__init__.py +0 -0
  14. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/base_scheduler.py +0 -0
  15. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/flow_match/__init__.py +0 -0
  16. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_beta.py +0 -0
  17. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_ddim.py +0 -0
  18. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/flow_match/recifited_flow.py +0 -0
  19. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/__init__.py +0 -0
  20. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/beta.py +0 -0
  21. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/ddim.py +0 -0
  22. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/exponential.py +0 -0
  23. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/karras.py +0 -0
  24. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/linear.py +0 -0
  25. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/sgm_uniform.py +0 -0
  26. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/__init__.py +0 -0
  27. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/flow_match/__init__.py +0 -0
  28. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/flow_match/flow_match_euler.py +0 -0
  29. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/__init__.py +0 -0
  30. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/brownian_tree.py +0 -0
  31. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/ddpm.py +0 -0
  32. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/deis.py +0 -0
  33. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m.py +0 -0
  34. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m_sde.py +0 -0
  35. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_3m_sde.py +0 -0
  36. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/epsilon.py +0 -0
  37. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler.py +0 -0
  38. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler_ancestral.py +0 -0
  39. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/components/vae.json +0 -0
  40. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/flux/flux_dit.json +0 -0
  41. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/flux/flux_text_encoder.json +0 -0
  42. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/flux/flux_vae.json +0 -0
  43. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_config.json +0 -0
  44. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_vision_config.json +0 -0
  45. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae.json +0 -0
  46. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae_keymap.json +0 -0
  47. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/sd/sd_text_encoder.json +0 -0
  48. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/sd/sd_unet.json +0 -0
  49. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/sd3/sd3_dit.json +0 -0
  50. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/sd3/sd3_text_encoder.json +0 -0
  51. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/sdxl/sdxl_text_encoder.json +0 -0
  52. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/sdxl/sdxl_unet.json +0 -0
  53. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.1-flf2v-14b.json +0 -0
  54. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.1-i2v-14b.json +0 -0
  55. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-1.3b.json +0 -0
  56. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-14b.json +0 -0
  57. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.2-i2v-a14b.json +0 -0
  58. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.2-t2v-a14b.json +0 -0
  59. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/dit/wan2.2-ti2v-5b.json +0 -0
  60. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/vae/wan-vae-keymap.json +0 -0
  61. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/vae/wan2.1-vae.json +0 -0
  62. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/models/wan/vae/wan2.2-vae.json +0 -0
  63. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/merges.txt +0 -0
  64. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/special_tokens_map.json +0 -0
  65. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/tokenizer_config.json +0 -0
  66. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/vocab.json +0 -0
  67. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/special_tokens_map.json +0 -0
  68. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/spiece.model +0 -0
  69. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer.json +0 -0
  70. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer_config.json +0 -0
  71. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/added_tokens.json +0 -0
  72. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/merges.txt +0 -0
  73. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/special_tokens_map.json +0 -0
  74. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer.json +0 -0
  75. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer_config.json +0 -0
  76. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/vocab.json +0 -0
  77. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/merges.txt +0 -0
  78. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/special_tokens_map.json +0 -0
  79. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/tokenizer_config.json +0 -0
  80. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/vocab.json +0 -0
  81. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/merges.txt +0 -0
  82. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/special_tokens_map.json +0 -0
  83. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/tokenizer_config.json +0 -0
  84. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/vocab.json +0 -0
  85. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/special_tokens_map.json +0 -0
  86. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/spiece.model +0 -0
  87. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer.json +0 -0
  88. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer_config.json +0 -0
  89. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/configs/__init__.py +0 -0
  90. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/configs/controlnet.py +0 -0
  91. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/configs/pipeline.py +0 -0
  92. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/kernels/__init__.py +0 -0
  93. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/__init__.py +0 -0
  94. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/base.py +0 -0
  95. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/__init__.py +0 -0
  96. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/attention.py +0 -0
  97. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/lora.py +0 -0
  98. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/relative_position_emb.py +0 -0
  99. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/timestep.py +0 -0
  100. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/transformer_helper.py +0 -0
  101. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/basic/unet_helper.py +0 -0
  102. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/__init__.py +0 -0
  103. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_controlnet.py +0 -0
  104. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_dit.py +0 -0
  105. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_dit_fbcache.py +0 -0
  106. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_ipadapter.py +0 -0
  107. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_redux.py +0 -0
  108. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_text_encoder.py +0 -0
  109. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/flux/flux_vae.py +0 -0
  110. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/qwen_image/__init__.py +0 -0
  111. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/qwen_image/qwen2_5_vl.py +0 -0
  112. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/qwen_image/qwen_image_dit.py +0 -0
  113. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/qwen_image/qwen_image_dit_fbcache.py +0 -0
  114. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/qwen_image/qwen_image_vae.py +0 -0
  115. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd/__init__.py +0 -0
  116. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd/sd_controlnet.py +0 -0
  117. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd/sd_text_encoder.py +0 -0
  118. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd/sd_unet.py +0 -0
  119. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd/sd_vae.py +0 -0
  120. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd3/__init__.py +0 -0
  121. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd3/sd3_dit.py +0 -0
  122. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd3/sd3_text_encoder.py +0 -0
  123. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sd3/sd3_vae.py +0 -0
  124. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sdxl/__init__.py +0 -0
  125. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sdxl/sdxl_controlnet.py +0 -0
  126. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sdxl/sdxl_text_encoder.py +0 -0
  127. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sdxl/sdxl_unet.py +0 -0
  128. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/sdxl/sdxl_vae.py +0 -0
  129. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/text_encoder/__init__.py +0 -0
  130. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/text_encoder/clip.py +0 -0
  131. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/text_encoder/siglip.py +0 -0
  132. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/text_encoder/t5.py +0 -0
  133. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/utils.py +0 -0
  134. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/vae/__init__.py +0 -0
  135. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/vae/vae.py +0 -0
  136. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/wan/__init__.py +0 -0
  137. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/wan/wan_dit.py +0 -0
  138. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/wan/wan_image_encoder.py +0 -0
  139. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/wan/wan_text_encoder.py +0 -0
  140. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/models/wan/wan_vae.py +0 -0
  141. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/__init__.py +0 -0
  142. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/base.py +0 -0
  143. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/flux_image.py +0 -0
  144. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/qwen_image.py +0 -0
  145. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/sd_image.py +0 -0
  146. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/sdxl_image.py +0 -0
  147. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/utils.py +0 -0
  148. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/pipelines/wan_video.py +0 -0
  149. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/processor/__init__.py +0 -0
  150. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/processor/canny_processor.py +0 -0
  151. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/processor/depth_processor.py +0 -0
  152. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tokenizers/__init__.py +0 -0
  153. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tokenizers/base.py +0 -0
  154. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tokenizers/clip.py +0 -0
  155. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tokenizers/qwen2.py +0 -0
  156. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tokenizers/t5.py +0 -0
  157. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tokenizers/wan.py +0 -0
  158. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tools/__init__.py +0 -0
  159. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tools/flux_inpainting_tool.py +0 -0
  160. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tools/flux_outpainting_tool.py +0 -0
  161. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tools/flux_reference_tool.py +0 -0
  162. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/tools/flux_replace_tool.py +0 -0
  163. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/__init__.py +0 -0
  164. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/cache.py +0 -0
  165. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/constants.py +0 -0
  166. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/download.py +0 -0
  167. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/env.py +0 -0
  168. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/flag.py +0 -0
  169. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/fp8_linear.py +0 -0
  170. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/gguf.py +0 -0
  171. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/image.py +0 -0
  172. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/loader.py +0 -0
  173. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/lock.py +0 -0
  174. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/logging.py +0 -0
  175. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/offload.py +0 -0
  176. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/onnx.py +0 -0
  177. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/platform.py +0 -0
  178. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/prompt.py +0 -0
  179. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/video.py +0 -0
  180. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine.egg-info/SOURCES.txt +0 -0
  181. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine.egg-info/dependency_links.txt +0 -0
  182. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine.egg-info/requires.txt +0 -0
  183. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine.egg-info/top_level.txt +0 -0
  184. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/docs/tutorial.md +0 -0
  185. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/docs/tutorial_zh.md +0 -0
  186. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/pyproject.toml +0 -0
  187. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/setup.cfg +0 -0
  188. {diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/setup.py +0 -0
{diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.4.3.dev4
3
+ Version: 0.4.3.dev5
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.dev4 → diffsynth_engine-0.4.3.dev5}/diffsynth_engine/utils/parallel.py RENAMED
@@ -331,19 +331,14 @@ class ParallelWrapper:
331
331
  master_port: int = 29500,
332
332
  device: str = "cuda",
333
333
  ):
334
- current_method = mp.get_start_method(allow_none=True)
335
- if current_method is None or current_method != "spawn":
336
- try:
337
- mp.set_start_method("spawn")
338
- except RuntimeError as e:
339
- raise RuntimeError("Failed to set start method to spawn:", e)
340
334
  super().__init__()
341
335
  self.config = module.config if isinstance(module, BasePipeline) else None
342
336
  self._module_name = module.__class__.__name__
343
337
 
344
338
  self.world_size = cfg_degree * sp_ulysses_degree * sp_ring_degree * tp_degree
345
- self.queue_in = mp.Queue()
346
- self.queue_out = mp.Queue()
339
+ spawn_ctx = mp.get_context("spawn")
340
+ self.queue_in = spawn_ctx.Queue()
341
+ self.queue_out = spawn_ctx.Queue()
347
342
  self.ctx = mp.spawn(
348
343
  _worker_loop,
349
344
  args=(
{diffsynth_engine-0.4.3.dev4 → diffsynth_engine-0.4.3.dev5}/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.dev4
3
+ Version: 0.4.3.dev5
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent