diff --git a/python/src/coreai_models/export/presets.py b/python/src/coreai_models/export/presets.py index bfca1312..85fed754 100644 --- a/python/src/coreai_models/export/presets.py +++ b/python/src/coreai_models/export/presets.py @@ -33,6 +33,7 @@ "coreai_models.primitives.macos.rope.RoPE": None, "coreai_models.primitives.macos.rms_norm.RMSNorm": None, "coreai_models.primitives.macos.rms_norm.RMSNormPlusOne": None, + "coreai_models.primitives.macos.rms_norm.RMSNormGated": None, } # Embedding modules excluded from iOS palettization.