Commit
·
58cd8b9
1
Parent(s):
8830b80
set cache, rm chmod, liger
Browse files- Dockerfile +1 -4
- requirements.txt +2 -1
Dockerfile
CHANGED
@@ -21,10 +21,7 @@ RUN pip install --upgrade pip && \
|
|
21 |
RUN install -m 755 sft.py /usr/local/bin/sft
|
22 |
|
23 |
# # Set environment variables
|
24 |
-
|
25 |
-
|
26 |
-
# chmod 777 app
|
27 |
-
RUN chmod 777 /
|
28 |
|
29 |
# Expose the Gradio port
|
30 |
EXPOSE 7860
|
|
|
21 |
RUN install -m 755 sft.py /usr/local/bin/sft
|
22 |
|
23 |
# # Set environment variables
|
24 |
+
ENV HF_HOME=/usr/src/app/.cache/huggingface
|
|
|
|
|
|
|
25 |
|
26 |
# Expose the Gradio port
|
27 |
EXPOSE 7860
|
requirements.txt
CHANGED
@@ -1,2 +1,3 @@
|
|
1 |
git+https://github.com/huggingface/trl.git@main#egg=trl
|
2 |
-
gradio
|
|
|
|
1 |
git+https://github.com/huggingface/trl.git@main#egg=trl
|
2 |
+
gradio
|
3 |
+
liger-kernel
|