tidoni/tti-provider
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

chmod +x /app/init.sh

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

1
tti-provider/Dockerfile
View file

@ -37,6 +37,7 @@ RUN touch /var/log/uvicorn.log
# copy scripts to folder # copy scripts to folder
COPY ./*.py /app COPY ./*.py /app
COPY ./init.sh /app COPY ./init.sh /app
RUN chmod +x /app/init.sh
COPY ./styles.json /app COPY ./styles.json /app
CMD ["bash", "init.sh"] CMD ["bash", "init.sh"]

4
tti-provider/runner.py
View file

@ -79,10 +79,10 @@ try:
# prompt = "a red------- cat playing with a ball" # prompt = "a red------- cat playing with a ball"
base = DiffusionPipeline.from_pretrained( base = DiffusionPipeline.from_pretrained(
"stablediffusionapi/juggernaut-xl-v8" "RunDiffusion/Juggernaut-XL-v8"
).to("cpu") ).to("cpu")
refiner = DiffusionPipeline.from_pretrained( refiner = DiffusionPipeline.from_pretrained(
"stablediffusionapi/juggernaut-xl-v8", "RunDiffusion/Juggernaut-XL-v8",
text_encoder_2=base.text_encoder_2, text_encoder_2=base.text_encoder_2,
vae=base.vae, vae=base.vae,
).to("cpu") ).to("cpu")