InvokeAI 6.10.0__py3-none-any.whl → 6.10.0rc1__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 (40) hide show
  1. invokeai/app/invocations/flux_denoise.py +1 -15
  2. invokeai/app/invocations/metadata_linked.py +0 -47
  3. invokeai/app/invocations/z_image_denoise.py +84 -244
  4. invokeai/app/services/config/config_default.py +1 -3
  5. invokeai/app/services/model_manager/model_manager_default.py +0 -7
  6. invokeai/backend/flux/denoise.py +11 -196
  7. invokeai/backend/model_manager/configs/lora.py +0 -36
  8. invokeai/backend/model_manager/load/model_cache/model_cache.py +2 -104
  9. invokeai/backend/model_manager/load/model_loaders/cogview4.py +1 -2
  10. invokeai/backend/model_manager/load/model_loaders/flux.py +6 -13
  11. invokeai/backend/model_manager/load/model_loaders/generic_diffusers.py +2 -4
  12. invokeai/backend/model_manager/load/model_loaders/onnx.py +0 -1
  13. invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py +1 -2
  14. invokeai/backend/model_manager/load/model_loaders/z_image.py +3 -37
  15. invokeai/backend/model_manager/starter_models.py +4 -13
  16. invokeai/backend/patches/lora_conversions/z_image_lora_conversion_utils.py +5 -39
  17. invokeai/backend/quantization/gguf/ggml_tensor.py +4 -15
  18. invokeai/backend/z_image/extensions/regional_prompting_extension.py +12 -10
  19. invokeai/frontend/web/dist/assets/App-CYhlZO3Q.js +161 -0
  20. invokeai/frontend/web/dist/assets/{browser-ponyfill-4xPFTMT3.js → browser-ponyfill-DHZxq1nk.js} +1 -1
  21. invokeai/frontend/web/dist/assets/{index-vCDSQboA.js → index-dgSJAY--.js} +51 -51
  22. invokeai/frontend/web/dist/index.html +1 -1
  23. invokeai/frontend/web/dist/locales/en.json +5 -11
  24. invokeai/version/invokeai_version.py +1 -1
  25. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/METADATA +2 -2
  26. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/RECORD +32 -39
  27. invokeai/app/invocations/pbr_maps.py +0 -59
  28. invokeai/backend/flux/schedulers.py +0 -62
  29. invokeai/backend/image_util/pbr_maps/architecture/block.py +0 -367
  30. invokeai/backend/image_util/pbr_maps/architecture/pbr_rrdb_net.py +0 -70
  31. invokeai/backend/image_util/pbr_maps/pbr_maps.py +0 -141
  32. invokeai/backend/image_util/pbr_maps/utils/image_ops.py +0 -93
  33. invokeai/frontend/web/dist/assets/App-BBELGD-n.js +0 -161
  34. invokeai/frontend/web/dist/locales/en-GB.json +0 -1
  35. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/WHEEL +0 -0
  36. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/entry_points.txt +0 -0
  37. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/licenses/LICENSE +0 -0
  38. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/licenses/LICENSE-SD1+SD2.txt +0 -0
  39. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/licenses/LICENSE-SDXL.txt +0 -0
  40. {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/top_level.txt +0 -0
@@ -18,7 +18,7 @@
18
18
  overflow: hidden;
19
19
  }
20
20
  </style>
21
- <script type="module" crossorigin src="./assets/index-vCDSQboA.js"></script>
21
+ <script type="module" crossorigin src="./assets/index-dgSJAY--.js"></script>
22
22
  </head>
23
23
 
24
24
  <body dir="ltr">
@@ -850,7 +850,6 @@
850
850
  "parsingFailed": "Parsing Failed",
851
851
  "positivePrompt": "Positive Prompt",
852
852
  "qwen3Encoder": "Qwen3 Encoder",
853
- "qwen3Source": "Qwen3 Source",
854
853
  "recallParameters": "Recall Parameters",
855
854
  "recallParameter": "Recall {{label}}",
856
855
  "scheduler": "Scheduler",
@@ -1049,10 +1048,10 @@
1049
1048
  "t5Encoder": "T5 Encoder",
1050
1049
  "qwen3Encoder": "Qwen3 Encoder",
1051
1050
  "zImageVae": "VAE (optional)",
1052
- "zImageVaePlaceholder": "From VAE source model",
1051
+ "zImageVaePlaceholder": "From Qwen3 Source (leave empty)",
1053
1052
  "zImageQwen3Encoder": "Qwen3 Encoder (optional)",
1054
- "zImageQwen3EncoderPlaceholder": "From Qwen3 source model",
1055
- "zImageQwen3Source": "Qwen3 & VAE Source Model",
1053
+ "zImageQwen3EncoderPlaceholder": "From Qwen3 Source (leave empty)",
1054
+ "zImageQwen3Source": "Qwen3 Source (Diffusers Z-Image)",
1056
1055
  "zImageQwen3SourcePlaceholder": "Required if VAE/Encoder empty",
1057
1056
  "upcastAttention": "Upcast Attention",
1058
1057
  "uploadImage": "Upload Image",
@@ -2502,9 +2501,6 @@
2502
2501
  "y": "Y (YCbCr)",
2503
2502
  "cb": "Cb (YCbCr)",
2504
2503
  "cr": "Cr (YCbCr)"
2505
- },
2506
- "pbr_maps": {
2507
- "label": "Create PBR Maps"
2508
2504
  }
2509
2505
  },
2510
2506
  "transform": {
@@ -2821,10 +2817,8 @@
2821
2817
  "whatsNew": {
2822
2818
  "whatsNewInInvoke": "What's New in Invoke",
2823
2819
  "items": [
2824
- "Support for Z-Image-Turbo: InvokeAI now supports the fast and accurate Z-Image-Turbo model. See 'Starter Models' to get started.",
2825
- "Workflow Editor: You can now organize your custom workflows with tags to facilitate search and retrieval.",
2826
- "Hotkeys: Add and edit keyboard shortcuts for any of Invoke's major functions.",
2827
- "Model Manager: The Model Manager user interface now supports bulk tagging and deletion."
2820
+ "Model Manager: If a model cannot be identified during installation, you can now manually select the model type and architecture.",
2821
+ "Internal: Improved model identification system, making it easier for contributors to add support for new models."
2828
2822
  ],
2829
2823
  "readReleaseNotes": "Read Release Notes",
2830
2824
  "watchRecentReleaseVideos": "Watch Recent Release Videos",
@@ -1 +1 @@
1
- __version__ = "6.10.0"
1
+ __version__ = "6.10.0rc1"
@@ -1,7 +1,7 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: InvokeAI
3
- Version: 6.10.0
4
- Summary: A full-featured AI-assisted image generation environment designed for creatives and enthusiasts.
3
+ Version: 6.10.0rc1
4
+ Summary: An implementation of Stable Diffusion which provides various new features and options to aid the image generation process
5
5
  Author-email: Invoke <support@invoke.ai>
6
6
  License: Apache License
7
7
  Version 2.0, January 2004
@@ -46,7 +46,7 @@ invokeai/app/invocations/facetools.py,sha256=wRdN9cxbM5-LhHvoJA0f4wcP9aAf-WPZAfe
46
46
  invokeai/app/invocations/fields.py,sha256=5aEA7iZm3-dmCMJzcXQJHcs-dtCzlGVzYGZMB0suU8U,35879
47
47
  invokeai/app/invocations/flux_control_lora_loader.py,sha256=vNyVB3BhO7MpgE7aeTseIE5A0ALUgStwq0BFbko1otw,1872
48
48
  invokeai/app/invocations/flux_controlnet.py,sha256=9EoVC8F5rrxRgPdZkC7hv3SKtD8fHOHM9iG9lBD7J8k,4316
49
- invokeai/app/invocations/flux_denoise.py,sha256=DSrGU9Orzs4uFHKhRBdRtl91yyJf6-YsqxCehThqfys,44024
49
+ invokeai/app/invocations/flux_denoise.py,sha256=tvss4h-JNYNavUgFqvdHSg5XL--jQLvLNYoPrHMkr1A,43283
50
50
  invokeai/app/invocations/flux_fill.py,sha256=__lEdjXjLHSz2D3SIQVmTWjZat9Icn8pPIVtjairy5k,1448
51
51
  invokeai/app/invocations/flux_ip_adapter.py,sha256=NnoRSoUceXDZC8PV17l8Vsqnq_32_lSxdRCxBamizqw,3962
52
52
  invokeai/app/invocations/flux_kontext.py,sha256=uEQhgIOIA1gVLrjRK1FbG4P_g93hnvXXwxW-dAVZtgA,1344
@@ -73,13 +73,12 @@ invokeai/app/invocations/mask.py,sha256=R9n9TkcFJrAL4UCm5I2xBngh7ItZ6vTPREMajiIj
73
73
  invokeai/app/invocations/math.py,sha256=y00c6frIIowhN2lwTHYDjnuozNAtHGBa8_7FfHcRxLk,10441
74
74
  invokeai/app/invocations/mediapipe_face.py,sha256=Gby8JWhELjR_0Hm_qgGurLexPzWpGLyE5NKuyw4xqXM,1308
75
75
  invokeai/app/invocations/metadata.py,sha256=xc5r6mnLGzlmiAR6ihfCh45G9B8RqZjMZg-UMq2X0nQ,13500
76
- invokeai/app/invocations/metadata_linked.py,sha256=AZCPTG9GNHt1CnOOUv64VBYA4WXQ3RvZbbiJbjv6afg,46655
76
+ invokeai/app/invocations/metadata_linked.py,sha256=RIiXR09bfMEYNgPnrFBH7j1wYjWiATl9dS1TBBdafOs,44765
77
77
  invokeai/app/invocations/mlsd.py,sha256=uokKzWTHB-poQTeFY4FYznOC4QRhUw7mNRg4E82T7_I,1737
78
78
  invokeai/app/invocations/model.py,sha256=CL6SzVpvGhj-1DM6gnL527me6i7DAZNpQZ6PJlHYDUI,21375
79
79
  invokeai/app/invocations/noise.py,sha256=7C7a2UM_zpGdUI35qOTr_uiW41zUIsHzZSz1oZV9pI8,3585
80
80
  invokeai/app/invocations/normal_bae.py,sha256=gNP8WEpy7XqlwbZJ3nUuftvSfPTi0p34VGiDOThV2-8,1298
81
81
  invokeai/app/invocations/param_easing.py,sha256=2qDOeBeSMM25KyFGQLwM2Yf3TE8EQFvSpFFhRTkUg04,1031
82
- invokeai/app/invocations/pbr_maps.py,sha256=UVj8gG7qkNoewdyPrZ6ktXcA9xttlrSaP4xqQSpvUd0,3075
83
82
  invokeai/app/invocations/pidi.py,sha256=PW0KjmwPzgA-FThtXiCkAedLX0BTJQvaoAxMk-KYvpE,1584
84
83
  invokeai/app/invocations/primitives.py,sha256=Aw72nkZzfYkKKiEHafM4LWW9RZuPUlFXyntndA5J02I,18112
85
84
  invokeai/app/invocations/prompt.py,sha256=NKjGhuM8oAF7P-ipyMF0lR3dl3mmgcqtIWlgndai4Go,3638
@@ -101,7 +100,7 @@ invokeai/app/invocations/tiles.py,sha256=Q_09_E533aKvcDQouyX6RP_C5b0qMddCig18BWY
101
100
  invokeai/app/invocations/upscale.py,sha256=C11Noz4-J0zIT1kXY7EBOuFVGhgkveAi7xkmRJaG2Gw,4429
102
101
  invokeai/app/invocations/util.py,sha256=mK2su6x-ZWHPFqdjcWwrv0ZO-70RBlw6kcBqsQuiKlQ,665
103
102
  invokeai/app/invocations/z_image_control.py,sha256=15QSRqqquZiz4xRkAUJ9l9ZImbqHKFUwCZ4jF36WQEs,3783
104
- invokeai/app/invocations/z_image_denoise.py,sha256=vibjkIOWgcegfptaF1hXrnYisCpePa4TPte1MS7J8W0,37535
103
+ invokeai/app/invocations/z_image_denoise.py,sha256=xJ-iw9kn_BXjtNhV8qEy-74LlXXokUBuwGhQGBxqzL0,28342
105
104
  invokeai/app/invocations/z_image_image_to_latents.py,sha256=cu4WMqme9T8ZmchIwmOnRgn354_-Et-2AtjlMDxeaYQ,4590
106
105
  invokeai/app/invocations/z_image_latents_to_image.py,sha256=zNeYJVtMn0VuSgkFfCBcRF_8iG6H_Kaftkgx0FAKK-U,4252
107
106
  invokeai/app/invocations/z_image_lora_loader.py,sha256=A-6rDAd0TnzeaSbTte3NZD0GOaa3dpG3cGX38sLUTk4,5615
@@ -133,7 +132,7 @@ invokeai/app/services/client_state_persistence/client_state_persistence_base.py,
133
132
  invokeai/app/services/client_state_persistence/client_state_persistence_sqlite.py,sha256=OHc1PQ724ua5US3hMzK19wnteg7PsFxWcWz9h6FRPds,2043
134
133
  invokeai/app/services/config/__init__.py,sha256=h3XnlBQvL7wPI2oZOvXXRW-DTBBX9obvCE0ELjHkldM,282
135
134
  invokeai/app/services/config/config_common.py,sha256=NwVkcdzv2VVbAL2Izu0AtF0ep2fMBT70qU0fhAoedqg,718
136
- invokeai/app/services/config/config_default.py,sha256=kSuRfA-70fm8HESyF3H0FskSztfmo25Ybp0qa3AS5ak,39389
135
+ invokeai/app/services/config/config_default.py,sha256=V9jULV8vLYXE6YICCp5dGApIciILI0bUgURHlh6mSDM,38715
137
136
  invokeai/app/services/download/__init__.py,sha256=DKuNv-oRbzR-IhJjP6N-tKyyI6VJ3OPlgaFqz25J0EA,512
138
137
  invokeai/app/services/download/download_base.py,sha256=kukndGneEWPrdekXjDVizpEv9iX5KcaDngjhCa5tE2M,13033
139
138
  invokeai/app/services/download/download_default.py,sha256=LtohvfyJSQMvUn9U9CzlIIQDYI-SAPp-7MrSBMvXYAI,24577
@@ -178,7 +177,7 @@ invokeai/app/services/model_load/model_load_default.py,sha256=jnkuIa8LxKAZ6IPqNQ
178
177
  invokeai/app/services/model_manager/__init__.py,sha256=fD9cM2SRoHoGP9HqGTbNX4ChZkMMIVV8uqUD1vj9avI,321
179
178
  invokeai/app/services/model_manager/model_manager_base.py,sha256=uUAXhQT_hlk8yKqFze2k179Xgdn29BcNzl_Qch9D_po,2380
180
179
  invokeai/app/services/model_manager/model_manager_common.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
181
- invokeai/app/services/model_manager/model_manager_default.py,sha256=_oDZ_P0IxTcSrYhD_YzSpceK_aMpGEiR-4dgkwtpn4c,4456
180
+ invokeai/app/services/model_manager/model_manager_default.py,sha256=d-VcNTXOur9BuaaCAfD9DJBeBFbpzr2yFYqFs0Vx-qY,4140
182
181
  invokeai/app/services/model_records/__init__.py,sha256=OzsXFpijEMROZWxLq2gaHgpDhUReybvB3AcVcHz0VHE,571
183
182
  invokeai/app/services/model_records/model_records_base.py,sha256=HScjV-5897UKt42oxMvp_l__5D-CH1WGHmbza6EY2So,9472
184
183
  invokeai/app/services/model_records/model_records_sql.py,sha256=jlZlRFMsDRGkMuO6bbD2jlbsY1NAsGbs88QGfA9VCdE,14667
@@ -314,12 +313,11 @@ invokeai/backend/raw_model.py,sha256=g7RfS65MHixII0uB0bczftXa6XZ2AJrjGes6vBdtWDY
314
313
  invokeai/backend/spandrel_image_to_image_model.py,sha256=8LEPnxmsq49VdDpJuM8k_sT22owSQU86wzYeyvI8mQc,5237
315
314
  invokeai/backend/textual_inversion.py,sha256=JHxVoMcTarnjL-FBxBi_lOdQZop6U1glrQ-iH0H5ttI,5277
316
315
  invokeai/backend/flux/custom_block_processor.py,sha256=mwuqEEo-_TkEn4rDCVMnrcMYKLeo7XfPKXEiJcKzrMc,5670
317
- invokeai/backend/flux/denoise.py,sha256=9Rt1W3eVv7z6dsTtzFuwYN_ya-XFSI8322y_NB5hmxo,18003
316
+ invokeai/backend/flux/denoise.py,sha256=SJ9VudtZjKY8g1hmnPU-JVQsuCgbpL_2OX3clZLrUJo,8588
318
317
  invokeai/backend/flux/flux_state_dict_utils.py,sha256=qk3zY_fjeEpfZLfqBeUV7npPFlFv1ecceFXpNA8YpE8,752
319
318
  invokeai/backend/flux/math.py,sha256=q0XdfT_kI2cnhBmreqo_jQ3aWrkUvcu4K0ktuFx7rwA,1365
320
319
  invokeai/backend/flux/model.py,sha256=2kqMUH7adnOHTYj8OGakAehrSAZnXOiaIN3aFGdmo7w,6188
321
320
  invokeai/backend/flux/sampling_utils.py,sha256=spFX8v5iq5x3ytVVUMqDSRwHhdoqDKfBAzqvjo47ae8,6531
322
- invokeai/backend/flux/schedulers.py,sha256=XczIzWrsOMHhVyA-bb4Hg_JTKWcZKM1KP_sscMrXfbI,1836
323
321
  invokeai/backend/flux/text_conditioning.py,sha256=AjxqRWTx1FH2LJtPg7KnXsrpmBw8yy8SZjoRTVRuw24,1332
324
322
  invokeai/backend/flux/util.py,sha256=af747OCqYTyS8Y1dAN-txhjDmEA7Yiy4VHS9OKy0qR8,3028
325
323
  invokeai/backend/flux/controlnet/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -412,10 +410,6 @@ invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet
412
410
  invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/activations/activations.py,sha256=m0KlKEZY5yNTDey-rZuNDmAONMsU547cwZx25X8dAeU,2690
413
411
  invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/activations/activations_jit.py,sha256=qGNmkNhseyrm8rG8VYv17Wx7SguHnqp48qfwoAy6A3I,2294
414
412
  invokeai/backend/image_util/normal_bae/nets/submodules/efficientnet_repo/geffnet/activations/activations_me.py,sha256=HzYPuHidaIq_L4K9PVATILvwQX5wwucUDw6H2QrMCTs,4549
415
- invokeai/backend/image_util/pbr_maps/pbr_maps.py,sha256=fj8mLM5kPcArmH_SOUSLPliuedmzxz8M4ApT9za09yU,5459
416
- invokeai/backend/image_util/pbr_maps/architecture/block.py,sha256=IwTtG_W8IR4WGERIM8LlB89J8YmHzxkbopI6vIoirP4,11483
417
- invokeai/backend/image_util/pbr_maps/architecture/pbr_rrdb_net.py,sha256=El1g_Ga288ws4mgzTQRdC5kRv5tjr4oMrPORdnxCg6o,2214
418
- invokeai/backend/image_util/pbr_maps/utils/image_ops.py,sha256=H3uI09rCHOAGFoyiBfPpM2E50WHJoSL8WHVsp7fO07w,3480
419
413
  invokeai/backend/image_util/pidi/__init__.py,sha256=nv2tcwotHB8kVh6k8mJzh2VEIEeRsRzAr8duwr-bYes,2599
420
414
  invokeai/backend/image_util/pidi/model.py,sha256=vjLYSwFxSJnoMmY-N0mRTrWI9adSXTPvk_wnor-dYo4,21813
421
415
  invokeai/backend/image_util/realesrgan/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -435,7 +429,7 @@ invokeai/backend/model_manager/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
435
429
  invokeai/backend/model_manager/model_on_disk.py,sha256=h-v3Q8e7svjtXy81zmXjSS1dp6Q6jWyTPg6j0V9G5Xc,5764
436
430
  invokeai/backend/model_manager/search.py,sha256=ahUgpryoemLVSq2quaom0BXFbYyCtAh29ASQXgfisYI,4902
437
431
  invokeai/backend/model_manager/single_file_config_files.py,sha256=JPF7Fa3223_xetd_9JFHmWTkgtaiUAj7fj1mIX8osKY,3406
438
- invokeai/backend/model_manager/starter_models.py,sha256=kp9HeiFI2L9sx96IeVFscwBZ8bn2lMc-1K-Xp4rTgq4,32726
432
+ invokeai/backend/model_manager/starter_models.py,sha256=EQ057A8puI8wP5z2FEoCCoLepdhY8rTC6TyBFyt9FXw,32406
439
433
  invokeai/backend/model_manager/taxonomy.py,sha256=wBP8Fx19gqy_LePiYQVKHFiOtm4aFFyCYCoQrfRNCmk,5603
440
434
  invokeai/backend/model_manager/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
441
435
  invokeai/backend/model_manager/configs/base.py,sha256=65b3MjalpzERdyulK-ejWxZgW7Qd4zc4psOvKh2ymDg,10471
@@ -448,7 +442,7 @@ invokeai/backend/model_manager/configs/flux_redux.py,sha256=9s9lwfsccg6f0T3RkzX7
448
442
  invokeai/backend/model_manager/configs/identification_utils.py,sha256=O10Sz3X70V7dZys92L5yVt_pT9WadFjQuLn3mNutMLA,8174
449
443
  invokeai/backend/model_manager/configs/ip_adapter.py,sha256=tJY6z5gIxosD9UPKfLlioWrNXLZYkQzogZiaaTBqs3s,6889
450
444
  invokeai/backend/model_manager/configs/llava_onevision.py,sha256=TxI8de2XHe9Fx76-oczqe2RHtjnZ3O3bhCFTkc_Cn7g,1233
451
- invokeai/backend/model_manager/configs/lora.py,sha256=3UAA39pwfxPml8y3v2WTJ332Uhzdpoi_fqZy_wjww_U,15383
445
+ invokeai/backend/model_manager/configs/lora.py,sha256=JKGJFisCRJcXa5hXD2Z7YY_OxVIzQuicYp7lbdJkYCU,14130
452
446
  invokeai/backend/model_manager/configs/main.py,sha256=87CG_hH_UutLTx8MFgPDeToyk9FgQMoAJw6ssn1sJ_g,31891
453
447
  invokeai/backend/model_manager/configs/qwen3_encoder.py,sha256=wdxcewabEdygAcdiL2Hb9LMwNQzzt4yOTOfOOe_2KnU,6181
454
448
  invokeai/backend/model_manager/configs/siglip.py,sha256=eUW7XguQazWBnYX-F4zy7H7EcEU1Sq5064wcjYWZD74,1265
@@ -469,7 +463,7 @@ invokeai/backend/model_manager/load/model_cache/__init__.py,sha256=47DEQpj8HBSa-
469
463
  invokeai/backend/model_manager/load/model_cache/cache_record.py,sha256=6433UMwWvtxVol3Tz_okMYyM1NFvpcc2Ci5RcjtGKwM,882
470
464
  invokeai/backend/model_manager/load/model_cache/cache_stats.py,sha256=Y6QN5itSe3sFP2QW8DQXf6NlmZ_nHe4QkFieyxdZJh8,506
471
465
  invokeai/backend/model_manager/load/model_cache/dev_utils.py,sha256=260Xa4O4DpMvaGpjzAD_fvRT17HeAqGUdA6lZFi0OHI,1065
472
- invokeai/backend/model_manager/load/model_cache/model_cache.py,sha256=yzsHUQLtToPMuoYyWvjbT9iljmOtR2dmdou6pT3jVt0,40803
466
+ invokeai/backend/model_manager/load/model_cache/model_cache.py,sha256=IUgwem_F8-fLy2_vIes9VwDeDFZX2vCGqiLt8gkaX2Q,36528
473
467
  invokeai/backend/model_manager/load/model_cache/utils.py,sha256=HQMeQ_VXodjjtmgJfLnsS0tWaqILILGtupIrzkblsUM,772
474
468
  invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_only_full_load.py,sha256=i20Y7e1SWnu9e7Ry0BcgytpqYeb6oxWdR2H7V-1irBs,4732
475
469
  invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_with_partial_load.py,sha256=VJ1UZRM2FpEBorSxZRVXFDqy6MDRK6C7sfb4tXXWv4U,15569
@@ -491,20 +485,20 @@ invokeai/backend/model_manager/load/model_cache/torch_module_autocast/custom_mod
491
485
  invokeai/backend/model_manager/load/model_cache/torch_module_autocast/custom_modules/utils.py,sha256=T7GckoNvrOUYeRy8PpxDq27QrHnJOK69LODbX7TUqAI,626
492
486
  invokeai/backend/model_manager/load/model_loaders/__init__.py,sha256=EDgdUKmYMiisOPDHtaRU2uwbfSlLbU6hoQybJqVgS7k,37
493
487
  invokeai/backend/model_manager/load/model_loaders/clip_vision.py,sha256=rFcaHsnK75II5YbahOAxnYeHfw1DFltM1004IaUa2jY,1418
494
- invokeai/backend/model_manager/load/model_loaders/cogview4.py,sha256=-kiwWNpP2f8Dso5jpBRRlgkbYxSaTbkWQ1actC7LJvc,2314
488
+ invokeai/backend/model_manager/load/model_loaders/cogview4.py,sha256=g6thyuL21xZ8PjwTQr-TytiJd7uXRckZGcYWN-hA12M,2252
495
489
  invokeai/backend/model_manager/load/model_loaders/controlnet.py,sha256=KkHv_oGphPcBqTfD3p9hkKrhPoV48ZH_-XwyTnVHXJQ,2002
496
- invokeai/backend/model_manager/load/model_loaders/flux.py,sha256=gg5dy_EwXriBUnTXID-b4ZSp_RrtHRG8030zxD5Bv_Q,18280
497
- invokeai/backend/model_manager/load/model_loaders/generic_diffusers.py,sha256=xQKRXpNyK8SzI3Gmj5ct2ap_wDfLjeWkQ9uwhqJ1DnA,4799
490
+ invokeai/backend/model_manager/load/model_loaders/flux.py,sha256=UDuCTjnfFGwmq9JafkTeZOjiGf1PuCF5CAsoOBe_76c,18005
491
+ invokeai/backend/model_manager/load/model_loaders/generic_diffusers.py,sha256=WYID2rL0dAoC9YdaxpR_z65Qn_jxo4LruF61z7zxVUQ,4723
498
492
  invokeai/backend/model_manager/load/model_loaders/ip_adapter.py,sha256=GPdwMWKhHPodF8yc6xf4jOqh3GdqsiVmfc81MYWqnmA,1394
499
493
  invokeai/backend/model_manager/load/model_loaders/llava_onevision.py,sha256=U8D2ac8iRM5Iip5C_isYNEnRE-iannvJHw_WOapOEBQ,1251
500
494
  invokeai/backend/model_manager/load/model_loaders/lora.py,sha256=SediPp6EvtvwTycdONzwe-Vl3VB7m7z7WSH1vn5INIM,8454
501
- invokeai/backend/model_manager/load/model_loaders/onnx.py,sha256=sFOdAx6J1KrwjxFjFpduZ915jYY8QbdJg-bjED5ZivQ,1713
495
+ invokeai/backend/model_manager/load/model_loaders/onnx.py,sha256=TZfc3Eg9SaCIxCJi67Y9Sj1TZqCJ4UzYLGBIu_LQ2pI,1678
502
496
  invokeai/backend/model_manager/load/model_loaders/sig_lip.py,sha256=Q9BoXewnDE3X1Lu1oHkmsAbKkMAi6ED_Xff1ofvx8ao,1084
503
497
  invokeai/backend/model_manager/load/model_loaders/spandrel_image_to_image.py,sha256=0NGwvcNsYeLTfFPyhv5JHopKsVeEzLzrYSy4kbrImew,1579
504
- invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py,sha256=t4fxWo-iakFrZ3TiQ0vgmg9k05t2qzGEgoobRUswzJ0,7224
498
+ invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py,sha256=u_hSnc4CuQIuQ3hfx-f6HXL5ZC6vqdu2VJf8gGSn0HU,7162
505
499
  invokeai/backend/model_manager/load/model_loaders/textual_inversion.py,sha256=S8ZckTM_tb7jPZvtvn3lybyUpFlK76od3XhEosYF3pk,1785
506
500
  invokeai/backend/model_manager/load/model_loaders/vae.py,sha256=cEgqLy2dxFLLFG38whlS1uIV0DwmDCcE-akFlSVlCcY,1415
507
- invokeai/backend/model_manager/load/model_loaders/z_image.py,sha256=Tw9w1EcVvMZ7U3SxaPTkBEbx6yGUb8mRrtgPPVCpcyw,42595
501
+ invokeai/backend/model_manager/load/model_loaders/z_image.py,sha256=E0uqcgUoUCF5A2uiz5gtjeoLls15EQectyGcDU9-Okc,40693
508
502
  invokeai/backend/model_manager/metadata/__init__.py,sha256=TRImeMpoWu7ZVEH35eLP_w6euT0frVJhupk1D5wZ7bc,1049
509
503
  invokeai/backend/model_manager/metadata/metadata_base.py,sha256=LqJhI70LMKxqNrTNlCgFPLFL9-LijxMvXCxAZn3uGc4,5903
510
504
  invokeai/backend/model_manager/metadata/fetch/__init__.py,sha256=4qbbpEPEGbiO6MPEekd6txNcPHlnvZakUxALDRwxZq4,533
@@ -559,11 +553,11 @@ invokeai/backend/patches/lora_conversions/kohya_key_utils.py,sha256=2_dNOX8zaq7M
559
553
  invokeai/backend/patches/lora_conversions/sd_lora_conversion_utils.py,sha256=D2KxH-EQqltjw5hHmrR8pYBxca0ENYzpp_0Gtq6pxcY,1058
560
554
  invokeai/backend/patches/lora_conversions/sdxl_lora_conversion_utils.py,sha256=uMCayjpvfqeWV3CmudWgzyXlMfAs7bwjklczTsrg8sI,6942
561
555
  invokeai/backend/patches/lora_conversions/z_image_lora_constants.py,sha256=KfFKRvDghRFB1rtBV8_9BxmDBYH5WmvJ5Cri9HX7pgg,328
562
- invokeai/backend/patches/lora_conversions/z_image_lora_conversion_utils.py,sha256=H2EG2tN9SlE3U2Vj31KEYeW_ErzJRKURDxTSxMQq_H0,6505
556
+ invokeai/backend/patches/lora_conversions/z_image_lora_conversion_utils.py,sha256=KCbHJ1zjAjmRbK57alWULKm5JP40ckIhoy8QeSk8urg,5489
563
557
  invokeai/backend/quantization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
564
558
  invokeai/backend/quantization/bnb_llm_int8.py,sha256=nMnp9zoKhs8q_pWOwFTPJstzcXW3EcvUgFPEqb2tiDI,7052
565
559
  invokeai/backend/quantization/bnb_nf4.py,sha256=r_Y97abXJh18cUcVkieKu-3WTfvARQ4YGvC9PxqQGoo,7486
566
- invokeai/backend/quantization/gguf/ggml_tensor.py,sha256=HxRLIawp_O8VMqUPE7jQqzfhjNqCxqgZZW4wnSKQS94,8266
560
+ invokeai/backend/quantization/gguf/ggml_tensor.py,sha256=L4m_1zXHkBviOdnpXJRWU2V2-ebvUbkr3Kjoh6xbtxs,7595
567
561
  invokeai/backend/quantization/gguf/loaders.py,sha256=YCsTIJ3pOK1Z8a3q-PiFTwIKoRJQnHr12bF_Pq_gk3c,1969
568
562
  invokeai/backend/quantization/gguf/utils.py,sha256=_QZ1bigJ8VlLFq01Jg5Lb3V2oahCBmkiPwaX-6VMamw,10472
569
563
  invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py,sha256=fbfDlbgLUmio6Av9dy622sZhCy4UJvgiboCQrBkJUqk,3660
@@ -626,7 +620,7 @@ invokeai/backend/z_image/z_image_controlnet_extension.py,sha256=4KV0wu0S6UsJjc2u
626
620
  invokeai/backend/z_image/z_image_patchify_utils.py,sha256=fP6TIvsx3ClBTaca9cOXwMgvZfMrsCrAlZH4BuQ78J4,4810
627
621
  invokeai/backend/z_image/z_image_transformer_patch.py,sha256=yqDEeCP6X_MI_e6yqsOnDgveMUqAQiYkzEx8P5U01lY,10516
628
622
  invokeai/backend/z_image/extensions/__init__.py,sha256=eGnEmURiqVvISyZY0RmD3VtvTcM3mZaXT9P8uOrv6RM,21
629
- invokeai/backend/z_image/extensions/regional_prompting_extension.py,sha256=Pws63Rl_eCihAfpp-_EI6uk_0bl8icsX14f08sGiM5M,8762
623
+ invokeai/backend/z_image/extensions/regional_prompting_extension.py,sha256=A6p1qZ-Zna2CsEyB0gjFcwYNvtmBj6kNiVhiBprmAgc,8795
630
624
  invokeai/configs/controlnet/cldm_v15.yaml,sha256=OeEzjEFDYYrbF2BPlsOj90DBq10VV9cbBE8DB6CmrbQ,1949
631
625
  invokeai/configs/controlnet/cldm_v21.yaml,sha256=BTXRzGMA3RnP0-ve5pezqjhPmWXlc-gFBr6ii8uvRVk,2216
632
626
  invokeai/configs/stable-diffusion/sd_xl_base.yaml,sha256=gYc6UzyIwLq0Vixezlk5ym0BGlDcPVRRYlXEPHuxbWc,3246
@@ -649,12 +643,12 @@ invokeai/frontend/cli/arg_parser.py,sha256=6_y2ZB0iDmMvy2b_MJRWSdlqW1hToptmOoFl0
649
643
  invokeai/frontend/install/__init__.py,sha256=aAlRfyPocCx3Bm7NRB2I3Y_88Ug5omU4M5thFiNaw1A,57
650
644
  invokeai/frontend/install/import_images.py,sha256=wddGRowy3HQGEnRHTGAkOA1fhzPz9qT4IbGoZqX9xlc,34417
651
645
  invokeai/frontend/web/__init__.py,sha256=H4ye3hqaqVMEuzTnUNGeYBSHDhYLDcUWzS1XOr1zUWw,54
652
- invokeai/frontend/web/dist/index.html,sha256=O5SMRoG7FhigZqK1EU1gmbrhXHrIK0klfjpozMPj5QU,716
653
- invokeai/frontend/web/dist/assets/App-BBELGD-n.js,sha256=gi_6axurpTIOFjqjba21wZiCLXNT5WMQmiGbS5dv0fo,2478891
646
+ invokeai/frontend/web/dist/index.html,sha256=eq6roreeYqDlW0lbgXHqbP911fj49e4dUXpemdiCpqg,716
654
647
  invokeai/frontend/web/dist/assets/App-BpMMXEFQ.css,sha256=likhIM6eQSX2T8VxT2Kip3oyIgSBXsiPuUJjGveEOOc,66521
655
- invokeai/frontend/web/dist/assets/browser-ponyfill-4xPFTMT3.js,sha256=d40jzDUAdEsqizv52jEq2eNdsgvHCRdJEvQXyaw3Zfw,10291
648
+ invokeai/frontend/web/dist/assets/App-CYhlZO3Q.js,sha256=OIrg3fasBt8yCQOq5X8qoGmtIycBCxUxCVa8ggVNlh0,2475953
649
+ invokeai/frontend/web/dist/assets/browser-ponyfill-DHZxq1nk.js,sha256=-EFYKQSzmG-6DZX8Ux_vzUZBq2JGc5TbKsIlpYpb4S4,10291
656
650
  invokeai/frontend/web/dist/assets/denoising-strength-4mznP2tj.png,sha256=gg-GjcPP0hcwpwj1_qqdjiJDnwrUF21IGFT8p89NLvc,916615
657
- invokeai/frontend/web/dist/assets/index-vCDSQboA.js,sha256=BxNtdZUlwPgULJa6Rf9gAeQHmsPagsVDZhbXv5F8zQQ,1705845
651
+ invokeai/frontend/web/dist/assets/index-dgSJAY--.js,sha256=c9Sh9XAqEuHfj2_GWv97vUhYpZ7PuB1w2CzExsErS_g,1704145
658
652
  invokeai/frontend/web/dist/assets/inter-cyrillic-ext-wght-normal-B2xhLi22.woff2,sha256=wpY9x2Wg6ArPJplunuyuJEXzN4PVPGFdwBSovnKqwXA,25888
659
653
  invokeai/frontend/web/dist/assets/inter-cyrillic-wght-normal-CMZtQduZ.woff2,sha256=YswB2u9yw-p2olhEU2jS9KuNBakfkcU_0S98QuMyWUI,18740
660
654
  invokeai/frontend/web/dist/assets/inter-greek-ext-wght-normal-CGAr0uHJ.woff2,sha256=akF8_kBanIieL9Y6ZyibhaaIaBwPpk8SFedKwUcaNv4,11200
@@ -690,8 +684,7 @@ invokeai/frontend/web/dist/locales/ar.json,sha256=mG_5DFNSHdxGZnLYrrQsBUAXQcbWK8
690
684
  invokeai/frontend/web/dist/locales/az.json,sha256=gssN5Q1lHAHJQj4-qvjg6dP3Sh-ZPNb6pb6FLE7xTxk,62
691
685
  invokeai/frontend/web/dist/locales/bg.json,sha256=BlR-CNtSfdYBxKpPNry8czGAHryF1kqC64CEft-nk7w,2797
692
686
  invokeai/frontend/web/dist/locales/de.json,sha256=TL0pRnw3nSWj_8iP9a1nm0eSfXZNlGu4rKey-DrIA7Y,78380
693
- invokeai/frontend/web/dist/locales/en-GB.json,sha256=yj0WO6sFU4GCciYUBWjzvvfqrBh869doeOC2Pp5EI1Y,3
694
- invokeai/frontend/web/dist/locales/en.json,sha256=zRApMg8ygAYIJGU7Gs5B4-IQVb6UWp0kekuLegS6O5E,142697
687
+ invokeai/frontend/web/dist/locales/en.json,sha256=-H396lA0xs2vxsZ_fFTX1-A5Tx2PZ3kSoERN_4pgPh4,142411
695
688
  invokeai/frontend/web/dist/locales/es.json,sha256=_3J-LGH6YZqhBfeClQ4k714BFCmNb9Nex2R8KWOQgIM,39819
696
689
  invokeai/frontend/web/dist/locales/fi.json,sha256=ZxJBUVFCA78eIZtlOozR7lvNmFDs4DiIDA3HEkiMGcY,1055
697
690
  invokeai/frontend/web/dist/locales/fr.json,sha256=J436OiJflkyrEVTpCk6C4zoWz1M1RKGhcve9qU-T50w,120939
@@ -717,12 +710,12 @@ invokeai/frontend/web/scripts/clean_translations.py,sha256=Hj43SbGSCT2B_9jRohUVv
717
710
  invokeai/frontend/web/static/docs/invoke-favicon-docs.svg,sha256=2DfWaNhjgplnQK8V79yXjQ34JDCk94ZPH29CiTk90y8,779
718
711
  invokeai/invocation_api/__init__.py,sha256=5_qthg3i_5VmptjcjeIq0mytlMD9P_gzeaGqeLNG1YY,6351
719
712
  invokeai/version/__init__.py,sha256=22G4uBmmfLuldDclGKINkVS-rgPfNFQUJH_ag8m_OcM,529
720
- invokeai/version/invokeai_version.py,sha256=JBby69U_szIzmtcZ6Vacle7cAVe7nhVz7gIF0NB3Fqo,23
721
- invokeai-6.10.0.dist-info/licenses/LICENSE,sha256=OTAVifLLp_QCl8faBcfl0WT3BMqpeWZbEXw1MgPJzwM,10146
722
- invokeai-6.10.0.dist-info/licenses/LICENSE-SD1+SD2.txt,sha256=GbsztR5dRQiKsmF33NYkGL-GFJhSc2oG9gQPwzo-7Tg,14555
723
- invokeai-6.10.0.dist-info/licenses/LICENSE-SDXL.txt,sha256=JV_wPIqRG79iLCB8HrwY1J8jBimjUblcUxfnDbUMaqs,14213
724
- invokeai-6.10.0.dist-info/METADATA,sha256=iNFeL482-KztspuMcqo4ZO4liFBzv49_iuzZ168YgKs,22865
725
- invokeai-6.10.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
726
- invokeai-6.10.0.dist-info/entry_points.txt,sha256=MyuHYE_-ewdP-nVIcsITkhCNq7q0HNkavfd49TfNYho,62
727
- invokeai-6.10.0.dist-info/top_level.txt,sha256=69I61eoKb6LXiuo6Zz2l_6lhY1eRGBCqP8i9SMqp9Gg,9
728
- invokeai-6.10.0.dist-info/RECORD,,
713
+ invokeai/version/invokeai_version.py,sha256=kFGSJ9Z-O2Tws8mLDjBq187WMy6A7mG2_tHz_FqYOoE,26
714
+ invokeai-6.10.0rc1.dist-info/licenses/LICENSE,sha256=OTAVifLLp_QCl8faBcfl0WT3BMqpeWZbEXw1MgPJzwM,10146
715
+ invokeai-6.10.0rc1.dist-info/licenses/LICENSE-SD1+SD2.txt,sha256=GbsztR5dRQiKsmF33NYkGL-GFJhSc2oG9gQPwzo-7Tg,14555
716
+ invokeai-6.10.0rc1.dist-info/licenses/LICENSE-SDXL.txt,sha256=JV_wPIqRG79iLCB8HrwY1J8jBimjUblcUxfnDbUMaqs,14213
717
+ invokeai-6.10.0rc1.dist-info/METADATA,sha256=ppO9uEJNvLvvuncxW-MfRy86-UudHfPYWl3mbgTyEQI,22893
718
+ invokeai-6.10.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
719
+ invokeai-6.10.0rc1.dist-info/entry_points.txt,sha256=MyuHYE_-ewdP-nVIcsITkhCNq7q0HNkavfd49TfNYho,62
720
+ invokeai-6.10.0rc1.dist-info/top_level.txt,sha256=69I61eoKb6LXiuo6Zz2l_6lhY1eRGBCqP8i9SMqp9Gg,9
721
+ invokeai-6.10.0rc1.dist-info/RECORD,,
@@ -1,59 +0,0 @@
1
- import pathlib
2
- from typing import Literal
3
-
4
- from invokeai.app.invocations.baseinvocation import BaseInvocation, BaseInvocationOutput, invocation, invocation_output
5
- from invokeai.app.invocations.fields import ImageField, InputField, OutputField, WithBoard, WithMetadata
6
- from invokeai.app.services.shared.invocation_context import InvocationContext
7
- from invokeai.backend.image_util.pbr_maps.architecture.pbr_rrdb_net import PBR_RRDB_Net
8
- from invokeai.backend.image_util.pbr_maps.pbr_maps import NORMAL_MAP_MODEL, OTHER_MAP_MODEL, PBRMapsGenerator
9
- from invokeai.backend.util.devices import TorchDevice
10
-
11
-
12
- @invocation_output("pbr_maps-output")
13
- class PBRMapsOutput(BaseInvocationOutput):
14
- normal_map: ImageField = OutputField(default=None, description="The generated normal map")
15
- roughness_map: ImageField = OutputField(default=None, description="The generated roughness map")
16
- displacement_map: ImageField = OutputField(default=None, description="The generated displacement map")
17
-
18
-
19
- @invocation("pbr_maps", title="PBR Maps", tags=["image", "material"], category="image", version="1.0.0")
20
- class PBRMapsInvocation(BaseInvocation, WithMetadata, WithBoard):
21
- """Generate Normal, Displacement and Roughness Map from a given image"""
22
-
23
- image: ImageField = InputField(description="Input image")
24
- tile_size: int = InputField(default=512, description="Tile size")
25
- border_mode: Literal["none", "seamless", "mirror", "replicate"] = InputField(
26
- default="none", description="Border mode to apply to eliminate any artifacts or seams"
27
- )
28
-
29
- def invoke(self, context: InvocationContext) -> PBRMapsOutput:
30
- image_pil = context.images.get_pil(self.image.image_name, mode="RGB")
31
-
32
- def loader(model_path: pathlib.Path):
33
- return PBRMapsGenerator.load_model(model_path, TorchDevice.choose_torch_device())
34
-
35
- torch_device = TorchDevice.choose_torch_device()
36
-
37
- with (
38
- context.models.load_remote_model(NORMAL_MAP_MODEL, loader) as normal_map_model,
39
- context.models.load_remote_model(OTHER_MAP_MODEL, loader) as other_map_model,
40
- ):
41
- assert isinstance(normal_map_model, PBR_RRDB_Net)
42
- assert isinstance(other_map_model, PBR_RRDB_Net)
43
- pbr_pipeline = PBRMapsGenerator(normal_map_model, other_map_model, torch_device)
44
- normal_map, roughness_map, displacement_map = pbr_pipeline.generate_maps(
45
- image_pil, self.tile_size, self.border_mode
46
- )
47
-
48
- normal_map = context.images.save(normal_map)
49
- normal_map_field = ImageField(image_name=normal_map.image_name)
50
-
51
- roughness_map = context.images.save(roughness_map)
52
- roughness_map_field = ImageField(image_name=roughness_map.image_name)
53
-
54
- displacement_map = context.images.save(displacement_map)
55
- displacement_map_field = ImageField(image_name=displacement_map.image_name)
56
-
57
- return PBRMapsOutput(
58
- normal_map=normal_map_field, roughness_map=roughness_map_field, displacement_map=displacement_map_field
59
- )
@@ -1,62 +0,0 @@
1
- """Flow Matching scheduler definitions and mapping.
2
-
3
- This module provides the scheduler types and mapping for Flow Matching models
4
- (Flux and Z-Image), supporting multiple schedulers from the diffusers library.
5
- """
6
-
7
- from typing import Literal, Type
8
-
9
- from diffusers import (
10
- FlowMatchEulerDiscreteScheduler,
11
- FlowMatchHeunDiscreteScheduler,
12
- )
13
- from diffusers.schedulers.scheduling_utils import SchedulerMixin
14
-
15
- # Note: FlowMatchLCMScheduler may not be available in all diffusers versions
16
- try:
17
- from diffusers import FlowMatchLCMScheduler
18
-
19
- _HAS_LCM = True
20
- except ImportError:
21
- _HAS_LCM = False
22
-
23
- # Scheduler name literal type for type checking
24
- FLUX_SCHEDULER_NAME_VALUES = Literal["euler", "heun", "lcm"]
25
-
26
- # Human-readable labels for the UI
27
- FLUX_SCHEDULER_LABELS: dict[str, str] = {
28
- "euler": "Euler",
29
- "heun": "Heun (2nd order)",
30
- "lcm": "LCM",
31
- }
32
-
33
- # Mapping from scheduler names to scheduler classes
34
- FLUX_SCHEDULER_MAP: dict[str, Type[SchedulerMixin]] = {
35
- "euler": FlowMatchEulerDiscreteScheduler,
36
- "heun": FlowMatchHeunDiscreteScheduler,
37
- }
38
-
39
- if _HAS_LCM:
40
- FLUX_SCHEDULER_MAP["lcm"] = FlowMatchLCMScheduler
41
-
42
-
43
- # Z-Image scheduler types (same schedulers as Flux, both use Flow Matching)
44
- # Note: Z-Image-Turbo is optimized for ~8 steps with Euler, but other schedulers
45
- # can be used for experimentation.
46
- ZIMAGE_SCHEDULER_NAME_VALUES = Literal["euler", "heun", "lcm"]
47
-
48
- # Human-readable labels for the UI
49
- ZIMAGE_SCHEDULER_LABELS: dict[str, str] = {
50
- "euler": "Euler",
51
- "heun": "Heun (2nd order)",
52
- "lcm": "LCM",
53
- }
54
-
55
- # Mapping from scheduler names to scheduler classes (same as Flux)
56
- ZIMAGE_SCHEDULER_MAP: dict[str, Type[SchedulerMixin]] = {
57
- "euler": FlowMatchEulerDiscreteScheduler,
58
- "heun": FlowMatchHeunDiscreteScheduler,
59
- }
60
-
61
- if _HAS_LCM:
62
- ZIMAGE_SCHEDULER_MAP["lcm"] = FlowMatchLCMScheduler