Skip to content

Commit

Permalink
Storagemonitor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
robballantyne committed Mar 4, 2024
1 parent 98f9d1b commit 05f9608
Show file tree
Hide file tree
Showing 4 changed files with 10 additions and 10 deletions.
2 changes: 0 additions & 2 deletions .github/workflows/docker-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,6 @@ jobs:
python:
- "3.10"
pytorch:
- "2.1.1"
- "2.1.2"
- "2.2.0"
steps:
Expand Down Expand Up @@ -100,7 +99,6 @@ jobs:
python:
- "3.10"
pytorch:
- "2.1.1"
- "2.1.2"
- "2.2.0"
cuda:
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,7 @@ You can also [build from source](#building-images) by editing `.env` and running

Supported Python versions: `3.10`

Supported Pytorch versions: `2.2.0`, `2.1.2` `2.1.1`
Supported Pytorch versions: `2.2.0`, `2.1.2`

Supported Platforms: `NVIDIA CUDA`, `AMD ROCm`, `CPU`

Expand Down
6 changes: 3 additions & 3 deletions config/provisioning/default.sh
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
#!/bin/false
#!/bin/bash
# This file will be sourced in init.sh
# Namespace functions with provisioning_

Expand Down Expand Up @@ -33,8 +33,8 @@ EXTENSIONS=(
CHECKPOINT_MODELS=(
"https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt"
#"https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.ckpt"
"https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors"
"https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors"
#"https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors"
#"https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-1.0/resolve/main/sd_xl_refiner_1.0.safetensors"
)

LORA_MODELS=(
Expand Down
10 changes: 6 additions & 4 deletions docker-compose.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,15 @@ version: "3.8"
# Compose file build variables set in .env
services:
supervisor:
platform: linux/amd64
build:
context: ./build
args:
IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/jupyter-pytorch:2.1.1-py3.10-cuda-11.8.0-runtime-22.04}
IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/jupyter-pytorch:2.2.1-py3.10-cuda-11.8.0-runtime-22.04}
tags:
- "ghcr.io/ai-dock/stable-diffusion-webui:${IMAGE_TAG:-jupyter-pytorch-2.1.1-py3.10-cuda-11.8.0-runtime-22.04}"
- "ghcr.io/ai-dock/stable-diffusion-webui:${IMAGE_TAG:-jupyter-pytorch-2.2.1-py3.10-cuda-11.8.0-runtime-22.04}"

image: ghcr.io/ai-dock/stable-diffusion-webui:${IMAGE_TAG:-jupyter-pytorch-2.1.1-py3.10-cuda-11.8.0-runtime-22.04}
image: ghcr.io/ai-dock/stable-diffusion-webui:${IMAGE_TAG:-jupyter-pytorch-2.2.1-py3.10-cuda-11.8.0-runtime-22.04}

devices:
- "/dev/dri:/dev/dri"
Expand Down Expand Up @@ -53,11 +54,12 @@ services:
- SSH_PORT_LOCAL=${SSH_PORT_LOCAL:-22}
- SERVICEPORTAL_PORT_HOST=${SERVICEPORTAL_PORT_HOST:-1111}
- SERVICEPORTAL_METRICS_PORT=${SERVICEPORTAL_METRICS_PORT:-21111}
- WEBUI_BRANCH=${WEBUI_BRANCH:-}
- WEBUI_FLAGS=${WEBUI_FLAGS:-}
- WEBUI_PORT_HOST=${WEBUI_PORT_HOST:-7860}
- WEBUI_PORT_LOCAL=${WEBUI_PORT_LOCAL:-17860}
- WEBUI_METRICS_PORT=${WEBUI_METRICS_PORT:-27860}
- JUPYTER_PORT_HOST=${JUPYTER_PORT_HOST:-8888}
- JUPYTER_METRICS_PORT=${JUPYTER_METRICS_PORT:-28888}
- SERVERLESS=${SERVERLESS:-false}
# - PROVISIONING_SCRIPT=${PROVISIONING_SCRIPT:-}
#- PROVISIONING_SCRIPT=${PROVISIONING_SCRIPT:-}

0 comments on commit 05f9608

Please sign in to comment.