diffsynth-engine 0.7.1.dev5__py3-none-any.whl → 0.7.1.dev6__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-0.7.1.dev5.dist-info → diffsynth_engine-0.7.1.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.7.1.dev5
+Version: 0.7.1.dev6
 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.dev6.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,7 +158,7 @@ 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
@@ -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.dev6.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
+diffsynth_engine-0.7.1.dev6.dist-info/METADATA,sha256=rriiDgtLmHVUmFGOPEP7kuSGi6wOYR8gggOZ9EGdSG8,1163
+diffsynth_engine-0.7.1.dev6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+diffsynth_engine-0.7.1.dev6.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
+diffsynth_engine-0.7.1.dev6.dist-info/RECORD,,