Update Dockerfile

Dockerfile

@@ -5,15 +5,13 @@ RUN apt-get update && apt-get install -y \
     git \
     python3.10 \
     python3-pip \
+    python-is-python3 \
     wget \
     ninja-build \
     gcc \
     g++ \
     && rm -rf /var/lib/apt/lists/*
 
-# Create symbolic link for python
-RUN ln -s /usr/bin/python3.10 /usr/bin/python
-
 WORKDIR /app
 
 # Install basic Python packages first
@@ -29,6 +27,11 @@ ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
+# Clone Self-Lengthen repository
+RUN git clone https://github.com/QwenLM/Self-Lengthen.git && \
+    mv Self-Lengthen/* . && \
+    rm -rf Self-Lengthen
+
 # Install dependencies in order
 COPY requirements.txt .
 RUN pip3 install --no-cache-dir \
@@ -56,11 +59,8 @@ RUN git clone -b self-lengthen https://github.com/quanshr/FastChat.git && \
 # Install LLaMA Factory
 RUN pip3 install --no-cache-dir llamafactory
 
-# Copy project structure
-COPY qwen /app/qwen
-COPY llama /app/llama
-COPY eval /app/eval
-COPY app.py /app/app.py
+# Create directories for models and results
+RUN mkdir -p models results
 
 # Set environment variables
 ENV CUDA_VISIBLE_DEVICES=0
@@ -68,20 +68,33 @@ ENV WORLD_SIZE=1
 ENV RANK=0
 ENV MASTER_ADDR=localhost
 ENV MASTER_PORT=29500
-
-# Create directories
-RUN mkdir -p /app/qwen/results
+ENV MODEL_PATH=/app/models/base_model
+ENV INSTRUCT_COUNT=5000
+ENV MAX_ITER=3
 
 # Create startup script
 RUN echo '#!/bin/bash\n\
-if [ "$USE_UI" = "true" ]; then\n\
-    python app.py\n\
-else\n\
-    cd /app/qwen\n\
-    bash run.sh --base_model=$MODEL_PATH --instruct_count=$INSTRUCT_COUNT --max_iter=$MAX_ITER\n\
-    python collect_data.py\n\
-fi' > /app/start.sh && \
+# Download model if needed\n\
+if [ ! -d "$MODEL_PATH" ]; then\n\
+    mkdir -p $MODEL_PATH\n\
+    git lfs install\n\
+    git clone https://huggingface.co/Qwen/Qwen2-7B-Instruct $MODEL_PATH\n\
+fi\n\
+\n\
+# Run the training process\n\
+cd /app/qwen\n\
+bash run.sh --base_model=$MODEL_PATH --instruct_count=$INSTRUCT_COUNT --max_iter=$MAX_ITER\n\
+python collect_data.py\n\
+\n\
+# Start the web interface\n\
+python app.py\n' > /app/start.sh && \
 chmod +x /app/start.sh
 
+# Create a simple web interface
+COPY app.py .
+
+# Expose port for web interface
+EXPOSE 7860
+
 # Command to run
 ENTRYPOINT ["/app/start.sh"]