chmod +x /app/init.sh
This commit is contained in:
parent
ba5da461e1
commit
f56d2ad3f2
2 changed files with 3 additions and 2 deletions
|
|
@ -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"]
|
||||
|
|
|
|||
|
|
@ -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")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue