diffsynth-engine 0.6.1.dev3__tar.gz → 0.6.1.dev4__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 (207) hide show
  1. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/PKG-INFO +1 -1
  2. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/hunyuan3d_shape.py +4 -0
  3. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine.egg-info/PKG-INFO +1 -1
  4. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/.gitattributes +0 -0
  5. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/.gitignore +0 -0
  6. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/.pre-commit-config.yaml +0 -0
  7. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/LICENSE +0 -0
  8. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/MANIFEST.in +0 -0
  9. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/README.md +0 -0
  10. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/assets/dingtalk.png +0 -0
  11. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/assets/showcase.jpeg +0 -0
  12. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/assets/tongyi.svg +0 -0
  13. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/__init__.py +0 -0
  14. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/__init__.py +0 -0
  15. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/__init__.py +0 -0
  16. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/base_scheduler.py +0 -0
  17. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/flow_match/__init__.py +0 -0
  18. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_beta.py +0 -0
  19. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_ddim.py +0 -0
  20. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/flow_match/recifited_flow.py +0 -0
  21. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/__init__.py +0 -0
  22. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/beta.py +0 -0
  23. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/ddim.py +0 -0
  24. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/exponential.py +0 -0
  25. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/karras.py +0 -0
  26. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/linear.py +0 -0
  27. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/sgm_uniform.py +0 -0
  28. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/__init__.py +0 -0
  29. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/flow_match/__init__.py +0 -0
  30. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/flow_match/flow_match_euler.py +0 -0
  31. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/__init__.py +0 -0
  32. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/brownian_tree.py +0 -0
  33. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/ddpm.py +0 -0
  34. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/deis.py +0 -0
  35. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m.py +0 -0
  36. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m_sde.py +0 -0
  37. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_3m_sde.py +0 -0
  38. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/epsilon.py +0 -0
  39. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler.py +0 -0
  40. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler_ancestral.py +0 -0
  41. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/components/vae.json +0 -0
  42. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/flux/flux_dit.json +0 -0
  43. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/flux/flux_text_encoder.json +0 -0
  44. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/flux/flux_vae.json +0 -0
  45. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_config.json +0 -0
  46. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_vision_config.json +0 -0
  47. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae.json +0 -0
  48. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae_keymap.json +0 -0
  49. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/sd/sd_text_encoder.json +0 -0
  50. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/sd/sd_unet.json +0 -0
  51. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/sd3/sd3_dit.json +0 -0
  52. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/sd3/sd3_text_encoder.json +0 -0
  53. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/sdxl/sdxl_text_encoder.json +0 -0
  54. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/sdxl/sdxl_unet.json +0 -0
  55. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.1-flf2v-14b.json +0 -0
  56. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.1-i2v-14b.json +0 -0
  57. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-1.3b.json +0 -0
  58. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-14b.json +0 -0
  59. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.2-i2v-a14b.json +0 -0
  60. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.2-s2v-14b.json +0 -0
  61. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.2-t2v-a14b.json +0 -0
  62. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/dit/wan2.2-ti2v-5b.json +0 -0
  63. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/vae/wan-vae-keymap.json +0 -0
  64. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/vae/wan2.1-vae.json +0 -0
  65. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/models/wan/vae/wan2.2-vae.json +0 -0
  66. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/merges.txt +0 -0
  67. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/special_tokens_map.json +0 -0
  68. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/tokenizer_config.json +0 -0
  69. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/vocab.json +0 -0
  70. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/special_tokens_map.json +0 -0
  71. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/spiece.model +0 -0
  72. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer.json +0 -0
  73. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer_config.json +0 -0
  74. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/qwen2_vl_image_processor.json +0 -0
  75. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/added_tokens.json +0 -0
  76. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/merges.txt +0 -0
  77. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/special_tokens_map.json +0 -0
  78. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer.json +0 -0
  79. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer_config.json +0 -0
  80. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/vocab.json +0 -0
  81. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/merges.txt +0 -0
  82. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/special_tokens_map.json +0 -0
  83. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/tokenizer_config.json +0 -0
  84. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/vocab.json +0 -0
  85. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/merges.txt +0 -0
  86. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/special_tokens_map.json +0 -0
  87. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/tokenizer_config.json +0 -0
  88. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/vocab.json +0 -0
  89. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/special_tokens_map.json +0 -0
  90. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/spiece.model +0 -0
  91. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer.json +0 -0
  92. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer_config.json +0 -0
  93. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/configs/__init__.py +0 -0
  94. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/configs/controlnet.py +0 -0
  95. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/configs/pipeline.py +0 -0
  96. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/kernels/__init__.py +0 -0
  97. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/__init__.py +0 -0
  98. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/base.py +0 -0
  99. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/__init__.py +0 -0
  100. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/attention.py +0 -0
  101. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/lora.py +0 -0
  102. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/relative_position_emb.py +0 -0
  103. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/timestep.py +0 -0
  104. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/transformer_helper.py +0 -0
  105. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/basic/unet_helper.py +0 -0
  106. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/__init__.py +0 -0
  107. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_controlnet.py +0 -0
  108. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_dit.py +0 -0
  109. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_dit_fbcache.py +0 -0
  110. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_ipadapter.py +0 -0
  111. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_redux.py +0 -0
  112. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_text_encoder.py +0 -0
  113. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/flux/flux_vae.py +0 -0
  114. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/__init__.py +0 -0
  115. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/dino_image_encoder.py +0 -0
  116. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/hunyuan3d_dit.py +0 -0
  117. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/hunyuan3d_vae.py +0 -0
  118. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/moe.py +0 -0
  119. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/surface_extractor.py +0 -0
  120. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/hunyuan3d/volume_decoder.py +0 -0
  121. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/qwen_image/__init__.py +0 -0
  122. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/qwen_image/qwen2_5_vl.py +0 -0
  123. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/qwen_image/qwen_image_dit.py +0 -0
  124. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/qwen_image/qwen_image_dit_fbcache.py +0 -0
  125. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/qwen_image/qwen_image_vae.py +0 -0
  126. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd/__init__.py +0 -0
  127. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd/sd_controlnet.py +0 -0
  128. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd/sd_text_encoder.py +0 -0
  129. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd/sd_unet.py +0 -0
  130. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd/sd_vae.py +0 -0
  131. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd3/__init__.py +0 -0
  132. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd3/sd3_dit.py +0 -0
  133. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd3/sd3_text_encoder.py +0 -0
  134. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sd3/sd3_vae.py +0 -0
  135. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sdxl/__init__.py +0 -0
  136. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sdxl/sdxl_controlnet.py +0 -0
  137. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sdxl/sdxl_text_encoder.py +0 -0
  138. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sdxl/sdxl_unet.py +0 -0
  139. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/sdxl/sdxl_vae.py +0 -0
  140. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/text_encoder/__init__.py +0 -0
  141. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/text_encoder/clip.py +0 -0
  142. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/text_encoder/siglip.py +0 -0
  143. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/text_encoder/t5.py +0 -0
  144. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/vae/__init__.py +0 -0
  145. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/vae/vae.py +0 -0
  146. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/__init__.py +0 -0
  147. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/wan_audio_encoder.py +0 -0
  148. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/wan_dit.py +0 -0
  149. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/wan_image_encoder.py +0 -0
  150. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/wan_s2v_dit.py +0 -0
  151. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/wan_text_encoder.py +0 -0
  152. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/models/wan/wan_vae.py +0 -0
  153. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/__init__.py +0 -0
  154. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/base.py +0 -0
  155. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/flux_image.py +0 -0
  156. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/qwen_image.py +0 -0
  157. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/sd_image.py +0 -0
  158. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/sdxl_image.py +0 -0
  159. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/utils.py +0 -0
  160. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/wan_s2v.py +0 -0
  161. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/wan_video.py +0 -0
  162. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/processor/__init__.py +0 -0
  163. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/processor/canny_processor.py +0 -0
  164. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/processor/depth_processor.py +0 -0
  165. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/__init__.py +0 -0
  166. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/base.py +0 -0
  167. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/clip.py +0 -0
  168. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/qwen2.py +0 -0
  169. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/qwen2_vl_image_processor.py +0 -0
  170. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/qwen2_vl_processor.py +0 -0
  171. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/t5.py +0 -0
  172. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tokenizers/wan.py +0 -0
  173. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tools/__init__.py +0 -0
  174. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tools/flux_inpainting_tool.py +0 -0
  175. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tools/flux_outpainting_tool.py +0 -0
  176. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tools/flux_reference_tool.py +0 -0
  177. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/tools/flux_replace_tool.py +0 -0
  178. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/__init__.py +0 -0
  179. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/cache.py +0 -0
  180. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/constants.py +0 -0
  181. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/download.py +0 -0
  182. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/env.py +0 -0
  183. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/flag.py +0 -0
  184. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/fp8_linear.py +0 -0
  185. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/gguf.py +0 -0
  186. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/image.py +0 -0
  187. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/loader.py +0 -0
  188. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/lock.py +0 -0
  189. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/logging.py +0 -0
  190. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/memory/__init__.py +0 -0
  191. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/memory/linear_regression.py +0 -0
  192. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/memory/memory_predcit_model.py +0 -0
  193. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/offload.py +0 -0
  194. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/onnx.py +0 -0
  195. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/parallel.py +0 -0
  196. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/platform.py +0 -0
  197. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/prompt.py +0 -0
  198. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/utils/video.py +0 -0
  199. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine.egg-info/SOURCES.txt +0 -0
  200. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine.egg-info/dependency_links.txt +0 -0
  201. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine.egg-info/requires.txt +0 -0
  202. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine.egg-info/top_level.txt +0 -0
  203. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/docs/tutorial.md +0 -0
  204. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/docs/tutorial_zh.md +0 -0
  205. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/pyproject.toml +0 -0
  206. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/setup.cfg +0 -0
  207. {diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/setup.py +0 -0
{diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.6.1.dev3
3
+ Version: 0.6.1.dev4
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent
{diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/diffsynth_engine/pipelines/hunyuan3d_shape.py RENAMED
@@ -1,4 +1,5 @@
1
1
  import torch
2
+ from typing import Optional, Callable
2
3
  from tqdm import tqdm
3
4
  from PIL import Image
4
5
  from diffsynth_engine.algorithm.noise_scheduler.flow_match.recifited_flow import RecifitedFlowScheduler
@@ -179,6 +180,7 @@ class Hunyuan3DShapePipeline(BasePipeline):
179
180
  num_inference_steps: int = 50,
180
181
  guidance_scale: float = 7.5,
181
182
  seed: int = 42,
183
+ progress_callback: Optional[Callable] = None, # def progress_callback(current, total, status)
182
184
  ):
183
185
  image_emb = self.encode_image(image)
184
186
 
@@ -197,4 +199,6 @@ class Hunyuan3DShapePipeline(BasePipeline):
197
199
  noise_pred, noise_pred_uncond = model_outputs.chunk(2)
198
200
  model_outputs = noise_pred_uncond + guidance_scale * (noise_pred - noise_pred_uncond)
199
201
  latents = self.sampler.step(latents, model_outputs, i)
202
+ if progress_callback is not None:
203
+ progress_callback(i, len(timesteps), "DENOISING")
200
204
  return self.decode_latents(latents)
{diffsynth_engine-0.6.1.dev3 → diffsynth_engine-0.6.1.dev4}/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.6.1.dev3
3
+ Version: 0.6.1.dev4
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent