diffsynth-engine 0.3.6.dev8__py3-none-any.whl → 0.3.6.dev10__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 (30) hide show
  1. diffsynth_engine/__init__.py +10 -8
  2. diffsynth_engine/configs/__init__.py +23 -0
  3. diffsynth_engine/configs/controlnet.py +17 -0
  4. diffsynth_engine/configs/pipeline.py +206 -0
  5. diffsynth_engine/models/basic/attention.py +43 -4
  6. diffsynth_engine/models/flux/flux_controlnet.py +8 -5
  7. diffsynth_engine/models/flux/flux_dit.py +22 -16
  8. diffsynth_engine/models/flux/flux_dit_fbcache.py +5 -5
  9. diffsynth_engine/models/flux/flux_ipadapter.py +5 -5
  10. diffsynth_engine/models/sd/sd_controlnet.py +2 -4
  11. diffsynth_engine/models/sdxl/sdxl_controlnet.py +1 -2
  12. diffsynth_engine/models/wan/wan_dit.py +15 -15
  13. diffsynth_engine/pipelines/__init__.py +5 -8
  14. diffsynth_engine/pipelines/base.py +14 -65
  15. diffsynth_engine/pipelines/flux_image.py +85 -158
  16. diffsynth_engine/pipelines/sd_image.py +30 -64
  17. diffsynth_engine/pipelines/sdxl_image.py +39 -71
  18. diffsynth_engine/pipelines/wan_video.py +66 -105
  19. diffsynth_engine/tools/flux_inpainting_tool.py +7 -3
  20. diffsynth_engine/tools/flux_outpainting_tool.py +7 -3
  21. diffsynth_engine/tools/flux_reference_tool.py +21 -5
  22. diffsynth_engine/tools/flux_replace_tool.py +15 -3
  23. diffsynth_engine/utils/fp8_linear.py +14 -5
  24. diffsynth_engine/utils/parallel.py +1 -1
  25. diffsynth_engine/utils/platform.py +9 -1
  26. {diffsynth_engine-0.3.6.dev8.dist-info → diffsynth_engine-0.3.6.dev10.dist-info}/METADATA +1 -1
  27. {diffsynth_engine-0.3.6.dev8.dist-info → diffsynth_engine-0.3.6.dev10.dist-info}/RECORD +30 -27
  28. {diffsynth_engine-0.3.6.dev8.dist-info → diffsynth_engine-0.3.6.dev10.dist-info}/WHEEL +0 -0
  29. {diffsynth_engine-0.3.6.dev8.dist-info → diffsynth_engine-0.3.6.dev10.dist-info}/licenses/LICENSE +0 -0
  30. {diffsynth_engine-0.3.6.dev8.dist-info → diffsynth_engine-0.3.6.dev10.dist-info}/top_level.txt +0 -0
@@ -1,4 +1,4 @@
1
- from diffsynth_engine import fetch_model, FluxControlNet, ControlNetParams, FluxImagePipeline
1
+ from diffsynth_engine import fetch_model, FluxPipelineConfig, FluxControlNet, ControlNetParams, FluxImagePipeline
2
2
  from typing import List, Tuple, Optional, Callable
3
3
  from PIL import Image
4
4
  import torch
@@ -12,9 +12,13 @@ class FluxInpaintingTool:
12
12
  dtype: torch.dtype = torch.bfloat16,
13
13
  offload_mode: Optional[str] = None,
14
14
  ):
15
- self.pipe = FluxImagePipeline.from_pretrained(
16
- flux_model_path, device=device, offload_mode=offload_mode, dtype=dtype
15
+ config = FluxPipelineConfig(
16
+ model_path=flux_model_path,
17
+ model_dtype=dtype,
18
+ device=device,
19
+ offload_mode=offload_mode,
17
20
  )
21
+ self.pipe = FluxImagePipeline.from_pretrained(config)
18
22
  self.controlnet = FluxControlNet.from_pretrained(
19
23
  fetch_model(
20
24
  "alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", path="diffusion_pytorch_model.safetensors"
@@ -1,4 +1,4 @@
1
- from diffsynth_engine import fetch_model, FluxControlNet, ControlNetParams, FluxImagePipeline
1
+ from diffsynth_engine import fetch_model, FluxPipelineConfig, FluxControlNet, ControlNetParams, FluxImagePipeline
2
2
  from typing import List, Tuple, Optional, Callable
3
3
  from PIL import Image
4
4
  import torch
@@ -12,9 +12,13 @@ class FluxOutpaintingTool:
12
12
  dtype: torch.dtype = torch.bfloat16,
13
13
  offload_mode: Optional[str] = None,
14
14
  ):
15
- self.pipe = FluxImagePipeline.from_pretrained(
16
- flux_model_path, device=device, offload_mode=offload_mode, dtype=dtype
15
+ config = FluxPipelineConfig(
16
+ model_path=flux_model_path,
17
+ model_dtype=dtype,
18
+ device=device,
19
+ offload_mode=offload_mode,
17
20
  )
21
+ self.pipe = FluxImagePipeline.from_pretrained(config)
18
22
  self.controlnet = FluxControlNet.from_pretrained(
19
23
  fetch_model(
20
24
  "alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", path="diffusion_pytorch_model.safetensors"
@@ -1,4 +1,11 @@
1
- from diffsynth_engine import ControlNetParams, FluxImagePipeline, FluxIPAdapter, FluxRedux, fetch_model
1
+ from diffsynth_engine import (
2
+ FluxPipelineConfig,
3
+ ControlNetParams,
4
+ FluxImagePipeline,
5
+ FluxIPAdapter,
6
+ FluxRedux,
7
+ fetch_model,
8
+ )
2
9
  from typing import List, Tuple, Optional
3
10
  from PIL import Image
4
11
  import torch
@@ -17,9 +24,14 @@ class FluxReduxRefTool:
17
24
  dtype: torch.dtype = torch.bfloat16,
18
25
  offload_mode: Optional[str] = None,
19
26
  ):
20
- self.pipe: FluxImagePipeline = FluxImagePipeline.from_pretrained(
21
- flux_model_path, load_text_encoder=load_text_encoder, device=device, offload_mode=offload_mode, dtype=dtype
27
+ config = FluxPipelineConfig(
28
+ model_path=flux_model_path,
29
+ model_dtype=dtype,
30
+ load_text_encoder=load_text_encoder,
31
+ device=device,
32
+ offload_mode=offload_mode,
22
33
  )
34
+ self.pipe: FluxImagePipeline = FluxImagePipeline.from_pretrained(config)
23
35
  redux_model_path = fetch_model("muse/flux1-redux-dev", path="flux1-redux-dev.safetensors", revision="v1")
24
36
  flux_redux = FluxRedux.from_pretrained(redux_model_path, device=device)
25
37
  self.pipe.load_redux(flux_redux)
@@ -68,9 +80,13 @@ class FluxIPAdapterRefTool:
68
80
  dtype: torch.dtype = torch.bfloat16,
69
81
  offload_mode: Optional[str] = None,
70
82
  ):
71
- self.pipe: FluxImagePipeline = FluxImagePipeline.from_pretrained(
72
- flux_model_path, device=device, offload_mode=offload_mode
83
+ config = FluxPipelineConfig(
84
+ model_path=flux_model_path,
85
+ model_dtype=dtype,
86
+ device=device,
87
+ offload_mode=offload_mode,
73
88
  )
89
+ self.pipe: FluxImagePipeline = FluxImagePipeline.from_pretrained(config)
74
90
  self.pipe.load_loras(lora_list)
75
91
  ip_adapter_path = fetch_model("muse/FLUX.1-dev-IP-Adapter", path="ip-adapter.safetensors", revision="v1")
76
92
  ip_adapter: FluxIPAdapter = FluxIPAdapter.from_pretrained(ip_adapter_path, device=device)
@@ -1,4 +1,11 @@
1
- from diffsynth_engine import ControlNetParams, FluxControlNet, FluxImagePipeline, FluxRedux, fetch_model
1
+ from diffsynth_engine import (
2
+ FluxPipelineConfig,
3
+ ControlNetParams,
4
+ FluxControlNet,
5
+ FluxImagePipeline,
6
+ FluxRedux,
7
+ fetch_model,
8
+ )
2
9
  from typing import List, Tuple, Optional, Callable
3
10
  from PIL import Image
4
11
  import torch
@@ -18,9 +25,14 @@ class FluxReplaceByControlTool:
18
25
  dtype: torch.dtype = torch.bfloat16,
19
26
  offload_mode: Optional[str] = None,
20
27
  ):
21
- self.pipe: FluxImagePipeline = FluxImagePipeline.from_pretrained(
22
- flux_model_path, load_text_encoder=load_text_encoder, device=device, offload_mode=offload_mode, dtype=dtype
28
+ config = FluxPipelineConfig(
29
+ model_path=flux_model_path,
30
+ model_dtype=dtype,
31
+ load_text_encoder=load_text_encoder,
32
+ device=device,
33
+ offload_mode=offload_mode,
23
34
  )
35
+ self.pipe: FluxImagePipeline = FluxImagePipeline.from_pretrained(config)
24
36
  redux_model_path = fetch_model("muse/flux1-redux-dev", path="flux1-redux-dev.safetensors", revision="v1")
25
37
  flux_redux = FluxRedux.from_pretrained(redux_model_path, device=device)
26
38
  self.pipe.load_redux(flux_redux)
@@ -2,6 +2,7 @@ import torch
2
2
  import torch.nn as nn
3
3
  import torch.nn.functional as F
4
4
  from contextlib import contextmanager
5
+ from diffsynth_engine.utils.platform import DTYPE_FP8
5
6
 
6
7
 
7
8
  def enable_fp8_autocast(module: nn.Module, compute_dtype: torch.dtype = torch.bfloat16, use_fp8_linear: bool = False):
@@ -51,7 +52,7 @@ def enable_fp8_linear(module: nn.Module):
51
52
  def _enable_fp8_linear(module: nn.Module):
52
53
  if isinstance(module, nn.Linear) and torch.is_floating_point(module.weight.data):
53
54
  # avoid conversion for int weights like GGUF
54
- module.weight.data = module.weight.data.to(torch.float8_e4m3fn)
55
+ module.weight.data = module.weight.data.to(DTYPE_FP8)
55
56
  for submodule in module.children():
56
57
  _enable_fp8_linear(submodule)
57
58
 
@@ -71,8 +72,16 @@ def fp8_inference(enabled=True):
71
72
  ) -> torch.Tensor:
72
73
  device = input.device
73
74
  origin_dtype = input.dtype
74
- input = input.to(torch.float8_e4m3fn)
75
- weight = weight.to(torch.float8_e4m3fn)
75
+ scale_a = 1.0
76
+ # For float8_e4m3fnuz, the maximum representable value is half of that of e4m3fn.
77
+ # To avoid overflow and ensure numerical compatibility during FP8 computation,
78
+ # we scale down the input by 2.0 in advance.
79
+ # This scaling will be compensated later during the final result scaling.
80
+ if DTYPE_FP8 == torch.float8_e4m3fnuz:
81
+ scale_a = 2.0
82
+ input = input / scale_a
83
+ input = input.to(DTYPE_FP8)
84
+ weight = weight.to(DTYPE_FP8)
76
85
 
77
86
  if len(input.shape) > 2:
78
87
  origin_shape = input.shape
@@ -80,7 +89,7 @@ def fp8_inference(enabled=True):
80
89
  result = torch._scaled_mm(
81
90
  input,
82
91
  weight.T,
83
- scale_a=torch.tensor(1.0).to(device=device),
92
+ scale_a=torch.tensor(scale_a).to(device=device),
84
93
  scale_b=torch.tensor(1.0).to(device=device),
85
94
  bias=bias,
86
95
  out_dtype=origin_dtype,
@@ -91,7 +100,7 @@ def fp8_inference(enabled=True):
91
100
  result = torch._scaled_mm(
92
101
  input,
93
102
  weight.T,
94
- scale_a=torch.tensor(1.0).to(device=device),
103
+ scale_a=torch.tensor(scale_a).to(device=device),
95
104
  scale_b=torch.tensor(1.0).to(device=device),
96
105
  bias=bias,
97
106
  out_dtype=origin_dtype,
@@ -325,7 +325,7 @@ class ParallelWrapper:
325
325
  sp_ulysses_degree: int,
326
326
  sp_ring_degree: int,
327
327
  tp_degree: int,
328
- use_fsdp: bool,
328
+ use_fsdp: bool = False,
329
329
  master_port: int = 29500,
330
330
  device: str = "cuda",
331
331
  ):
@@ -1,7 +1,15 @@
1
+ # cross-platform definitions and utilities
1
2
  import torch
2
3
  import gc
3
4
 
4
- # 存放跨平台的工具类
5
+
6
+ # data type
7
+ # AMD only supports float8_e4m3fnuz
8
+ # https://onnx.ai/onnx/technical/float8.html
9
+ if torch.version.hip and "gfx94" in torch.cuda.get_device_properties(0).gcnArchName:
10
+ DTYPE_FP8 = torch.float8_e4m3fnuz
11
+ else:
12
+ DTYPE_FP8 = torch.float8_e4m3fn
5
13
 
6
14
 
7
15
  def empty_cache():
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: diffsynth_engine
3
- Version: 0.3.6.dev8
3
+ Version: 0.3.6.dev10
4
4
  Author: MuseAI x ModelScope
5
5
  Classifier: Programming Language :: Python :: 3
6
6
  Classifier: Operating System :: OS Independent
@@ -1,4 +1,4 @@
1
- diffsynth_engine/__init__.py,sha256=eJPIItR6g1yK6Weyv5j8RnAx_wnwWC073orD3SWQvjE,1230
1
+ diffsynth_engine/__init__.py,sha256=PnsxBE7qAW_5yDsrl1S-I3UraXMOQKTHWxAfKHbwIYQ,1279
2
2
  diffsynth_engine/algorithm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  diffsynth_engine/algorithm/noise_scheduler/__init__.py,sha256=YvcwE2tCNua-OAX9GEPm0EXsINNWH4XvJMNZb-uaZMM,745
4
4
  diffsynth_engine/algorithm/noise_scheduler/base_scheduler.py,sha256=WICrLEh7b2TdZMMEN14NqiYydj7dxXT6RolXymKiMk8,188
@@ -60,27 +60,30 @@ diffsynth_engine/conf/tokenizers/wan/umt5-xxl/special_tokens_map.json,sha256=e4q
60
60
  diffsynth_engine/conf/tokenizers/wan/umt5-xxl/spiece.model,sha256=45CaZ7eAZQs1z1Kax4KtK2sm5tH4SdP7tqhykF9FJFg,4548313
61
61
  diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer.json,sha256=bhl7TT29cdoUtOslX0-pHJwfIGiyCi3iRylnyj0iYCs,16837417
62
62
  diffsynth_engine/conf/tokenizers/wan/umt5-xxl/tokenizer_config.json,sha256=7Zo6iw-qcacKMoR-BDX-A25uES1N9O23u0ipIeNE3AU,61728
63
+ diffsynth_engine/configs/__init__.py,sha256=dZ80g2GB3B2YdmoGMp9yvwK3FRJI5j8vShhB9L95j1U,460
64
+ diffsynth_engine/configs/controlnet.py,sha256=OF_cznEw-NpGTM9vP_mIApr4MAJCywWoDWkcUWCz-bs,434
65
+ diffsynth_engine/configs/pipeline.py,sha256=NPQlNz-AOpi8qFzRob0RNnOqSc8C-vCdHbstLyUugeo,7703
63
66
  diffsynth_engine/kernels/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
64
67
  diffsynth_engine/models/__init__.py,sha256=8Ze7cSE8InetgXWTNb0neVA2Q44K7WlE-h7O-02m2sY,119
65
68
  diffsynth_engine/models/base.py,sha256=sbyyGP-ENnqicr6cxjEmXRf6dWrmKjCu6k5yamuJ518,2665
66
69
  diffsynth_engine/models/utils.py,sha256=r5xLSEog1_ODaFrpqzJvAj3r23PQiEpgivzErClTZTg,1561
67
70
  diffsynth_engine/models/basic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
68
- diffsynth_engine/models/basic/attention.py,sha256=olX6-_3znlCIxEIGKX0qWWXeqwmQ4QQ-EuSr8wMtmbA,7804
71
+ diffsynth_engine/models/basic/attention.py,sha256=Hquc9H57N37hz5JbCJHpuPGX1smG403DNiOZTnlCYYA,9103
69
72
  diffsynth_engine/models/basic/lora.py,sha256=qEh44zfh7ZBblLpjmKzwzAxmTlVyY0wu9IkGsnr7Ih8,10614
70
73
  diffsynth_engine/models/basic/relative_position_emb.py,sha256=rCXOweZMcayVnNUVvBcYXMdhHS257B_PC8PZSWxvhNQ,2540
71
74
  diffsynth_engine/models/basic/timestep.py,sha256=WJODYqkSXEM0wcS42YkkfrGwxWt0e60zMTkDdUBQqBw,2810
72
75
  diffsynth_engine/models/basic/transformer_helper.py,sha256=wJIM7W0fkUOwXGFp8rWZfotIzpJol0mDBhIcSMpDZVM,2941
73
76
  diffsynth_engine/models/basic/unet_helper.py,sha256=4lN6F80Ubm6ip4dkLVmB-Og5-Y25Wduhs9Q8qjyzK6E,9044
74
77
  diffsynth_engine/models/flux/__init__.py,sha256=x0JoxL0CdiiVrY0BjkIrGinud7mcXecLleGO0km91XQ,686
75
- diffsynth_engine/models/flux/flux_controlnet.py,sha256=YvQ5gfzttP092gwLzTzf7IcEnG4smOvILghfoCVuHOo,8163
76
- diffsynth_engine/models/flux/flux_dit.py,sha256=tbj0pVjuOexP4Zn3xjCDiddt0ZW0JSh6uNYOEb-EdlM,23379
77
- diffsynth_engine/models/flux/flux_dit_fbcache.py,sha256=1uG24Cv7IOfrHzwqBv1H_BGAbBsJkylvADaA_gnNjss,8497
78
- diffsynth_engine/models/flux/flux_ipadapter.py,sha256=SHN2hlibLy0OIScGTygV2yNNJZ7FLzpFdBXzykH6wTo,7129
78
+ diffsynth_engine/models/flux/flux_controlnet.py,sha256=xcnokpQ_9z2eToaS-z9dkltA6FI66FzLYkAGh9gvxT4,8248
79
+ diffsynth_engine/models/flux/flux_dit.py,sha256=he1-72wz1gjP5Q-nAsgc_fbr7Vi-OwOyh8rq_x6z8WE,23640
80
+ diffsynth_engine/models/flux/flux_dit_fbcache.py,sha256=i7n08IJZgn_KHZ94_mFpl8gOw9RBj96TNn8R5N1Pq9c,8536
81
+ diffsynth_engine/models/flux/flux_ipadapter.py,sha256=v2gBn1_hOr9JfBAg-zYUYKDUNLODmAnGy5AZoiVhTmg,7231
79
82
  diffsynth_engine/models/flux/flux_redux.py,sha256=tVK4hZxHtC_jx-6vpc0os6dtohBUWK_V8ucdZBkeb-o,2560
80
83
  diffsynth_engine/models/flux/flux_text_encoder.py,sha256=Qcs277RIPP-O5AkcAb5Fb0jToV5o6Qn8hh8nw8zx9g8,3663
81
84
  diffsynth_engine/models/flux/flux_vae.py,sha256=YoeZZSTbXo5rpYQ5-XY3A-Bd1swb8H1yOmtVjm0g5ZI,2994
82
85
  diffsynth_engine/models/sd/__init__.py,sha256=hjoKRnwoXOLD0wude-w7I6wK5ak7ACMbnbkPuBB2oU0,380
83
- diffsynth_engine/models/sd/sd_controlnet.py,sha256=K4iHSCf_UgIlogfLcwvcnnu4RwtgkI-IL-PsWq3e-gM,50761
86
+ diffsynth_engine/models/sd/sd_controlnet.py,sha256=_O5VXm5s3sAunso0UXr0zHLEhHxPSrBtgEAbY5ThsZ8,50648
84
87
  diffsynth_engine/models/sd/sd_text_encoder.py,sha256=gTj8GOGnrUjLDvIFgbHCMJAkdh8nn0E6FtumQ-6L2Xs,5406
85
88
  diffsynth_engine/models/sd/sd_unet.py,sha256=EbF0Vdt4tKp0VjqozBK71-utFAPp6OHZOdWn0czqYNo,12897
86
89
  diffsynth_engine/models/sd/sd_vae.py,sha256=neHAC2HJoiSCQZsyH-ufvIbg1VWmXk3OHz5N_-I5BTQ,1650
@@ -89,7 +92,7 @@ diffsynth_engine/models/sd3/sd3_dit.py,sha256=3G7XVQRw2iD3lFDvzfYXykB3M0QpX42adP
89
92
  diffsynth_engine/models/sd3/sd3_text_encoder.py,sha256=9I57W5AScVoV7L2If4CsOxD_k903S5jlUxEOiUF-xf4,7081
90
93
  diffsynth_engine/models/sd3/sd3_vae.py,sha256=nxZpg616aAagSWALVC4r2B-NLYeA2-u9kPYN0lgBm1I,1428
91
94
  diffsynth_engine/models/sdxl/__init__.py,sha256=3DyhN7p2N2n0MF7RqnTl54JNL0qNQHjIqBMb7W8V2zA,468
92
- diffsynth_engine/models/sdxl/sdxl_controlnet.py,sha256=rfLDGbe9txKS7DCEtCCn1RjD08SnC5BJsXR3LEunCXM,15380
95
+ diffsynth_engine/models/sdxl/sdxl_controlnet.py,sha256=RZH5qfbzm-laiDBmGDG6H-y8Cmd5Gj1lJYvScN1Rg7M,15329
93
96
  diffsynth_engine/models/sdxl/sdxl_text_encoder.py,sha256=VNSx1hfqKAOK-T3m34SaONscAJK8wFkyuVtfcTWmeRA,12369
94
97
  diffsynth_engine/models/sdxl/sdxl_unet.py,sha256=rnD-ZmGf2g_iRuS8I1zBjFJkcoMAE4TxuTjmISWlQhk,12156
95
98
  diffsynth_engine/models/sdxl/sdxl_vae.py,sha256=kNGnn5wKipD31qgMYYLeFyyi8iCc8bsAMD1pEtDhue8,1654
@@ -100,17 +103,17 @@ diffsynth_engine/models/text_encoder/t5.py,sha256=iSYyYQF4DUU0zpN65V_slWoftBTDVw
100
103
  diffsynth_engine/models/vae/__init__.py,sha256=TFSIXZ-UyRaZbEr5KUXm1d4koS5gbgsCi7Soh6jDV0Y,140
101
104
  diffsynth_engine/models/vae/vae.py,sha256=FWMVqahY1BdnIkzLi8ykCp_VWHs05l0JF21wk7763LI,15844
102
105
  diffsynth_engine/models/wan/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
103
- diffsynth_engine/models/wan/wan_dit.py,sha256=pFWSj9Z7ey3PqWBtY7-xIqD8ytUrM-cLopay1K3ndJg,18719
106
+ diffsynth_engine/models/wan/wan_dit.py,sha256=gUd9KeMl7y_VPLntGoGtT2Io94opPiKlrrfo8E8O-8o,18869
104
107
  diffsynth_engine/models/wan/wan_image_encoder.py,sha256=LYwcfCcQmXf9FP08DGaU2bfaPgFfdpJ23OpJP8UCggo,14397
105
108
  diffsynth_engine/models/wan/wan_text_encoder.py,sha256=bkphxtqNNwXcEA_OaUrwV9CvICV-s16awu5Z9gjjzsM,10912
106
109
  diffsynth_engine/models/wan/wan_vae.py,sha256=RxyuHExQmRjGBAqhZdIbtwZFdCibTzh__U4-Sa00zdI,29004
107
- diffsynth_engine/pipelines/__init__.py,sha256=TQnwuOGtnlIBdi-afGvS0W4PXPDVkOyjCAAys0DMowA,605
108
- diffsynth_engine/pipelines/base.py,sha256=5i4pOYE2Mtv9tdVuQSNhkMfN3JwPDdwBfyf6G4b07Sg,14519
110
+ diffsynth_engine/pipelines/__init__.py,sha256=Ewarnhf4K-sYFfSG4mghDoJh5FZKG9Xiz2DFZizNZ-I,452
111
+ diffsynth_engine/pipelines/base.py,sha256=yVp4hSPCqk98azzy3ykKBfPAufvq_ncTFOURN95z7d0,12178
109
112
  diffsynth_engine/pipelines/controlnet_helper.py,sha256=b6HnJFJfMKZq9s5DQ-9Se8OTSDeHVk4AskONSwcRShg,680
110
- diffsynth_engine/pipelines/flux_image.py,sha256=hXCMFScPw6He-FFjWMsP-Eo5cx5rcY1YugsfEUNIJZw,51191
111
- diffsynth_engine/pipelines/sd_image.py,sha256=A92oBYZt37x4uOYhl3dAiCzkcgsPmRmH1z4_IWEULeA,18807
112
- diffsynth_engine/pipelines/sdxl_image.py,sha256=P1bFvFg3Dd49BAEihig-e5QkoNtztJqVxdfAuD693FU,22671
113
- diffsynth_engine/pipelines/wan_video.py,sha256=LqDBZILcIOK6SR0zPrDJgBjpBgkIah8N6Q7I-8XHchY,21584
113
+ diffsynth_engine/pipelines/flux_image.py,sha256=CPI7WwXJz60rpQN_ZfVV5kcWxg8WlrYNNe6QI1E5EPk,48851
114
+ diffsynth_engine/pipelines/sd_image.py,sha256=hiL2gvQcPgniRm8TlzUKhoo5bGUmmVlYDS__E_WFDiE,17834
115
+ diffsynth_engine/pipelines/sdxl_image.py,sha256=qWpE5q0CeDrsKZVIxYHpYLyZAkKMqACDzf4RPPvdn7A,21587
116
+ diffsynth_engine/pipelines/wan_video.py,sha256=vi_xW-jU4PeMtZzjkfQbnj8eOymJrTZMrOQau6tx6ks,20187
114
117
  diffsynth_engine/processor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
115
118
  diffsynth_engine/processor/canny_processor.py,sha256=hV30NlblTkEFUAmF_O-LJrNlGVM2SFrqq6okfF8VpOo,602
116
119
  diffsynth_engine/processor/depth_processor.py,sha256=dQvs3JsnyMbz4dyI9QoR8oO-mMFBFAgNvgqeCoaU5jk,1532
@@ -120,16 +123,16 @@ diffsynth_engine/tokenizers/clip.py,sha256=TbcJu5-Z7x9CJNWJuekasy4xuLJImBwZnXyX9
120
123
  diffsynth_engine/tokenizers/t5.py,sha256=5ENHjfLx0n5oYo0hVFFTpgJCNorZgskHl3nitjVJW8g,7453
121
124
  diffsynth_engine/tokenizers/wan.py,sha256=4bvibHZKNQYHnl0oSyN_pJK5PAxkUC0TWgbNDgckdCQ,2265
122
125
  diffsynth_engine/tools/__init__.py,sha256=3sgNSD5sts3cHSyKVH5qziTykJQ6gDRMx_RyrrxSzZ8,388
123
- diffsynth_engine/tools/flux_inpainting_tool.py,sha256=aBsrXAzjR5cZXDIL4rUbXeHFrQODgiB7nD4HCjd_dbQ,2414
124
- diffsynth_engine/tools/flux_outpainting_tool.py,sha256=7km8XWfl-BsCHcMpflZsyDx6HtvuAD-TrveK0u2xdfk,2738
125
- diffsynth_engine/tools/flux_reference_tool.py,sha256=qE2I05bN0_5VlZBckB87S8KOklh7ilY3LjKfpswjLUM,3435
126
- diffsynth_engine/tools/flux_replace_tool.py,sha256=rs0EJtZpwaJ8ZxT_IxGE4XdFUBotTzQeC8AdnufcQd4,3177
126
+ diffsynth_engine/tools/flux_inpainting_tool.py,sha256=AoGmBYMnk62joYYg5wN9orj1eNNqByGUQZLuHAhlZzQ,2532
127
+ diffsynth_engine/tools/flux_outpainting_tool.py,sha256=sxGRAiht27he9CT_dL9KkXVvMUeHC052IrKUIYD1fO8,2856
128
+ diffsynth_engine/tools/flux_reference_tool.py,sha256=BJlXQxH8j3AhEhlymIlE6OnIH2gU_l_qv5k10JDZKng,3705
129
+ diffsynth_engine/tools/flux_replace_tool.py,sha256=M_q8KnsBEwNi4w8NOK-F2Bmj7cKUNcA9QMzwrp3zm6E,3336
127
130
  diffsynth_engine/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
128
131
  diffsynth_engine/utils/constants.py,sha256=L7sIxGNMfCvcZG66ul7GIT6fDctkcwhePAjMjG6WXx8,1969
129
132
  diffsynth_engine/utils/download.py,sha256=NCgfL9tUca-sOhT41k6w4o__Ktbw-1aDwFTR4JDkT28,5639
130
133
  diffsynth_engine/utils/env.py,sha256=43x-kBjt5zI2cwZ9G4BOeTbedi2k6TuBzHGOBeFbFvU,280
131
134
  diffsynth_engine/utils/flag.py,sha256=6zQLnoEaU69pBEyhavCgydQfP0khw5ppCU7sue4yRqg,1370
132
- diffsynth_engine/utils/fp8_linear.py,sha256=qu6Hzi7dqmDFgtoP-Uf0p-GDKW03AK9338YeLuzw2nw,3589
135
+ diffsynth_engine/utils/fp8_linear.py,sha256=NosnWMoAr_IpFcLn-OYbAx-vXySphjxutDZqmXLNjJI,4064
133
136
  diffsynth_engine/utils/gguf.py,sha256=ZWvw46V4g4uVyAR_oCq-4K5nPdKVrYk3u47uXMgA9lU,14092
134
137
  diffsynth_engine/utils/image.py,sha256=_46CVs1Qe7GdZNulWWJISnR_Y6FotC2tZGLKtr04gIE,562
135
138
  diffsynth_engine/utils/loader.py,sha256=Z5v1WNDWFY0OrVubB70j5VU3zeaAfEK_j8c1KrGI4yM,1240
@@ -137,12 +140,12 @@ diffsynth_engine/utils/lock.py,sha256=1Ipgst9eEFfFdViAvD5bxdB6HnHHBcqWYOb__fGaPU
137
140
  diffsynth_engine/utils/logging.py,sha256=XB0xTT8PBN6btkOjFtOvjlrOCRVgDGT8PFAp1vmse28,467
138
141
  diffsynth_engine/utils/offload.py,sha256=jUR4u7J60o4KZIRxHhMCwaeDkiXJvBa0KJkYKKT6mrg,1587
139
142
  diffsynth_engine/utils/onnx.py,sha256=jeWUudJHnESjuiEAHyUZYUZz7dCj34O9aGjHCe8yjWo,1149
140
- diffsynth_engine/utils/parallel.py,sha256=2WISMBTTmW0v2qPvpms421-B59v3bYlS6YrLq9BZ5Zo,16909
141
- diffsynth_engine/utils/platform.py,sha256=q9ifmdzoa66Cj9YKfwps21DsDdwA0JGpwroKQbG6shU,224
143
+ diffsynth_engine/utils/parallel.py,sha256=gbIeilfOYsqeDcgkaP68TfLjIXxvD0KfLiAsR_8gJco,16917
144
+ diffsynth_engine/utils/platform.py,sha256=2lXdw6YkqcRONCeT98n4cyg1Ii8Ybbyj2Ns72Se9tlk,496
142
145
  diffsynth_engine/utils/prompt.py,sha256=YItMchoVzsG6y-LB4vzzDUWrkhKRVlt1HfVhxZjSxMQ,280
143
146
  diffsynth_engine/utils/video.py,sha256=Ne0rd2lb59UT1q5EotpjlY7OT8F9oTCFDyo1ST77uoQ,1004
144
- diffsynth_engine-0.3.6.dev8.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
145
- diffsynth_engine-0.3.6.dev8.dist-info/METADATA,sha256=bZRH8-guipkEhR276zMywqxu72Ayr-c2XPttJedLZ2o,1068
146
- diffsynth_engine-0.3.6.dev8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
147
- diffsynth_engine-0.3.6.dev8.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
148
- diffsynth_engine-0.3.6.dev8.dist-info/RECORD,,
147
+ diffsynth_engine-0.3.6.dev10.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
148
+ diffsynth_engine-0.3.6.dev10.dist-info/METADATA,sha256=fmz3b9c9qbb08cBHcF9Bj4VrTqseZOyvBck3xEcHU4c,1069
149
+ diffsynth_engine-0.3.6.dev10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
150
+ diffsynth_engine-0.3.6.dev10.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
151
+ diffsynth_engine-0.3.6.dev10.dist-info/RECORD,,