diffsynth-engine 0.7.1.dev7__py3-none-any.whl → 0.7.1.dev9__py3-none-any.whl

diffsynth_engine/utils/flag.py

@@ -25,7 +25,7 @@ def check_module_available(module_path: str, module_name: str = None) -> bool:
 # 无损
 FLASH_ATTN_4_AVAILABLE = check_module_available("flash_attn.cute.interface", "Flash attention 4")
 FLASH_ATTN_3_AVAILABLE = check_module_available("flash_attn_interface", "Flash attention 3")
-FLASH_ATTN_2_AVAILABLE = check_module_available("flash_attn", "Flash attention 2")
+FLASH_ATTN_2_AVAILABLE = check_module_available("flash_attn.flash_attn_func", "Flash attention 2")
 XFORMERS_AVAILABLE = check_module_available("xformers", "XFormers")
 AITER_AVAILABLE = check_module_available("aiter.flash_attn_fp8_pertensor_func", "Aiter")
 

{diffsynth_engine-0.7.1.dev7.dist-info → diffsynth_engine-0.7.1.dev9.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.7.1.dev7
+Version: 0.7.1.dev9
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.10
 License-File: LICENSE
-Requires-Dist: torch>=2.6
+Requires-Dist: torch<=2.8,>=2.6
 Requires-Dist: torchvision
 Requires-Dist: numpy
 Requires-Dist: safetensors

{diffsynth_engine-0.7.1.dev7.dist-info → diffsynth_engine-0.7.1.dev9.dist-info}/RECORD

@@ -192,7 +192,7 @@ diffsynth_engine/utils/cache.py,sha256=Ivef22pCuhEq-4H00gSvkLS8ceVZoGis7OSitYL6g
 diffsynth_engine/utils/constants.py,sha256=Tsn3EAByfZra-nGcx0NEcP9nWTPKaDGdatosE3BuPGE,3846
 diffsynth_engine/utils/download.py,sha256=w9QQjllPfTUEY371UTREU7o_vvdMY-Q2DymDel3ZEZY,6792
 diffsynth_engine/utils/env.py,sha256=k749eYt_qKGq38GocDiXfkhp8nZrowFefNVTZ8R755I,363
-diffsynth_engine/utils/flag.py,sha256=PSssT1agxSMMAVRZXBnilQ1q3rZwzk8lGCGfTJgtqa4,2483
+diffsynth_engine/utils/flag.py,sha256=Q7byWPKegE2wq5j_2KU73swsLvmFG7JrwRxslsAabLA,2499
 diffsynth_engine/utils/fp8_linear.py,sha256=k34YFWo2dc3t8aKjHaCW9CbQMOTqXxaDHk8aw8aKif4,3857
 diffsynth_engine/utils/gguf.py,sha256=ZWvw46V4g4uVyAR_oCq-4K5nPdKVrYk3u47uXMgA9lU,14092
 diffsynth_engine/utils/image.py,sha256=jqx-UKfdc2YRBtHoL-RP2M8yce_0h2rTIJgf6mux-aU,12695
@@ -209,8 +209,8 @@ diffsynth_engine/utils/video.py,sha256=8FCaeqIdUsWMgWI_6SO9SPynsToGcLCQAVYFTc4CD
 diffsynth_engine/utils/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 diffsynth_engine/utils/memory/linear_regression.py,sha256=oW_EQEw13oPoyUrxiL8A7Ksa5AuJ2ynI2qhCbfAuZbg,3930
 diffsynth_engine/utils/memory/memory_predcit_model.py,sha256=EXprSl_zlVjgfMWNXP-iw83Ot3hyMcgYaRPv-dvyL84,3943
-diffsynth_engine-0.7.1.dev7.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
-diffsynth_engine-0.7.1.dev7.dist-info/METADATA,sha256=Tv8IEL6r7NbkQjxss_9VKtc59mXWTFyAWmWqYh5f6Nk,1163
-diffsynth_engine-0.7.1.dev7.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
-diffsynth_engine-0.7.1.dev7.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
-diffsynth_engine-0.7.1.dev7.dist-info/RECORD,,
+diffsynth_engine-0.7.1.dev9.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
+diffsynth_engine-0.7.1.dev9.dist-info/METADATA,sha256=RyzJ__ltkdqqqgrqHdWjxnavdxU3hqRDaeySDxJS3oo,1169
+diffsynth_engine-0.7.1.dev9.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+diffsynth_engine-0.7.1.dev9.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
+diffsynth_engine-0.7.1.dev9.dist-info/RECORD,,

{diffsynth_engine-0.7.1.dev7.dist-info → diffsynth_engine-0.7.1.dev9.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (82.0.0)
+Generator: setuptools (82.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any