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

diffsynth_engine/models/flux2/flux2_dit.py

@@ -1058,7 +1058,8 @@ class Flux2DiT(PreTrainedModel):
         dtype: torch.dtype = torch.float32,
         **kwargs,
     ) -> "Flux2DiT":
-        model = cls(device="meta", dtype=dtype, **kwargs)
+        with torch.device("meta"):
+            model = cls(device="meta", dtype=dtype, **kwargs)
         model = model.requires_grad_(False)
         model.load_state_dict(state_dict, assign=True)
         model.to(device=device, dtype=dtype, non_blocking=True)

diffsynth_engine/pipelines/flux2_klein_image.py

@@ -202,7 +202,8 @@ class Flux2KleinPipeline(BasePipeline):
         else:
             with open(FLUX2_TEXT_ENCODER_8B_CONF_PATH, "r", encoding="utf-8") as f:
                 qwen3_config = Qwen3Config(**json.load(f))
-                state_dicts.encoder.pop("lm_head.weight")
+                if "lm_head.weight" in state_dicts.encoder:
+                    state_dicts.encoder.pop("lm_head.weight")
             dit_config = {"guidance_embeds": False, "joint_attention_dim": 12288, "num_attention_heads": 32, "num_layers": 8, "num_single_layers": 24}
         text_encoder = Qwen3Model.from_state_dict(
             state_dicts.encoder, config=qwen3_config, device=init_device, dtype=config.encoder_dtype

diffsynth_engine/pipelines/qwen_image.py

@@ -695,7 +695,7 @@ class QwenImagePipeline(BasePipeline):
 
         if width is None and height is None:
             width, height = 1328, 1328
-        self.validate_image_size(height, width, minimum=64, multiple_of=16)
+        self.validate_image_size(height, width, minimum=16, multiple_of=16)
 
         if not isinstance(controlnet_params, list):
             controlnet_params = [controlnet_params]

diffsynth_engine/utils/flag.py

@@ -8,8 +8,9 @@ logger = logging.get_logger(__name__)
 
 def check_module_available(module_path: str, module_name: str = None) -> bool:
     try:
-        available = importlib.util.find_spec(module_path) is not None
-    except (ModuleNotFoundError, AttributeError, ValueError):
+        importlib.import_module(module_path)
+        available = True
+    except Exception:
         available = False
 
     if module_name:
@@ -26,7 +27,7 @@ FLASH_ATTN_4_AVAILABLE = check_module_available("flash_attn.cute.interface", "Fl
 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")
 XFORMERS_AVAILABLE = check_module_available("xformers", "XFormers")
-AITER_AVAILABLE = check_module_available("aiter", "Aiter")
+AITER_AVAILABLE = check_module_available("aiter.flash_attn_fp8_pertensor_func", "Aiter")
 
 SDPA_AVAILABLE = hasattr(torch.nn.functional, "scaled_dot_product_attention")
 if SDPA_AVAILABLE:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.7.1.dev5
+Version: 0.7.1.dev7
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent

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

@@ -109,7 +109,7 @@ diffsynth_engine/models/flux/flux_redux.py,sha256=-X-eu4NGEu-4PmK9hM-u4nRrWqvW6a
 diffsynth_engine/models/flux/flux_text_encoder.py,sha256=ruqFICZXQfZtHUasXHw5FK6uOi-IJdFjC-_GM0m5bco,3629
 diffsynth_engine/models/flux/flux_vae.py,sha256=lsiqqsl6eT_ry3vGkrqgLi5EVrFFXYm8wUkezspowD8,3705
 diffsynth_engine/models/flux2/__init__.py,sha256=Vpm-oP80mBsiOA1HPHfe0Fjrva4a0i0EB5cRzebUsMk,111
-diffsynth_engine/models/flux2/flux2_dit.py,sha256=VKcd1UQGfOTS_D4kOyr95m9x_N3mnRoKWLtvxd_2PwQ,42107
+diffsynth_engine/models/flux2/flux2_dit.py,sha256=aWl7sdfKQYEKSJJmub4Bb_LlLIQeFOlsTINtX0721j8,42146
 diffsynth_engine/models/flux2/flux2_vae.py,sha256=zQ1BQe3S2wnfTHp3t8AwRX_K0SONDZvima2w6Qd8uzE,83021
 diffsynth_engine/models/hunyuan3d/__init__.py,sha256=iWB2mu95TBhOwDseuwQnUjDS_IQxLIZLmcuEW2-bbIg,206
 diffsynth_engine/models/hunyuan3d/dino_image_encoder.py,sha256=KWhVNOlMjaL7sEA_W19OgoFBchauWAaShC-OfQK677k,14273
@@ -158,10 +158,10 @@ diffsynth_engine/models/z_image/z_image_dit.py,sha256=la_fcNRbLBuP7FS26ixS0hkcEo
 diffsynth_engine/models/z_image/z_image_dit_omni_base.py,sha256=cfdUFTwGFYRiyBhB_4ptn0lAvYuLAulF6zf0ABqlAzs,44854
 diffsynth_engine/pipelines/__init__.py,sha256=pcqwWR-K3jOMYiko82gSKjixYPTrfWCpZAa6qrPmdFg,880
 diffsynth_engine/pipelines/base.py,sha256=h6xOqT1LMFGrJYoTD68_VoHcfRX04je8KUE_y3BUZfM,17279
-diffsynth_engine/pipelines/flux2_klein_image.py,sha256=murraFbh_CTZmE1Ukpd7Gjv1naRXe3k8R5bwDVLvqV8,24844
+diffsynth_engine/pipelines/flux2_klein_image.py,sha256=ZNdr0j_o4JlHfdKT1MaRhAiPivv3iuLF6R5tXOyl1Es,24908
 diffsynth_engine/pipelines/flux_image.py,sha256=L0ggxpthLD8a5-zdPHu9z668uWBei9YzPb4PFVypDNU,50707
 diffsynth_engine/pipelines/hunyuan3d_shape.py,sha256=TNV0Wr09Dj2bzzlpua9WioCClOj3YiLfE6utI9aWL8A,8164
-diffsynth_engine/pipelines/qwen_image.py,sha256=Xc3H5LiQj2MUdi2KgFD2G2VqDwUa2ehqj4H35sr8iro,35627
+diffsynth_engine/pipelines/qwen_image.py,sha256=0GvGoZ1Ju5MPzZcszbSp8vvVKi3IIhPg8zMw6tsFjz4,35627
 diffsynth_engine/pipelines/sd_image.py,sha256=nr-Nhsnomq8CsUqhTM3i2l2zG01YjwXdfRXgr_bC3F0,17891
 diffsynth_engine/pipelines/sdxl_image.py,sha256=v7ZACGPb6EcBunL6e5E9jynSQjE7GQx8etEV-ZLP91g,21704
 diffsynth_engine/pipelines/utils.py,sha256=HZbJHErNJS1DhlwJKvZ9dY7Kh8Zdlsw3zE2e88TYGRY,2277
@@ -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=Ubm7FF0vHG197bmJGEplp4XauBlUaQVv-zr-w6VyEIM,2493
+diffsynth_engine/utils/flag.py,sha256=PSssT1agxSMMAVRZXBnilQ1q3rZwzk8lGCGfTJgtqa4,2483
 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.dev5.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
-diffsynth_engine-0.7.1.dev5.dist-info/METADATA,sha256=76gzYfIIeo_71jVybkzGLWiMpkm95ifPNZkL12gCRj8,1163
-diffsynth_engine-0.7.1.dev5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-diffsynth_engine-0.7.1.dev5.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
-diffsynth_engine-0.7.1.dev5.dist-info/RECORD,,
+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.dev5.dist-info → diffsynth_engine-0.7.1.dev7.dist-info}/WHEEL

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