diffsynth-engine 0.6.1.dev17__py3-none-any.whl → 0.6.1.dev18__py3-none-any.whl

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 (6) hide show
  1. diffsynth_engine/pipelines/flux_image.py +3 -2
  2. {diffsynth_engine-0.6.1.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/METADATA +1 -1
  3. {diffsynth_engine-0.6.1.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/RECORD +6 -6
  4. {diffsynth_engine-0.6.1.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/WHEEL +0 -0
  5. {diffsynth_engine-0.6.1.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/licenses/LICENSE +0 -0
  6. {diffsynth_engine-0.6.1.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/top_level.txt +0 -0
diffsynth_engine/pipelines/flux_image.py CHANGED
@@ -983,8 +983,9 @@ class FluxImagePipeline(BasePipeline):
983
983
  elif self.ip_adapter is not None:
984
984
  image_emb = self.ip_adapter.encode_image(ref_image)
985
985
  elif self.redux is not None:
986
- image_prompt_embeds = self.redux(ref_image)
987
- positive_prompt_emb = torch.cat([positive_prompt_emb, image_prompt_embeds], dim=1)
986
+ ref_prompt_embeds = self.redux(ref_image)
987
+ flattened_ref_emb = ref_prompt_embeds.view(1, -1, ref_prompt_embeds.size(-1))
988
+ positive_prompt_emb = torch.cat([positive_prompt_emb, flattened_ref_emb], dim=1)
988
989
 
989
990
  # Extra input
990
991
  image_ids, text_ids, guidance = self.prepare_extra_input(
{diffsynth_engine-0.6.1.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.6.1.dev17
3
+ Version: 0.6.1.dev18
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.dev17.dist-info → diffsynth_engine-0.6.1.dev18.dist-info}/RECORD RENAMED
@@ -140,7 +140,7 @@ diffsynth_engine/models/wan/wan_text_encoder.py,sha256=OERlmwOqthAFPNnnT2sXJ4Ojy
140
140
  diffsynth_engine/models/wan/wan_vae.py,sha256=dC7MoUFeXRL7SIY0LG1OOUiZW-pp9IbXCghutMxpXr4,38889
141
141
  diffsynth_engine/pipelines/__init__.py,sha256=jh-4LSJ0vqlXiT8BgFgRIQxuAr2atEPyHrxXWj-Ud1U,604
142
142
  diffsynth_engine/pipelines/base.py,sha256=B6Md10eeAK4itILjx3biRCFwYk2usgSv7v2V9vd4fjA,14842
143
- diffsynth_engine/pipelines/flux_image.py,sha256=xoCZA2IbxARjETcni3AL6B5KcNb-RqABQA1FULi5msw,50833
143
+ diffsynth_engine/pipelines/flux_image.py,sha256=Dpy8AkwywuLAhvJ6cjg5TgzhSUgFQtv6p2JTTkzUHbo,50919
144
144
  diffsynth_engine/pipelines/hunyuan3d_shape.py,sha256=TNV0Wr09Dj2bzzlpua9WioCClOj3YiLfE6utI9aWL8A,8164
145
145
  diffsynth_engine/pipelines/qwen_image.py,sha256=o16Dffrhwel5SzCmBTE4-XnIeoq6fm3YwdPWbdsk2as,33257
146
146
  diffsynth_engine/pipelines/sd_image.py,sha256=nr-Nhsnomq8CsUqhTM3i2l2zG01YjwXdfRXgr_bC3F0,17891
@@ -185,8 +185,8 @@ diffsynth_engine/utils/video.py,sha256=8FCaeqIdUsWMgWI_6SO9SPynsToGcLCQAVYFTc4CD
185
185
  diffsynth_engine/utils/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
186
186
  diffsynth_engine/utils/memory/linear_regression.py,sha256=oW_EQEw13oPoyUrxiL8A7Ksa5AuJ2ynI2qhCbfAuZbg,3930
187
187
  diffsynth_engine/utils/memory/memory_predcit_model.py,sha256=EXprSl_zlVjgfMWNXP-iw83Ot3hyMcgYaRPv-dvyL84,3943
188
- diffsynth_engine-0.6.1.dev17.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
189
- diffsynth_engine-0.6.1.dev17.dist-info/METADATA,sha256=PypFK6NGQju_6lQq1kfFjGFsUnwft8v8MMbVjL8d_mg,1164
190
- diffsynth_engine-0.6.1.dev17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
191
- diffsynth_engine-0.6.1.dev17.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
192
- diffsynth_engine-0.6.1.dev17.dist-info/RECORD,,
188
+ diffsynth_engine-0.6.1.dev18.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
189
+ diffsynth_engine-0.6.1.dev18.dist-info/METADATA,sha256=c8oHG8aD0mbUn6T3jM-resfjrrej4jZYHb8KfZq-qYc,1164
190
+ diffsynth_engine-0.6.1.dev18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
191
+ diffsynth_engine-0.6.1.dev18.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
192
+ diffsynth_engine-0.6.1.dev18.dist-info/RECORD,,