Update src/diffusers/models/activations.py

williamberman · hypnopump · web-flow · commit 2169bc467fbe · 2023-06-05T10:30:20.000-07:00
Co-authored-by: Eric Alcaide &lt;ericalcaide1@gmail.com&gt;
diff --git a/src/diffusers/models/activations.py b/src/diffusers/models/activations.py
@@ -2,7 +2,7 @@
 
 
 def get_activation(act_fn):
-    if act_fn in ["swish", "silu"]:
+    if act_fn in {"swish", "silu"}:
         return nn.SiLU()
     elif act_fn == "mish":
         return nn.Mish()
