diff --git a/logos/creation_interfaces/kojii_chebel.py b/logos/creation_interfaces/kojii_chebel.py index 9b9eb11..096cead 100644 --- 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: @@ -42,7 +42,7 @@ def kojii_chebel(request: KojiiChebelRequest, callback=None): 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 diff --git a/logos/creation_interfaces/kojii_untitledxyz.py b/logos/creation_interfaces/kojii_untitledxyz.py index 7330871..07f9123 100644 --- a/logos/creation_interfaces/kojii_untitledxyz.py +++ b/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, } diff --git a/logos/creation_interfaces/kojii_violetforest.py b/logos/creation_interfaces/kojii_violetforest.py index e8a8988..a91ab52 100644 --- a/logos/creation_interfaces/kojii_violetforest.py +++ b/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") diff --git a/logos/plugins/replicate.py b/logos/plugins/replicate.py index fc088b8..a30105f 100644 --- a/logos/plugins/replicate.py +++ b/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)