Add back missed deleted GELU_QUICK type. (#47)

google-ai-edge · Jun 11, 2024 · c84db1c · c84db1c
1 parent 913831c
commit c84db1c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ai_edge_torch/generative/layers/model_config.py b/ai_edge_torch/generative/layers/model_config.py
@@ -27,6 +27,7 @@ class ActivationType(enum.Enum):
   SILU = enum.auto()
   GELU = enum.auto()
   GELU_TANH = enum.auto()
+  GELU_QUICK = enum.auto()
   GE_GLU = enum.auto()
   RELU = enum.auto()