JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-12-03 11:12:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

cuda 11.3

Browse Source
This commit is contained in:
Stephan Auerhahn 2022-07-25 08:29:56 -07:00
parent abfc7af8f9
commit 30edd2a800
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cog.yaml
View File

@ -3,7 +3,8 @@
build:
gpu: true
cuda: "11.3"
system_packages:
- "ffmpeg"
- "libsm6"
@ -17,15 +18,16 @@ build:
- "icetk==0.0.4"
- "opencv-python==4.6.0.66"
- "SwissArmyTransformer==0.2.9"
- "torch==1.9.0"
- "torchvision==0.10.0"
- "torch==1.11.0"
- "torchvision==0.12.0"
- "tqdm==4.64.0"
run:
- "pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 pytorch-lightning torch-fidelity -f https://download.pytorch.org/whl/torch_stable.html"
- "mkdir -p /sharefs/cogview-new; cd /sharefs/cogview-new; wget https://models.nmb.ai/cogvideo/cogvideo-stage1.tar.gz -O - | tar xz"
- "cd /sharefs/cogview-new; wget https://models.nmb.ai/cogvideo/cogvideo-stage2.tar.gz -O - | tar xz"
- "mkdir -p /root/.icetk_models; wget -O /root/.icetk_models/ice_text.model https://models.nmb.ai/cogvideo/ice_text.model"
- "mkdir -p /root/.tcetk_models; wget -O /root/.icetk_models/ice_image.pt https://models.nmb.ai/cogvideo/ice_image.pt"
- "mkdir -p /root/.tcetk_models; wget -O /root/.icetk_models/ice_image.pt https://models.nmb.ai/cogvideo/ice_image.pt"
#- "cd /sharefs/cogview-new; wget https://models.nmb.ai/cogview2/cogview2-dsr.tar.gz -O - | tar xz"
predict: "predict.py:Predictor"