Files changed (1) hide show
  1. Dockerfile +6 -5
Dockerfile CHANGED
@@ -20,20 +20,21 @@ WORKDIR /app
20
  # Create cache directory with proper permissions
21
  RUN mkdir -p /app/cache && chmod -R 777 /app/cache
22
  RUN mkdir -p /app/output && chmod -R 777 /app/output
 
23
  # Set the cache environment variable
24
  ENV HF_HOME=/app/cache
25
 
26
  # Install Python dependencies
 
27
  RUN pip install --upgrade pip
 
28
 
29
  # Copy the current directory contents into the container at /app
30
  COPY . /app
31
 
32
- # Install requirements
33
- RUN pip install -r requirements.txt
34
-
35
  # Unzip the data.zip file into the /app/data directory
36
  RUN unzip data.zip -d /app/data
37
 
38
- # Set the default command to run when starting the container
39
- CMD ["python", "x.py"]
 
 
20
  # Create cache directory with proper permissions
21
  RUN mkdir -p /app/cache && chmod -R 777 /app/cache
22
  RUN mkdir -p /app/output && chmod -R 777 /app/output
23
+
24
  # Set the cache environment variable
25
  ENV HF_HOME=/app/cache
26
 
27
  # Install Python dependencies
28
+ RUN git clone https://github.com/facebookresearch/MobileLLM
29
  RUN pip install --upgrade pip
30
+ RUN pip install -r requirement.txt
31
 
32
  # Copy the current directory contents into the container at /app
33
  COPY . /app
34
 
 
 
 
35
  # Unzip the data.zip file into the /app/data directory
36
  RUN unzip data.zip -d /app/data
37
 
38
+ # data pre-process and specify the data path in pretrain.sh
39
+ # run pretraining
40
+ RUN sudo bash pretrain.sh