Spaces:
Build error
Build error
no
Browse files- Dockerfile +14 -1
Dockerfile
CHANGED
@@ -1,5 +1,7 @@
|
|
1 |
# Utilizza l'immagine base NVIDIA CUDA con supporto GPU
|
2 |
FROM huggingface/transformers-pytorch-gpu:latest
|
|
|
|
|
3 |
# Installa Python, pip e altre dipendenze di sistema
|
4 |
RUN apt-get update && apt-get install -y \
|
5 |
python3 \
|
@@ -18,7 +20,7 @@ RUN pip install torch transformers datasets
|
|
18 |
RUN mkdir -p /app/.cache
|
19 |
|
20 |
# Imposta le variabili d'ambiente per la cache e altre configurazioni
|
21 |
-
ENV
|
22 |
ENV HUGGINGFACE_TOKEN=hf_1234567890abcdef1234567890abcdef12345678
|
23 |
ENV HUGGINGFACE_USER=Rathalos
|
24 |
ENV HUGGINGFACE_REPO=training_incite
|
@@ -29,3 +31,14 @@ WORKDIR /app
|
|
29 |
|
30 |
# Comando per eseguire lo script di training
|
31 |
CMD ["python", "train_model.py"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
# Utilizza l'immagine base NVIDIA CUDA con supporto GPU
|
2 |
FROM huggingface/transformers-pytorch-gpu:latest
|
3 |
+
# Installa Python, pip e altre dipendenze di sistema
|
4 |
+
|
5 |
# Installa Python, pip e altre dipendenze di sistema
|
6 |
RUN apt-get update && apt-get install -y \
|
7 |
python3 \
|
|
|
20 |
RUN mkdir -p /app/.cache
|
21 |
|
22 |
# Imposta le variabili d'ambiente per la cache e altre configurazioni
|
23 |
+
ENV HF_HOME=/app/.cache
|
24 |
ENV HUGGINGFACE_TOKEN=hf_1234567890abcdef1234567890abcdef12345678
|
25 |
ENV HUGGINGFACE_USER=Rathalos
|
26 |
ENV HUGGINGFACE_REPO=training_incite
|
|
|
31 |
|
32 |
# Comando per eseguire lo script di training
|
33 |
CMD ["python", "train_model.py"]
|
34 |
+
Struttura dei File Locali
|
35 |
+
Assicurati che i tuoi file locali siano organizzati come segue:
|
36 |
+
|
37 |
+
bash
|
38 |
+
Copia codice
|
39 |
+
/project_directory
|
40 |
+
Dockerfile
|
41 |
+
requirements.txt
|
42 |
+
train_model.py
|
43 |
+
/data
|
44 |
+
text_file.txt
|