fix(api): allow dash in network names

ssube · Mar 19, 2023 · 645da3d · 645da3d
1 parent b195b59
commit 645da3d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
@@ -9,8 +9,8 @@
 logger = getLogger(__name__)
 
 
-INVERSION_TOKEN = compile(r"\<inversion:(\w+):([\.|\d]+)\>")
-LORA_TOKEN = compile(r"\<lora:(\w+):([\.|\d]+)\>")
+INVERSION_TOKEN = compile(r"\<inversion:([-\w]+):([\.|\d]+)\>")
+LORA_TOKEN = compile(r"\<lora:([-\w]+):([\.|\d]+)\>")
 MAX_TOKENS_PER_GROUP = 77
 PATTERN_RANGE = compile(r"(\w+)-{(\d+),(\d+)(?:,(\d+))?}")