InvokeAI/invokeai/backend/flux/modules/layers.py at ce286363d0e1639b97dbd76ec00cf00c85bb2e4e - InvokeAI - AtHeartEngineering

github/InvokeAI

mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-03 08:25:00 -05:00

Files

Ryan Dick db9c0cad7c Replace custom RMSNorm implementation with torch.nn.functional.rms_norm(...) for improved speed.

2024-11-29 12:32:50 -05:00

9.1 KiB

Raw Blame History

View Raw