chmod +x /app/init.sh

This commit is contained in:
Niklas Müller 2024-03-14 11:48:31 +01:00
parent ba5da461e1
commit f56d2ad3f2
2 changed files with 3 additions and 2 deletions

View file

@ -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"]

View file

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