Angel Tsvetkov 1e8a9cef85

Update model.py ...

Why print twice the same VAE?

Before:
Using pytorch attention in VAE
Using pytorch attention in VAE

After:
Using pytorch attention in VAE encoder (1/2) - Model: ae.safetensors
Using pytorch attention in VAE decoder (2/2) - Model: ae.safetensors

2025-08-02 16:38:43 +03:00

..

ace

Fix torch warning about deprecated function. (#8075)

2025-05-12 14:32:01 -04:00

audio

Fix torch warning about deprecated function. (#8075)

2025-05-12 14:32:01 -04:00

aura

Replace print with logging (#6138)

2024-12-20 16:24:55 -05:00

cascade

Fix stable cascade VAE on some lowvram machines.

2025-03-08 20:24:04 -05:00

chroma

Un-hardcode chroma patch_size (#8840)

2025-07-08 15:56:59 -04:00

cosmos

Replace manual x * sigmoid(x) with torch silu in VAE nonlinearity (#9057)

2025-07-30 19:25:56 -04:00

flux

Support Flux Kontext Dev model. (#8679)

2025-06-26 11:28:41 -04:00

genmo

Switch mochi and wan modes to use pytorch RMSNorm. (#7925)

2025-05-03 19:07:55 -04:00

hidream

Support HiDream E1 model. (#7857)

2025-04-28 15:01:15 -04:00

hunyuan3d

Fix orientation of hunyuan 3d model.

2025-03-19 19:55:24 -04:00

hunyuan_video

Hunyuan Custom initial untested implementation. (#8101)

2025-05-13 15:53:47 -04:00

hydit

Change cosmos and hydit models to use the native RMSNorm. (#7934)

2025-05-04 06:26:20 -04:00

lightricks

LTXV VAE decoder: switch default padding mode (#8930)

2025-07-16 13:54:38 -04:00

lumina

Change lumina to native RMSNorm. (#7935)

2025-05-04 06:39:23 -04:00

models

Don't add tiny bit of random noise when VAE encoding. (#8705)

2025-06-27 14:14:56 -04:00

modules

Update model.py

2025-08-02 16:38:43 +03:00

omnigen

Omnigen2 model implementation. (#8669)

2025-06-25 19:35:57 -04:00

pixart

Remove windows line endings. (#8866)

2025-07-11 02:37:51 -04:00

wan

Tiny wan vae optimizations. (#9136)

2025-08-01 05:25:38 -04:00

common_dit.py

add RMSNorm to comfy.ops

2025-04-14 18:00:33 -04:00

util.py

Fix and enforce new lines at the end of files.

2024-12-30 04:14:59 -05:00