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

github/InvokeAI

mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04: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