Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(docker): change 'nvidia' profile name to 'cuda'

Browse Source
This commit is contained in:
Eugene Brodsky 2024-06-24 16:56:07 -04:00
parent b672cc37a7
commit 42c2dea202
3 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docker/.env.sample
View File

@ -19,8 +19,9 @@
## INVOKEAI_PORT is the port on which the InvokeAI web interface will be available ## INVOKEAI_PORT is the port on which the InvokeAI web interface will be available
# INVOKEAI_PORT=9090 # INVOKEAI_PORT=9090
## GPU_DRIVER can be set to either `nvidia` or `rocm` to enable GPU support in the container accordingly. ## GPU_DRIVER can be set to either `cuda` or `rocm` to enable GPU support in the container accordingly.
# GPU_DRIVER=nvidia #| rocm # GPU_DRIVER=cuda #| rocm
## CONTAINER_UID can be set to the UID of the user on the host system that should own the files in the container. ## CONTAINER_UID can be set to the UID of the user on the host system that should own the files in the container.
## It is usually not necessary to change this. Use `id -u` on the host system to find the UID.
# CONTAINER_UID=1000 # CONTAINER_UID=1000

4
docker/docker-compose.yml
View File

@ -1,7 +1,5 @@
# Copyright (c) 2023 Eugene Brodsky https://github.com/ebr # Copyright (c) 2023 Eugene Brodsky https://github.com/ebr
version: '3.8'
x-invokeai: &invokeai x-invokeai: &invokeai
image: "local/invokeai:latest" image: "local/invokeai:latest"
build: build:
@ -32,7 +30,7 @@ x-invokeai: &invokeai
services: services:
invokeai-nvidia: invokeai-cuda:
<<: *invokeai <<: *invokeai
deploy: deploy:
resources: resources:

6
docker/run.sh
View File

@ -8,11 +8,15 @@ run() {
local build_args="" local build_args=""
local profile="" local profile=""
# create .env file if it doesn't exist, otherwise docker compose will fail
touch .env touch .env
# parse .env file for build args
build_args=$(awk '$1 ~ /=[^$]/ && $0 !~ /^#/ {print "--build-arg " $0 " "}' .env) && build_args=$(awk '$1 ~ /=[^$]/ && $0 !~ /^#/ {print "--build-arg " $0 " "}' .env) &&
profile="$(awk -F '=' '/GPU_DRIVER/ {print $2}' .env)" profile="$(awk -F '=' '/GPU_DRIVER/ {print $2}' .env)"
[[ -z "$profile" ]] && profile="nvidia" # default to 'cuda' profile
[[ -z "$profile" ]] && profile="cuda"
local service_name="invokeai-$profile" local service_name="invokeai-$profile"