doublelotus commited on
Commit
490e8f8
·
1 Parent(s): 1937f91
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. requirements.txt +2 -1
Dockerfile CHANGED
@@ -17,4 +17,5 @@ ENV HF_HOME=/code/transformers_cache
17
 
18
  COPY . .
19
 
20
- CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app", "--timeout", "600"]
 
 
17
 
18
  COPY . .
19
 
20
+ # CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app", "--timeout", "600"]
21
+ CMD ["gunicorn", "main:app", "--workers", "1", "--worker-class", "uvicorn.workers.UvicornWorker", "--bind", "0.0.0.0:7860", "--timeout", "600"]
requirements.txt CHANGED
@@ -11,4 +11,5 @@ matplotlib # Required for image processing and mask visualization
11
  transformers
12
  diffusers
13
  numpy
14
- accelerate
 
 
11
  transformers
12
  diffusers
13
  numpy
14
+ accelerate
15
+ xformers