Skip to content

Commit fbc5c35

Browse files
Resolve memory usage situation in Auto (#620)
Fixes #612 --------- Co-authored-by: AbdBarho <ka70911@gmail.com>
1 parent 90affeb commit fbc5c35

2 files changed

Lines changed: 5 additions & 1 deletion

File tree

docker-compose.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ services:
2929
<<: *base_service
3030
profiles: ["auto"]
3131
build: ./services/AUTOMATIC1111
32-
image: sd-auto:68
32+
image: sd-auto:69
3333
environment:
3434
- CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api
3535

services/AUTOMATIC1111/Dockerfile

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,10 @@ RUN --mount=type=cache,target=/root/.cache/pip \
4848
git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \
4949
git+https://github.com/mlfoundations/open_clip.git@bb6e834e9c70d9c27d0dc3ecedeebeaeb1ffad6b
5050

51+
# there seems to be a memory leak (or maybe just memory not being freed fast eno8ugh) that is fixed by this version of malloc
52+
# maybe move this up to the dependencies list.
53+
RUN apt-get -y install libgoogle-perftools-dev && apt-get clean
54+
ENV LD_PRELOAD=libtcmalloc.so
5155

5256
COPY . /docker
5357

0 commit comments

Comments
 (0)