Update Dockerfile

Dockerfile

@@ -20,20 +20,21 @@ WORKDIR /app
 # Create cache directory with proper permissions
 RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 RUN mkdir -p /app/output && chmod -R 777 /app/output
+
 # Set the cache environment variable
 ENV HF_HOME=/app/cache
 
 # Install Python dependencies
+RUN git clone https://github.com/facebookresearch/MobileLLM
 RUN pip install --upgrade pip
+RUN pip install -r requirement.txt
 
 # Copy the current directory contents into the container at /app
 COPY . /app
 
-# Install requirements
-RUN pip install -r requirements.txt
-
 # Unzip the data.zip file into the /app/data directory
 RUN unzip data.zip -d /app/data
 
-# Set the default command to run when starting the container
-CMD ["python", "x.py"]
+# data pre-process and specify the data path in pretrain.sh
+# run pretraining
+RUN sudo bash pretrain.sh