Spaces:
Paused
Paused
Commit
·
ea61392
1
Parent(s):
5de8cee
Fixing model download issue v10
Browse files- Dockerfile +15 -2
- README.md +0 -1
Dockerfile
CHANGED
|
@@ -32,9 +32,22 @@ RUN mkdir -p /app/main/checkpoints && \
|
|
| 32 |
ARG HF_TOKEN
|
| 33 |
ENV HF_TOKEN=${HF_TOKEN}
|
| 34 |
|
| 35 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 36 |
RUN if [ -n "$HF_TOKEN" ]; then \
|
| 37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
else \
|
| 39 |
echo "No Hugging Face token provided. Models will need to be downloaded separately."; \
|
| 40 |
exit 1; \
|
|
|
|
| 32 |
ARG HF_TOKEN
|
| 33 |
ENV HF_TOKEN=${HF_TOKEN}
|
| 34 |
|
| 35 |
+
# Verify token and environment
|
| 36 |
+
RUN echo "Token length: ${#HF_TOKEN}" && \
|
| 37 |
+
echo "Cache directory:" && \
|
| 38 |
+
ls -la /app/.cache/huggingface && \
|
| 39 |
+
echo "Checkpoints directory:" && \
|
| 40 |
+
ls -la /app/main/checkpoints && \
|
| 41 |
+
echo "Testing huggingface-cli" && \
|
| 42 |
+
huggingface-cli whoami
|
| 43 |
+
|
| 44 |
+
# Download model using litgpt command line with verbose output
|
| 45 |
RUN if [ -n "$HF_TOKEN" ]; then \
|
| 46 |
+
echo "Starting model download..." && \
|
| 47 |
+
litgpt download mistralai/Mistral-7B-Instruct-v0.3 \
|
| 48 |
+
--access_token ${HF_TOKEN} \
|
| 49 |
+
--checkpoint_dir /app/main/checkpoints \
|
| 50 |
+
--verbose || { echo "Download failed with status $?"; exit 1; }; \
|
| 51 |
else \
|
| 52 |
echo "No Hugging Face token provided. Models will need to be downloaded separately."; \
|
| 53 |
exit 1; \
|
README.md
CHANGED
|
@@ -24,5 +24,4 @@ folders
|
|
| 24 |
LLM-Engine
|
| 25 |
Main
|
| 26 |
main.py
|
| 27 |
-
utils.py
|
| 28 |
```
|
|
|
|
| 24 |
LLM-Engine
|
| 25 |
Main
|
| 26 |
main.py
|
|
|
|
| 27 |
```
|