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")