psychedelicious fd4c3bd27a refactor: estimate working vae memory during encode/decode ...

- Move the estimation logic to utility functions
- Estimate memory _within_ the encode and decode methods, ensuring we
_always_ estimate working memory when running a VAE

2025-08-18 21:43:14 +10:00

..

custom_nodes

feat(app): less janky custom node loading

2025-03-08 09:42:13 +11:00

__init__.py

feat(app): improved custom load loading ordering

2025-02-27 12:39:37 +11:00

baseinvocation.py

fix(nodes): ensure each invocation overrides _original_model_fields with own field data

2025-06-20 15:03:55 +10:00

batch.py

fix(nodes): do not provide invalid defaults for batch nodes

2025-05-14 06:55:15 +10:00

blend_latents.py

chore: ruff

2024-11-19 07:02:37 -08:00

canny.py

feat(nodes): add CannyEdgeDetectionInvocation

2024-09-11 08:12:48 -04:00

cogview4_denoise.py

fix(app): import CogView4Transformer2DModel from the module that exports it

2025-04-10 10:50:13 +10:00

cogview4_image_to_latents.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

cogview4_latents_to_image.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

cogview4_model_loader.py

feat(nodes): rename CogView4 nodes to match naming format

2025-04-10 10:50:13 +10:00

cogview4_text_encoder.py

feat(nodes): rename CogView4 nodes to match naming format

2025-04-10 10:50:13 +10:00

collections.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

color_map.py

feat(nodes): update color map node

2024-09-11 08:12:48 -04:00

compel.py

fix(app): reduce peak memory usage

2025-06-11 12:56:16 +10:00

composition-nodes.py

fix(nodes): fix spots where default of None was provided for non-optional fields

2025-05-20 10:47:02 +10:00

constants.py

fix(app): remove obsolete DEFAULT_PRECISION variable

2025-01-06 11:14:58 +11:00

content_shuffle.py

feat(nodes): update content shuffle node

2024-09-11 08:12:48 -04:00

controlnet.py

feat(nodes): much faster heuristic resize utility

2025-05-29 13:49:07 +10:00

create_denoise_mask.py

fix(app): fixed InputField default values

2024-12-20 09:30:56 +11:00

create_gradient_mask.py

Fix: inpaint model mask using wrong tensor name

2025-06-05 11:31:35 -04:00

crop_latents.py

Move CropLatentsCoreInvocation to its own file. No functional changes.

2024-06-07 09:28:42 -04:00

cv.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

denoise_latents.py

ruff fixes

2025-05-19 13:50:04 +10:00

depth_anything.py

feat(nodes): add DepthAnythingDepthEstimationInvocation

2024-09-11 08:12:48 -04:00

dw_openpose.py

tidy(nodes): remove unused old dw openpose detector class

2025-04-04 18:42:13 +11:00

facetools.py

chore: bump nodes versions

2024-03-20 10:28:07 +11:00

fields.py

Add FLUX Kontext conditioning support for reference images

2025-06-27 18:27:46 +10:00

flux_control_lora_loader.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

flux_controlnet.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

flux_denoise.py

feat(mm): prepare kontext latents before loading transformer

2025-08-12 10:51:05 +10:00

flux_fill.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

flux_ip_adapter.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

flux_kontext.py

Add FLUX Kontext conditioning support for reference images

2025-06-27 18:27:46 +10:00

flux_lora_loader.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

flux_model_loader.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

flux_redux.py

feat(mm): siglip model loading supports partial loading

2025-04-18 10:12:03 +10:00

flux_text_encoder.py

use "\u2581" instead of the character itself for clarity

2025-06-30 10:40:31 +10:00

flux_vae_decode.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

flux_vae_encode.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

grounding_dino.py

Expose all model options in the GroundingDinoInvocation and the SegmentAnythingInvocation.

2024-08-01 14:23:32 -04:00

hed.py

feat(nodes): add HEDEdgeDetectionInvocation

2024-09-11 08:12:48 -04:00

ideal_size.py

fix(nodes): fix spots where default of None was provided for non-optional fields

2025-05-20 10:47:02 +10:00

image_panels.py

Add utility ImagePanelLayoutInvocation for working with In-Context LoRA workflows.

2024-11-26 20:58:31 -08:00

image_to_latents.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

image.py

feat(nodes): add node to prep images for FLUX Kontext

2025-08-04 23:03:12 +10:00

infill.py

feat(nodes): extract LaMa model url to constant

2025-04-15 07:13:25 +10:00

ip_adapter.py

ruff fix

2025-05-19 13:50:04 +10:00

latents_to_image.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

lineart_anime.py

feat(nodes): add LineartAnimeEdgeDetectionInvocation

2024-09-11 08:12:48 -04:00

lineart.py

feat(nodes): add LineartEdgeDetectionInvocation

2024-09-11 08:12:48 -04:00

llava_onevision_vllm.py

feat(mm): llava model loading supports partial loading; fix OOM crash on initial load

2025-04-18 10:12:03 +10:00

load_custom_nodes.py

feat(app): less janky custom node loading

2025-03-08 09:42:13 +11:00

mask.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

math.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

mediapipe_face.py

feat(nodes): add MediaPipeFaceDetectionInvocation

2024-09-11 08:12:48 -04:00

metadata_linked.py

Update metadata_linked.py

2025-04-16 06:28:22 +10:00

metadata.py

add new method types to metadata

2025-05-22 14:09:10 -04:00

mlsd.py

tidy(nodes): MLSDEdgeDetection -> MLSDDetection

2024-09-11 08:12:48 -04:00

model.py

Add CogView4ModelLoaderInvocation. (Not wired up with frontend yet.)

2025-04-10 10:50:13 +10:00

noise.py

chore(nodes): update titles of all model-specific nodes to reference their models

2025-03-17 10:32:19 +11:00

normal_bae.py

feat(nodes): add NormalMapInvocation

2024-09-11 08:12:48 -04:00

param_easing.py

remove GPL-3 licensed package easing-functions

2024-11-18 08:55:17 -05:00

pidi.py

feat(nodes): update pidinet node

2024-09-11 08:12:48 -04:00

primitives.py

Update primitives.py

2025-07-17 06:33:56 +10:00

prompt.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

resize_latents.py

Move ScaleLatentsInvocation and ResizeLatentsInvocation to their own file. No functional changes.

2024-06-07 09:28:42 -04:00

scheduler.py

Fix static type errors with SCHEDULER_NAME_VALUES. And, avoid bi-directional cross-directory imports, which contribute to circular import issues.

2024-07-05 07:38:35 -07:00

sd3_denoise.py

Consolidate InpaintExtension implementations for SD3 and FLUX.

2025-04-10 10:50:13 +10:00

sd3_image_to_latents.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

sd3_latents_to_image.py

refactor: estimate working vae memory during encode/decode

2025-08-18 21:43:14 +10:00

sd3_model_loader.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

sd3_text_encoder.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

sdxl.py

Imports

2025-03-26 12:55:10 +11:00

segment_anything.py

fix(nodes): transformers bug with SAM

2025-05-22 11:32:37 +10:00

spandrel_image_to_image.py

We should not trust the value of since the model could be partially-loaded.

2025-01-07 19:22:31 -05:00

strings.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

t2i_adapter.py

chore(nodes): update titles of all model-specific nodes to reference their models

2025-03-17 10:32:19 +11:00

tiled_multi_diffusion_denoise_latents.py

tidy(nodes): rename controlnet_image_processors.py -> controlnet.py

2025-04-04 18:42:13 +11:00

tiles.py

chore(nodes): remove beta/prototype flag from a lot of stable nodes

2025-03-27 08:08:44 +11:00

upscale.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

util.py

Update util.py

2025-03-05 14:39:17 +11:00