InvokeAI/invokeai/backend/flux/modules/autoencoder.py at 6eafe53b2d46acf255ff86312ef5be5ffd7fe8d2 - InvokeAI - AtHeartEngineering

github/InvokeAI

mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-15 04:05:28 -05:00

Files

Ryan Dick a808ce81fd Replace swish() with torch.nn.functional.silu(h). They are functionally equivalent, but in my test VAE deconding was ~8% faster after the change.

2024-08-26 20:17:50 -04:00

10 KiB

Raw Blame History

View Raw