extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-01-23 06:39:50 +01:00
Code Issues Packages Projects Releases Wiki Activity

Can feed a custom image

Browse Source
This commit is contained in:
cmdr2 2022-08-24 11:18:25 +05:30
parent 6c0c6c50e8
commit 6dac5341db
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server.sh
View File

@ -1,7 +1,12 @@
#!/bin/bash
IMAGE_NAME="r8.im/stability-ai/stable-diffusion@sha256:06eb78b36068500c616a7f33c15e6fa40404f8e14b5bfad57ebe0c7fe0f6bdf1"
IMAGE_NAME=$1
if [ -z "$IMAGE_NAME" ]; then
IMAGE_NAME="r8.im/stability-ai/stable-diffusion@sha256:06eb78b36068500c616a7f33c15e6fa40404f8e14b5bfad57ebe0c7fe0f6bdf1"
fi
echo "Using docker image $IMAGE_NAME"
docker run --name sd -d -p 5000:5000 --gpus all $IMAGE_NAME
uvicorn main:app --reload