Yan Bai commited on
Commit
9b8469c
·
1 Parent(s): 60b1ccb
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -20,6 +20,8 @@ RUN groupadd -g 1000 user && \
20
  WORKDIR $HOME/app
21
  RUN cp -r /opt/mbridge/memory_estimator $HOME/app && chown -R user:user $HOME/app
22
 
 
 
23
  # HF Spaces 默认通过 $PORT 注入端口
24
  ENV PYTHONPATH=/opt/Megatron-LM:$PYTHONPATH
25
  ENV PORT=7860
 
20
  WORKDIR $HOME/app
21
  RUN cp -r /opt/mbridge/memory_estimator $HOME/app && chown -R user:user $HOME/app
22
 
23
+ RUN echo "from webui.main import app" > $HOME/app/app.py
24
+
25
  # HF Spaces 默认通过 $PORT 注入端口
26
  ENV PYTHONPATH=/opt/Megatron-LM:$PYTHONPATH
27
  ENV PORT=7860