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

github/InvokeAI

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