diff --git a/docker-compose.yml b/docker-compose.yml index 46c17b0..3b0b257 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -28,7 +28,15 @@ services: environment: - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api - forge: &forge + forge: &reforge + <<: *base_service + profiles: ["reforge"] + build: ./services/reforge + image: sd-reforge:80 + environment: + - CLI_ARGS=--allow-code --xformers --enable-insecure-extension-access --api --pin-shared-memory --cuda-malloc --cuda-stream + + reforge: &forge <<: *base_service profiles: ["forge"] build: ./services/forge