diffsynth-engine 0.6.1.dev6__tar.gz → 0.6.1.dev7__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.dev6 → diffsynth_engine-0.6.1.dev7}/PKG-INFO +1 -1
  2. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_dit.py +1 -1
  3. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_text_encoder.py +1 -1
  4. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_vae.py +1 -1
  5. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/qwen_image/qwen_image_vae.py +1 -1
  6. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd/sd_text_encoder.py +1 -1
  7. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd/sd_unet.py +1 -1
  8. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd3/sd3_dit.py +1 -1
  9. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd3/sd3_text_encoder.py +1 -1
  10. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sdxl/sdxl_text_encoder.py +1 -1
  11. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sdxl/sdxl_unet.py +1 -1
  12. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/vae/vae.py +1 -1
  13. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/wan_dit.py +1 -1
  14. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/wan_s2v_dit.py +1 -1
  15. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/wan_vae.py +2 -2
  16. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/flux_image.py +1 -1
  17. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/qwen_image.py +3 -3
  18. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/memory/memory_predcit_model.py +2 -2
  19. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine.egg-info/PKG-INFO +1 -1
  20. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/.gitattributes +0 -0
  21. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/.gitignore +0 -0
  22. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/.pre-commit-config.yaml +0 -0
  23. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/LICENSE +0 -0
  24. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/MANIFEST.in +0 -0
  25. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/README.md +0 -0
  26. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/assets/dingtalk.png +0 -0
  27. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/assets/showcase.jpeg +0 -0
  28. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/assets/tongyi.svg +0 -0
  29. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/__init__.py +0 -0
  30. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/__init__.py +0 -0
  31. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/__init__.py +0 -0
  32. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/base_scheduler.py +0 -0
  33. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/flow_match/__init__.py +0 -0
  34. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_beta.py +0 -0
  35. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/flow_match/flow_ddim.py +0 -0
  36. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/flow_match/recifited_flow.py +0 -0
  37. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/__init__.py +0 -0
  38. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/beta.py +0 -0
  39. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/ddim.py +0 -0
  40. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/exponential.py +0 -0
  41. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/karras.py +0 -0
  42. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/linear.py +0 -0
  43. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/noise_scheduler/stable_diffusion/sgm_uniform.py +0 -0
  44. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/__init__.py +0 -0
  45. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/flow_match/__init__.py +0 -0
  46. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/flow_match/flow_match_euler.py +0 -0
  47. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/__init__.py +0 -0
  48. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/brownian_tree.py +0 -0
  49. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/ddpm.py +0 -0
  50. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/deis.py +0 -0
  51. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m.py +0 -0
  52. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_2m_sde.py +0 -0
  53. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/dpmpp_3m_sde.py +0 -0
  54. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/epsilon.py +0 -0
  55. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler.py +0 -0
  56. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/algorithm/sampler/stable_diffusion/euler_ancestral.py +0 -0
  57. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/components/vae.json +0 -0
  58. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/flux/flux_dit.json +0 -0
  59. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/flux/flux_text_encoder.json +0 -0
  60. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/flux/flux_vae.json +0 -0
  61. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_config.json +0 -0
  62. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/qwen_image/qwen2_5_vl_vision_config.json +0 -0
  63. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae.json +0 -0
  64. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/qwen_image/qwen_image_vae_keymap.json +0 -0
  65. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/sd/sd_text_encoder.json +0 -0
  66. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/sd/sd_unet.json +0 -0
  67. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/sd3/sd3_dit.json +0 -0
  68. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/sd3/sd3_text_encoder.json +0 -0
  69. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/sdxl/sdxl_text_encoder.json +0 -0
  70. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/sdxl/sdxl_unet.json +0 -0
  71. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.1-flf2v-14b.json +0 -0
  72. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.1-i2v-14b.json +0 -0
  73. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-1.3b.json +0 -0
  74. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.1-t2v-14b.json +0 -0
  75. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.2-i2v-a14b.json +0 -0
  76. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.2-s2v-14b.json +0 -0
  77. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.2-t2v-a14b.json +0 -0
  78. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/dit/wan2.2-ti2v-5b.json +0 -0
  79. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/vae/wan-vae-keymap.json +0 -0
  80. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/vae/wan2.1-vae.json +0 -0
  81. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/models/wan/vae/wan2.2-vae.json +0 -0
  82. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/merges.txt +0 -0
  83. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/special_tokens_map.json +0 -0
  84. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/tokenizer_config.json +0 -0
  85. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_1/vocab.json +0 -0
  86. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/special_tokens_map.json +0 -0
  87. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/spiece.model +0 -0
  88. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer.json +0 -0
  89. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/flux/tokenizer_2/tokenizer_config.json +0 -0
  90. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/qwen2_vl_image_processor.json +0 -0
  91. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/added_tokens.json +0 -0
  92. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/merges.txt +0 -0
  93. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/special_tokens_map.json +0 -0
  94. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer.json +0 -0
  95. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/tokenizer_config.json +0 -0
  96. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/qwen_image/tokenizer/vocab.json +0 -0
  97. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/merges.txt +0 -0
  98. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/special_tokens_map.json +0 -0
  99. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/tokenizer_config.json +0 -0
  100. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer/vocab.json +0 -0
  101. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/merges.txt +0 -0
  102. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/special_tokens_map.json +0 -0
  103. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/tokenizer_config.json +0 -0
  104. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/sdxl/tokenizer_2/vocab.json +0 -0
  105. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/special_tokens_map.json +0 -0
  106. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/spiece.model +0 -0
  107. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer.json +0 -0
  108. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer_config.json +0 -0
  109. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/configs/__init__.py +0 -0
  110. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/configs/controlnet.py +0 -0
  111. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/configs/pipeline.py +0 -0
  112. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/kernels/__init__.py +0 -0
  113. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/__init__.py +0 -0
  114. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/base.py +0 -0
  115. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/__init__.py +0 -0
  116. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/attention.py +0 -0
  117. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/lora.py +0 -0
  118. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/relative_position_emb.py +0 -0
  119. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/timestep.py +0 -0
  120. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/transformer_helper.py +0 -0
  121. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/basic/unet_helper.py +0 -0
  122. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/__init__.py +0 -0
  123. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_controlnet.py +0 -0
  124. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_dit_fbcache.py +0 -0
  125. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_ipadapter.py +0 -0
  126. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/flux/flux_redux.py +0 -0
  127. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/__init__.py +0 -0
  128. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/dino_image_encoder.py +0 -0
  129. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/hunyuan3d_dit.py +0 -0
  130. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/hunyuan3d_vae.py +0 -0
  131. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/moe.py +0 -0
  132. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/surface_extractor.py +0 -0
  133. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/hunyuan3d/volume_decoder.py +0 -0
  134. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/qwen_image/__init__.py +0 -0
  135. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/qwen_image/qwen2_5_vl.py +0 -0
  136. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/qwen_image/qwen_image_dit.py +0 -0
  137. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/qwen_image/qwen_image_dit_fbcache.py +0 -0
  138. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd/__init__.py +0 -0
  139. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd/sd_controlnet.py +0 -0
  140. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd/sd_vae.py +0 -0
  141. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd3/__init__.py +0 -0
  142. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sd3/sd3_vae.py +0 -0
  143. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sdxl/__init__.py +0 -0
  144. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sdxl/sdxl_controlnet.py +0 -0
  145. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/sdxl/sdxl_vae.py +0 -0
  146. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/text_encoder/__init__.py +0 -0
  147. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/text_encoder/clip.py +0 -0
  148. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/text_encoder/siglip.py +0 -0
  149. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/text_encoder/t5.py +0 -0
  150. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/vae/__init__.py +0 -0
  151. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/__init__.py +0 -0
  152. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/wan_audio_encoder.py +0 -0
  153. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/wan_image_encoder.py +0 -0
  154. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/models/wan/wan_text_encoder.py +0 -0
  155. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/__init__.py +0 -0
  156. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/base.py +0 -0
  157. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/hunyuan3d_shape.py +0 -0
  158. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/sd_image.py +0 -0
  159. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/sdxl_image.py +0 -0
  160. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/utils.py +0 -0
  161. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/wan_s2v.py +0 -0
  162. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/pipelines/wan_video.py +0 -0
  163. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/processor/__init__.py +0 -0
  164. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/processor/canny_processor.py +0 -0
  165. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/processor/depth_processor.py +0 -0
  166. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/__init__.py +0 -0
  167. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/base.py +0 -0
  168. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/clip.py +0 -0
  169. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/qwen2.py +0 -0
  170. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/qwen2_vl_image_processor.py +0 -0
  171. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/qwen2_vl_processor.py +0 -0
  172. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/t5.py +0 -0
  173. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tokenizers/wan.py +0 -0
  174. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tools/__init__.py +0 -0
  175. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tools/flux_inpainting_tool.py +0 -0
  176. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tools/flux_outpainting_tool.py +0 -0
  177. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tools/flux_reference_tool.py +0 -0
  178. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/tools/flux_replace_tool.py +0 -0
  179. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/__init__.py +0 -0
  180. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/cache.py +0 -0
  181. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/constants.py +0 -0
  182. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/download.py +0 -0
  183. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/env.py +0 -0
  184. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/flag.py +0 -0
  185. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/fp8_linear.py +0 -0
  186. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/gguf.py +0 -0
  187. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/image.py +0 -0
  188. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/loader.py +0 -0
  189. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/lock.py +0 -0
  190. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/logging.py +0 -0
  191. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/memory/__init__.py +0 -0
  192. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/memory/linear_regression.py +0 -0
  193. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/offload.py +0 -0
  194. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/onnx.py +0 -0
  195. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/parallel.py +0 -0
  196. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/platform.py +0 -0
  197. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/prompt.py +0 -0
  198. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine/utils/video.py +0 -0
  199. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine.egg-info/SOURCES.txt +0 -0
  200. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine.egg-info/dependency_links.txt +0 -0
  201. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine.egg-info/requires.txt +0 -0
  202. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/diffsynth_engine.egg-info/top_level.txt +0 -0
  203. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/docs/tutorial.md +0 -0
  204. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/docs/tutorial_zh.md +0 -0
  205. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/pyproject.toml +0 -0
  206. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/setup.cfg +0 -0
  207. {diffsynth_engine-0.6.1.dev6 → diffsynth_engine-0.6.1.dev7}/setup.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.6.1.dev6
3
+ Version: 0.6.1.dev7
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent
@@ -28,7 +28,7 @@ from diffsynth_engine.utils import logging
28
28
 
29
29
  logger = logging.get_logger(__name__)
30
30
 
31
- with open(FLUX_DIT_CONFIG_FILE, "r") as f:
31
+ with open(FLUX_DIT_CONFIG_FILE, "r", encoding="utf-8") as f:
32
32
  config = json.load(f)
33
33
 
34
34
 
@@ -10,7 +10,7 @@ from diffsynth_engine.utils import logging
10
10
 
11
11
  logger = logging.get_logger(__name__)
12
12
 
13
- with open(FLUX_TEXT_ENCODER_CONFIG_FILE, "r") as f:
13
+ with open(FLUX_TEXT_ENCODER_CONFIG_FILE, "r", encoding="utf-8") as f:
14
14
  config = json.load(f)
15
15
 
16
16
 
@@ -8,7 +8,7 @@ from diffsynth_engine.utils import logging
8
8
 
9
9
  logger = logging.get_logger(__name__)
10
10
 
11
- with open(FLUX_VAE_CONFIG_FILE, "r") as f:
11
+ with open(FLUX_VAE_CONFIG_FILE, "r", encoding="utf-8") as f:
12
12
  config = json.load(f)
13
13
 
14
14
 
@@ -12,7 +12,7 @@ from diffsynth_engine.utils.constants import QWEN_IMAGE_VAE_KEYMAP_FILE
12
12
 
13
13
  CACHE_T = 2
14
14
 
15
- with open(QWEN_IMAGE_VAE_KEYMAP_FILE, "r") as f:
15
+ with open(QWEN_IMAGE_VAE_KEYMAP_FILE, "r", encoding="utf-8") as f:
16
16
  config = json.load(f)
17
17
 
18
18
 
@@ -10,7 +10,7 @@ from diffsynth_engine.utils import logging
10
10
 
11
11
  logger = logging.get_logger(__name__)
12
12
 
13
- with open(SD_TEXT_ENCODER_CONFIG_FILE, "r") as f:
13
+ with open(SD_TEXT_ENCODER_CONFIG_FILE, "r", encoding="utf-8") as f:
14
14
  config = json.load(f)
15
15
 
16
16
 
@@ -18,7 +18,7 @@ from diffsynth_engine.utils import logging
18
18
 
19
19
  logger = logging.get_logger(__name__)
20
20
 
21
- with open(SD_UNET_CONFIG_FILE) as f:
21
+ with open(SD_UNET_CONFIG_FILE, encoding="utf-8") as f:
22
22
  config = json.load(f)
23
23
 
24
24
 
@@ -13,7 +13,7 @@ from diffsynth_engine.utils import logging
13
13
 
14
14
  logger = logging.get_logger(__name__)
15
15
 
16
- with open(SD3_DIT_CONFIG_FILE, "r") as f:
16
+ with open(SD3_DIT_CONFIG_FILE, "r", encoding="utf-8") as f:
17
17
  config = json.load(f)
18
18
 
19
19
 
@@ -11,7 +11,7 @@ from diffsynth_engine.utils import logging
11
11
 
12
12
  logger = logging.get_logger(__name__)
13
13
 
14
- with open(SD3_TEXT_ENCODER_CONFIG_FILE, "r") as f:
14
+ with open(SD3_TEXT_ENCODER_CONFIG_FILE, "r", encoding="utf-8") as f:
15
15
  config = json.load(f)
16
16
 
17
17
 
@@ -10,7 +10,7 @@ from diffsynth_engine.utils import logging
10
10
 
11
11
  logger = logging.get_logger(__name__)
12
12
 
13
- with open(SDXL_TEXT_ENCODER_CONFIG_FILE, "r") as f:
13
+ with open(SDXL_TEXT_ENCODER_CONFIG_FILE, "r", encoding="utf-8") as f:
14
14
  config = json.load(f)
15
15
 
16
16
 
@@ -18,7 +18,7 @@ from diffsynth_engine.utils import logging
18
18
 
19
19
  logger = logging.get_logger(__name__)
20
20
 
21
- with open(SDXL_UNET_CONFIG_FILE, "r") as f:
21
+ with open(SDXL_UNET_CONFIG_FILE, "r", encoding="utf-8") as f:
22
22
  config = json.load(f)
23
23
 
24
24
 
@@ -12,7 +12,7 @@ from diffsynth_engine.utils import logging
12
12
 
13
13
  logger = logging.get_logger(__name__)
14
14
 
15
- with open(VAE_CONFIG_FILE, "r") as f:
15
+ with open(VAE_CONFIG_FILE, "r", encoding="utf-8") as f:
16
16
  config = json.load(f)
17
17
 
18
18
 
@@ -398,7 +398,7 @@ class WanDiT(PreTrainedModel):
398
398
  raise ValueError(f"Unsupported model type: {model_type}")
399
399
 
400
400
  config_file = MODEL_CONFIG_FILES[model_type]
401
- with open(config_file, "r") as f:
401
+ with open(config_file, "r", encoding="utf-8") as f:
402
402
  config = json.load(f)
403
403
  return config
404
404
 
@@ -360,7 +360,7 @@ class WanS2VDiT(WanDiT):
360
360
  raise ValueError(f"Unsupported model type: {model_type}")
361
361
 
362
362
  config_file = MODEL_CONFIG_FILES[model_type]
363
- with open(config_file, "r") as f:
363
+ with open(config_file, "r", encoding="utf-8") as f:
364
364
  config = json.load(f)
365
365
  return config
366
366
 
@@ -12,7 +12,7 @@ from diffsynth_engine.utils.constants import WAN2_1_VAE_CONFIG_FILE, WAN2_2_VAE_
12
12
 
13
13
  CACHE_T = 2
14
14
 
15
- with open(WAN_VAE_KEYMAP_FILE, "r") as f:
15
+ with open(WAN_VAE_KEYMAP_FILE, "r", encoding="utf-8") as f:
16
16
  config = json.load(f)
17
17
 
18
18
 
@@ -855,7 +855,7 @@ class WanVideoVAE(PreTrainedModel):
855
855
  raise ValueError(f"Unsupported model type: {model_type}")
856
856
 
857
857
  config_file = MODEL_CONFIG_FILES[model_type]
858
- with open(config_file, "r") as f:
858
+ with open(config_file, "r", encoding="utf-8") as f:
859
859
  config = json.load(f)
860
860
  return config
861
861
 
@@ -34,7 +34,7 @@ from diffsynth_engine.utils.constants import FLUX_DIT_CONFIG_FILE
34
34
 
35
35
  logger = logging.get_logger(__name__)
36
36
 
37
- with open(FLUX_DIT_CONFIG_FILE, "r") as f:
37
+ with open(FLUX_DIT_CONFIG_FILE, "r", encoding="utf-8") as f:
38
38
  config = json.load(f)
39
39
 
40
40
 
@@ -182,9 +182,9 @@ class QwenImagePipeline(BasePipeline):
182
182
  tokenizer_config_path=QWEN_IMAGE_TOKENIZER_CONF_PATH,
183
183
  image_processor_config_path=QWEN_IMAGE_PROCESSOR_CONFIG_FILE,
184
184
  )
185
- with open(QWEN_IMAGE_VISION_CONFIG_FILE, "r") as f:
185
+ with open(QWEN_IMAGE_VISION_CONFIG_FILE, "r", encoding="utf-8") as f:
186
186
  vision_config = Qwen2_5_VLVisionConfig(**json.load(f))
187
- with open(QWEN_IMAGE_CONFIG_FILE, "r") as f:
187
+ with open(QWEN_IMAGE_CONFIG_FILE, "r", encoding="utf-8") as f:
188
188
  text_config = Qwen2_5_VLConfig(**json.load(f))
189
189
  encoder = Qwen2_5_VLForConditionalGeneration.from_state_dict(
190
190
  state_dicts.encoder,
@@ -193,7 +193,7 @@ class QwenImagePipeline(BasePipeline):
193
193
  device=init_device,
194
194
  dtype=config.encoder_dtype,
195
195
  )
196
- with open(QWEN_IMAGE_VAE_CONFIG_FILE, "r") as f:
196
+ with open(QWEN_IMAGE_VAE_CONFIG_FILE, "r", encoding="utf-8") as f:
197
197
  vae_config = json.load(f)
198
198
  vae = QwenImageVAE.from_state_dict(
199
199
  state_dicts.vae, config=vae_config, device=init_device, dtype=config.vae_dtype
@@ -87,7 +87,7 @@ class MemoryPredictModel:
87
87
  return r2
88
88
 
89
89
  def save_model(self, model_path):
90
- with open(model_path, "wb") as f:
90
+ with open(model_path, "wb", encoding="utf-8") as f:
91
91
  torch.save(
92
92
  {
93
93
  "key_inputs": self.key_inputs,
@@ -98,7 +98,7 @@ class MemoryPredictModel:
98
98
 
99
99
  @classmethod
100
100
  def from_pretrained(cls, model_path):
101
- with open(model_path, "rb") as f:
101
+ with open(model_path, "rb", encoding="utf-8") as f:
102
102
  data = torch.load(f)
103
103
  model = cls(data["key_inputs"])
104
104
  model.model = LinearRegression.deserialize(data["model"])
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.6.1.dev6
3
+ Version: 0.6.1.dev7
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent