easydiffusion/server.sh

16 lines
374 B
Bash
Raw Normal View History

2022-08-23 22:28:18 +02:00
#!/bin/bash
2022-08-24 07:48:25 +02:00
IMAGE_NAME=$1
2022-08-23 22:28:18 +02:00
2022-08-24 07:48:25 +02:00
if [ -z "$IMAGE_NAME" ]; then
IMAGE_NAME="r8.im/stability-ai/stable-diffusion@sha256:06eb78b36068500c616a7f33c15e6fa40404f8e14b5bfad57ebe0c7fe0f6bdf1"
fi
echo "Using docker image $IMAGE_NAME"
2022-08-23 22:28:18 +02:00
docker run --name sd -d -p 5000:5000 --gpus all $IMAGE_NAME
echo "Please open http://localhost:8000 in your favorite web browser!"
uvicorn main:app