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.
- invokeai/app/invocations/flux_denoise.py +1 -15
- invokeai/app/invocations/metadata_linked.py +0 -47
- invokeai/app/invocations/z_image_denoise.py +84 -244
- invokeai/app/services/config/config_default.py +1 -3
- invokeai/app/services/model_manager/model_manager_default.py +0 -7
- invokeai/backend/flux/denoise.py +11 -196
- invokeai/backend/model_manager/configs/lora.py +0 -36
- invokeai/backend/model_manager/load/model_cache/model_cache.py +2 -104
- invokeai/backend/model_manager/load/model_loaders/cogview4.py +1 -2
- invokeai/backend/model_manager/load/model_loaders/flux.py +6 -13
- invokeai/backend/model_manager/load/model_loaders/generic_diffusers.py +2 -4
- invokeai/backend/model_manager/load/model_loaders/onnx.py +0 -1
- invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py +1 -2
- invokeai/backend/model_manager/load/model_loaders/z_image.py +3 -37
- invokeai/backend/model_manager/starter_models.py +4 -13
- invokeai/backend/patches/lora_conversions/z_image_lora_conversion_utils.py +5 -39
- invokeai/backend/quantization/gguf/ggml_tensor.py +4 -15
- invokeai/backend/z_image/extensions/regional_prompting_extension.py +12 -10
- invokeai/frontend/web/dist/assets/App-CYhlZO3Q.js +161 -0
- invokeai/frontend/web/dist/assets/{browser-ponyfill-4xPFTMT3.js → browser-ponyfill-DHZxq1nk.js} +1 -1
- invokeai/frontend/web/dist/assets/{index-vCDSQboA.js → index-dgSJAY--.js} +51 -51
- invokeai/frontend/web/dist/index.html +1 -1
- invokeai/frontend/web/dist/locales/en.json +5 -11
- invokeai/version/invokeai_version.py +1 -1
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/METADATA +2 -2
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/RECORD +32 -39
- invokeai/app/invocations/pbr_maps.py +0 -59
- invokeai/backend/flux/schedulers.py +0 -62
- invokeai/backend/image_util/pbr_maps/architecture/block.py +0 -367
- invokeai/backend/image_util/pbr_maps/architecture/pbr_rrdb_net.py +0 -70
- invokeai/backend/image_util/pbr_maps/pbr_maps.py +0 -141
- invokeai/backend/image_util/pbr_maps/utils/image_ops.py +0 -93
- invokeai/frontend/web/dist/assets/App-BBELGD-n.js +0 -161
- invokeai/frontend/web/dist/locales/en-GB.json +0 -1
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/WHEEL +0 -0
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/entry_points.txt +0 -0
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/licenses/LICENSE +0 -0
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/licenses/LICENSE-SD1+SD2.txt +0 -0
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/licenses/LICENSE-SDXL.txt +0 -0
- {invokeai-6.10.0.dist-info → invokeai-6.10.0rc1.dist-info}/top_level.txt +0 -0
|
@@ -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
|
|
1051
|
+
"zImageVaePlaceholder": "From Qwen3 Source (leave empty)",
|
|
1053
1052
|
"zImageQwen3Encoder": "Qwen3 Encoder (optional)",
|
|
1054
|
-
"zImageQwen3EncoderPlaceholder": "From Qwen3
|
|
1055
|
-
"zImageQwen3Source": "Qwen3
|
|
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
|
-
"
|
|
2825
|
-
"
|
|
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.
|
|
1
|
+
__version__ = "6.10.0rc1"
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: InvokeAI
|
|
3
|
-
Version: 6.10.
|
|
4
|
-
Summary:
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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
|
|
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=
|
|
497
|
-
invokeai/backend/model_manager/load/model_loaders/generic_diffusers.py,sha256=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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=
|
|
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/
|
|
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-
|
|
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
|
|
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=
|
|
721
|
-
invokeai-6.10.
|
|
722
|
-
invokeai-6.10.
|
|
723
|
-
invokeai-6.10.
|
|
724
|
-
invokeai-6.10.
|
|
725
|
-
invokeai-6.10.
|
|
726
|
-
invokeai-6.10.
|
|
727
|
-
invokeai-6.10.
|
|
728
|
-
invokeai-6.10.
|
|
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
|