apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): allow dash in network names

This commit is contained in:
Sean Sube 2023-03-18 21:48:20 -05:00
parent b195b59301
commit 645da3d1a6
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/utils.py
View File

@ -9,8 +9,8 @@ from diffusers import OnnxStableDiffusionPipeline
logger = getLogger(__name__) logger = getLogger(__name__)
INVERSION_TOKEN = compile(r"\<inversion:(\w+):([\.|\d]+)\>") INVERSION_TOKEN = compile(r"\<inversion:([-\w]+):([\.|\d]+)\>")
LORA_TOKEN = compile(r"\<lora:(\w+):([\.|\d]+)\>") LORA_TOKEN = compile(r"\<lora:([-\w]+):([\.|\d]+)\>")
MAX_TOKENS_PER_GROUP = 77 MAX_TOKENS_PER_GROUP = 77
PATTERN_RANGE = compile(r"(\w+)-{(\d+),(\d+)(?:,(\d+))?}") PATTERN_RANGE = compile(r"(\w+)-{(\d+),(\d+)(?:,(\d+))?}")