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
|
|
|
|
|
|
|
|
|
|
uvicorn main:app --reload
|
