Teddy-Project commited on
Commit
c24e581
·
verified ·
1 Parent(s): e1ab702

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -1
Dockerfile CHANGED
@@ -1,4 +1,11 @@
1
  FROM ghcr.io/huggingface/text-generation-inference:1.3
2
 
 
3
  ENV MODEL_ID=HuggingFaceH4/zephyr-7b-beta
4
- ENV HF_HOME=/tmp/huggingface
 
 
 
 
 
 
 
1
  FROM ghcr.io/huggingface/text-generation-inference:1.3
2
 
3
+ # Definimos el modelo
4
  ENV MODEL_ID=HuggingFaceH4/zephyr-7b-beta
5
+
6
+ # Definimos el cache para Hugging Face
7
+ ENV HF_HOME=/tmp/huggingface
8
+ ENV TRANSFORMERS_CACHE=/tmp/huggingface
9
+
10
+ # Creamos el directorio por si no existe
11
+ RUN mkdir -p /tmp/huggingface && chmod -R 777 /tmp/huggingface