mindchain commited on
Commit
5977b18
Β·
verified Β·
1 Parent(s): d52670a

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -8
Dockerfile CHANGED
@@ -7,10 +7,7 @@ RUN apt-get update && apt-get install -y \
7
  git \
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
- # Copy requirements
11
- COPY requirements_v6.txt requirements.txt
12
-
13
- # Install core Python packages (required)
14
  RUN pip install --no-cache-dir \
15
  torch \
16
  transformers \
@@ -19,14 +16,13 @@ RUN pip install --no-cache-dir \
19
  datasets \
20
  huggingface_hub \
21
  peft \
22
- bitsandbytes \
23
- optimum[openvino]
24
 
25
  # Copy training script
26
- COPY train_arithmetic_v6_openvino.py .
27
 
28
  # Set HF token from secret
29
  ENV HF_TOKEN=${HF_TOKEN}
30
 
31
  # Run training
32
- CMD ["python", "train_arithmetic_v6_openvino.py"]
 
7
  git \
8
  && rm -rf /var/lib/apt/lists/*
9
 
10
+ # Install core Python packages ONLY - no IPEX, no OpenVINO
 
 
 
11
  RUN pip install --no-cache-dir \
12
  torch \
13
  transformers \
 
16
  datasets \
17
  huggingface_hub \
18
  peft \
19
+ bitsandbytes
 
20
 
21
  # Copy training script
22
+ COPY train_arithmetic_v7_clean.py .
23
 
24
  # Set HF token from secret
25
  ENV HF_TOKEN=${HF_TOKEN}
26
 
27
  # Run training
28
+ CMD ["python", "train_arithmetic_v7_clean.py"]