From 27bd10d2f7e29a6c4cc6eeb161afe386ab1cf198 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 19 Mar 2023 09:13:40 -0500
Subject: [PATCH] fix(api): restrict CLIP skip token to integers, allow
 negative LoRA and Inversion weights

---
 api/onnx_web/diffusers/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
index c97ea65e..7c0cb2a5 100644
--- a/api/onnx_web/diffusers/utils.py
+++ b/api/onnx_web/diffusers/utils.py
@@ -10,9 +10,9 @@ from diffusers import OnnxStableDiffusionPipeline
 logger = getLogger(__name__)
 
 
-CLIP_TOKEN = compile(r"\<clip:([-\w]+):([\.|\d]+)\>")
-INVERSION_TOKEN = compile(r"\<inversion:([-\w]+):([\.|\d]+)\>")
-LORA_TOKEN = compile(r"\<lora:([-\w]+):([\.|\d]+)\>")
+CLIP_TOKEN = compile(r"\<clip:([-\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+))?}")