From f56d2ad3f21bc896eafd6f36b18bb763d84853be Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Niklas=20M=C3=BCller?= Date: Thu, 14 Mar 2024 11:48:31 +0100 Subject: [PATCH] chmod +x /app/init.sh --- tti-provider/Dockerfile | 1 + tti-provider/runner.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tti-provider/Dockerfile b/tti-provider/Dockerfile index 64e1a4a..2a0cb61 100644 --- a/tti-provider/Dockerfile +++ b/tti-provider/Dockerfile @@ -37,6 +37,7 @@ RUN touch /var/log/uvicorn.log # copy scripts to folder COPY ./*.py /app COPY ./init.sh /app +RUN chmod +x /app/init.sh COPY ./styles.json /app CMD ["bash", "init.sh"] diff --git a/tti-provider/runner.py b/tti-provider/runner.py index b83e372..02dce96 100644 --- a/tti-provider/runner.py +++ b/tti-provider/runner.py @@ -79,10 +79,10 @@ try: # prompt = "a red------- cat playing with a ball" base = DiffusionPipeline.from_pretrained( - "stablediffusionapi/juggernaut-xl-v8" + "RunDiffusion/Juggernaut-XL-v8" ).to("cpu") refiner = DiffusionPipeline.from_pretrained( - "stablediffusionapi/juggernaut-xl-v8", + "RunDiffusion/Juggernaut-XL-v8", text_encoder_2=base.text_encoder_2, vae=base.vae, ).to("cpu")