cap lora_scale value between 0 and 1

edenartlab · Apr 15, 2024 · be2651d · be2651d
1 parent ac239a4
commit be2651d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/logos/creation_interfaces/kojii_chebel.py b/logos/creation_interfaces/kojii_chebel.py
@@ -34,7 +34,7 @@ class KojiiChebelRequest(BaseModel):
 
 
 def kojii_chebel(request: KojiiChebelRequest, callback=None):
-    lora_scale = 0.44 + 0.52 * request.abstract
+    lora_scale = max(0.0, min(1.0, 0.44 + 0.52 * request.abstract))
     seed = request.seed if request.seed else random.randint(0, 1000000)
 
     if request.color == ColorType.color: