E-Hospital commited on
Commit
089ee0c
1 Parent(s): 5ec9a40

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -6,6 +6,10 @@ COPY ./requirements.txt /code/requirements.txt
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
 
 
 
 
9
  COPY . .
10
 
11
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
 
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
+ RUN pip install --upgrade --force-reinstall scipy
10
+
11
+ RUN pip install flash-attn --no-build-isolation
12
+
13
  COPY . .
14
 
15
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]