fix dockerfile

Author: tastelikefeet

Dockerfile

@@ -1,7 +1,5 @@
 FROM modelscope-registry.cn-hangzhou.cr.aliyuncs.com/modelscope-repo/modelscope:ubuntu22.04-cuda12.8.1-py311-torch2.9.1-1.35.0
 
-RUN echo '47.110.159.78 github.com' >> /etc/hosts
-
 # Install miniconda with Python 3.12
 RUN curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
     bash Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda && \
@@ -24,4 +22,5 @@ RUN pip install --no-cache-dir tinker==0.14.0 "ray[serve]" transformers peft acc
 
 RUN pip install -e . --no-build-isolation
 
-CMD ["bash", "cookbook/client/server/megatron/run.sh"]
+ENV TWINKLE_WORKDIR=/data
+CMD ["bash", "-c", "mkdir -p $TWINKLE_WORKDIR && cd $TWINKLE_WORKDIR && bash /twinkle/cookbook/client/server/megatron/run.sh 2>&1 | tee $TWINKLE_WORKDIR/run.log"]

INSTALL_MEGATRON.sh

@@ -95,7 +95,7 @@ echo ""
 python -c "
 import pkg_resources
 
-packages = ['peft', 'accelerate', 'transformers', 'modelscope', 'oss2', 'vllm', 'transformer_engine', 'megatron_core', 'flash_attn', 'numpy']
+packages = ['peft', 'accelerate', 'transformers', 'modelscope', 'vllm', 'transformer_engine', 'megatron_core', 'flash_attn', 'numpy']
 
 print('Installed package versions:')
 print('-' * 40)