Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/open-webui.git" did not exist on "9a6cbafdef7a1a44c7e3ad914996204d07c4a77e"
Commit c4f2176f authored by chenpangpang's avatar chenpangpang
Browse files

feat: dcu版初始提交

parent 2616a5e1
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 9 deletions
CogVideoX-5B-Space/requirements.txt
View file @ c4f2176f
safetensors>=0.4.4 safetensors>=0.4.4
spandrel>=0.3.4 spandrel>=0.3.4
tqdm>=4.66.5 tqdm>=4.66.5
opencv-python>=4.10.0.84
scikit-video>=1.1.11 scikit-video>=1.1.11
diffusers>=0.30.1 diffusers>=0.30.1
transformers>=4.44.0 transformers>=4.44.0
accelerate>=0.33.0 accelerate>=0.33.0
sentencepiece>=0.2.0 sentencepiece>=0.2.0
SwissArmyTransformer>=0.4.12 SwissArmyTransformer>=0.4.12
numpy==1.26.0
torch
torchvision
gradio>=4.42.0 gradio>=4.42.0
streamlit>=1.37.1 streamlit>=1.37.1
imageio==2.34.2 imageio==2.34.2
......
Dockerfile
View file @ c4f2176f
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
ARG IMAGE=cogvideox-5b ARG IMAGE=cogvideox-5b
ARG IMAGE_UPPER=CogVideoX-5B-Space ARG IMAGE_UPPER=CogVideoX-5B-Space
ARG BRANCH=gpu ARG BRANCH=dcu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER WORKDIR /root/$IMAGE/$IMAGE_UPPER
RUN pip install --no-cache-dir -r requirements.txt RUN pip install --no-cache-dir -r requirements.txt
...@@ -10,11 +10,11 @@ RUN pip install --no-cache-dir -r requirements.txt ...@@ -10,11 +10,11 @@ RUN pip install --no-cache-dir -r requirements.txt
######### #########
# Prod # # Prod #
######### #########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
ARG IMAGE=cogvideox-5b ARG IMAGE=cogvideox-5b
ARG IMAGE_UPPER=CogVideoX-5B-Space ARG IMAGE_UPPER=CogVideoX-5B-Space
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.2 /opt/conda/lib/python3.10/site-packages/gradio/ COPY chenyh/$IMAGE/frpc_linux_amd64_v0.3 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2 RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.3
COPY chenyh/$IMAGE/model_real_esran /root/$IMAGE_UPPER/model_real_esran COPY chenyh/$IMAGE/model_real_esran /root/$IMAGE_UPPER/model_real_esran
COPY chenyh/$IMAGE/model_rife /root/$IMAGE_UPPER/model_rife COPY chenyh/$IMAGE/model_rife /root/$IMAGE_UPPER/model_rife
COPY chenyh/$IMAGE/THUDM/CogVideoX-5b /root/$IMAGE_UPPER/THUDM/CogVideoX-5b COPY chenyh/$IMAGE/THUDM/CogVideoX-5b /root/$IMAGE_UPPER/THUDM/CogVideoX-5b
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment