update stage

logos/creation_interfaces/kojii_chebel.py

@@ -34,15 +34,15 @@ 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:
         prompt = "oil painting, soft pastel colors, skin tones, woman, dance, expressive brushstrokes"
         negative_prompt = "colorful, yellow, blue, watermark, text, tiling, out of frame, blurry, blurred, grainy, signature, cut off, draft"
     else:
         prompt = "oil painting, black and white, woman, dance, expressive brushstrokes, charcoal drawing"
-        negative_prompt = "color, colors, red, pink, purple, green, orange, yellow, blue,  colorful, watermark, text, tiling, out of frame, blurry, blurred, grainy, signature, cut off, draft"
+        negative_prompt = "color, colors, red, pink, purple, green, orange, yellow, blue, colorful, watermark, text, tiling, out of frame, blurry, blurred, grainy, signature, cut off, draft"
 
     # Hardcoded to 1024x1024 as per Chebel's request (aspect ratio of init_img will still get adopted)
     w, h = 1024, 1024

logos/creation_interfaces/kojii_untitledxyz.py

@@ -121,7 +121,9 @@ def kojii_untitledxyz(request: KojiiUntitledxyzRequest, callback=None):
         "text_inputs_to_interpolate_weights": "|".join(
             [str(t) for t in text_inputs_to_interpolate_weights],
         ),
-        "lora_scale": 0.8,
+        "lora": "https://edenartlab-prod-data.s3.us-east-1.amazonaws.com/d2e6d1f8ccfca428ba42fa56a0384a4261d32bf1ee8b0dc952d99da9011daf39.tar",
+        "lora_scale": 0.8, 
+        "uc_text": "watermark, text, nude, naked, nsfw, poorly drawn face, ugly, tiling, out of frame, blurry, blurred, grainy, signature, cut off, draft, person, car, sky, clouds, city",
         "seed": seed,
         "guidance_scale": 8.0,
     }

logos/creation_interfaces/kojii_violetforest.py

@@ -28,7 +28,7 @@ class KojiiVioletforestRequest(BaseModel):
     seed: Optional[int] = Field(default=None, description="Random seed")
 
 
-def kojii_violetforest(request: KojiiVioletforestRequest, callback=None):
+def kojii_violetforest2(request: KojiiVioletforestRequest, callback=None):
     import requests
     from ..creation_interfaces import KojiiVioletforestRequest
 
@@ -44,7 +44,7 @@ def kojii_violetforest(request: KojiiVioletforestRequest, callback=None):
                 f.write(requests.get(image_url).content)
 
 
-def kojii_violetforest1(request: KojiiVioletforestRequest, callback=None):
+def kojii_violetforest(request: KojiiVioletforestRequest, callback=None):
     seed = request.seed if request.seed else random.randint(0, 1000000)
 
     if request.style == Style.Kawaii:
@@ -73,8 +73,8 @@ def kojii_violetforest1(request: KojiiVioletforestRequest, callback=None):
         "text_inputs_to_interpolate_weights": " | ".join(
             [str(t) for t in text_inputs_to_interpolate_weights],
         ),
-        # "lora": "https://edenartlab-prod-data.s3.us-east-1.amazonaws.com/e3b036c0a9949de0a5433cb6c7e54b540c47535ce7ae252948177304542ca4da.tar",
-        # "lora_scale": 0.7,
+        "lora": "https://edenartlab-prod-data.s3.us-east-1.amazonaws.com/e3b036c0a9949de0a5433cb6c7e54b540c47535ce7ae252948177304542ca4da.tar",
+        "lora_scale": 0.7,
         "seed": seed,
     }
     print("CONFIG")

logos/plugins/replicate.py

@@ -152,7 +152,7 @@ def audiocraft(
         "text_input": prompt,
         "duration_seconds": seconds,
     }
-
+    print(config)
     output = run_task(config, model_name="edenartlab/audiocraft")
 
     output = list(output)