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

Updated README to add Docker

Browse Source
This commit is contained in:
Blake 2022-07-27 00:28:22 -04:00 committed by GitHub
parent 30fcd4f9ff
commit 11877f6173
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
README.md
View File

@ -55,10 +55,19 @@ A 4-second clip of 32 frames is shown below.
* LocalAttention: Make sure you have CUDA installed and compile the local attention kernel. * LocalAttention: Make sure you have CUDA installed and compile the local attention kernel.
```shell ```shell
git clone https://github.com/Sleepychord/Image-Local-Attention pip install git+https://github.com/Sleepychord/Image-Local-Attention
cd Image-Local-Attention && python setup.py install
``` ```
## Docker
Alternatively you can use Docker to handle all dependencies.
1. Run ```./build_image.sh```
2. Run ```./run_image.sh```
3. Run ```./install_image_local_attention```
Optionally, after that you can recommit the image to avoid having to install image local attention again.
### Download ### Download
Our code will automatically download or detect the models into the path defined by environment variable `SAT_HOME`. You can also manually download [CogVideo-Stage1](https://lfs.aminer.cn/misc/cogvideo/cogvideo-stage1.zip) and [CogVideo-Stage2](https://lfs.aminer.cn/misc/cogvideo/cogvideo-stage2.zip) and place them under SAT_HOME (with folders named `cogvideo-stage1` and `cogvideo-stage2`) Our code will automatically download or detect the models into the path defined by environment variable `SAT_HOME`. You can also manually download [CogVideo-Stage1](https://lfs.aminer.cn/misc/cogvideo/cogvideo-stage1.zip) and [CogVideo-Stage2](https://lfs.aminer.cn/misc/cogvideo/cogvideo-stage2.zip) and place them under SAT_HOME (with folders named `cogvideo-stage1` and `cogvideo-stage2`)