biib

Dockerfile

@@ -1,7 +1,6 @@
 # Utilizza l'immagine base NVIDIA CUDA con supporto GPU
 FROM huggingface/transformers-pytorch-gpu:latest
 # Installa Python, pip e altre dipendenze di sistema
-
 # Installa Python, pip e altre dipendenze di sistema
 RUN apt-get update && apt-get install -y \
     python3 \
@@ -17,10 +16,11 @@ RUN ln -s /usr/bin/python3 /usr/bin/python
 RUN pip install torch transformers datasets
 
 # Crea una directory per la cache di Hugging Face
-RUN mkdir -p /app/.cache
+RUN mkdir -p /root/.cache/huggingface
 
 # Imposta le variabili d'ambiente per la cache e altre configurazioni
-ENV HF_HOME=/app/.cache
+ENV HF_HOME=/root/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/root/.cache/huggingface
 ENV HUGGINGFACE_TOKEN=hf_1234567890abcdef1234567890abcdef12345678
 ENV HUGGINGFACE_USER=Rathalos
 ENV HUGGINGFACE_REPO=training_incite