hardik-0212 commited on
Commit
cfdd363
·
verified ·
1 Parent(s): 724c305

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. requirements.txt +2 -1
Dockerfile CHANGED
@@ -1,5 +1,7 @@
1
  FROM python:3.10-slim
2
 
 
 
3
  RUN apt-get update && apt-get install -y git libgl1 libglib2.0-0 ffmpeg && \
4
  apt-get clean && rm -rf /var/lib/apt/lists/*
5
 
 
1
  FROM python:3.10-slim
2
 
3
+ ENV HF_HOME=/data/.cache
4
+
5
  RUN apt-get update && apt-get install -y git libgl1 libglib2.0-0 ffmpeg && \
6
  apt-get clean && rm -rf /var/lib/apt/lists/*
7
 
requirements.txt CHANGED
@@ -6,4 +6,5 @@ torch
6
  diffusers
7
  transformers
8
  Pillow
9
- python-dotenv
 
 
6
  diffusers
7
  transformers
8
  Pillow
9
+ python-dotenv
10
+ accelerate