Skip to content

Commit cfcd844

Browse files
committed
Add requirements and instructions to run on windows
1 parent 4d8d8e2 commit cfcd844

File tree

4 files changed

+25
-0
lines changed

4 files changed

+25
-0
lines changed

Diff for: .gitignore

+8
Original file line numberDiff line numberDiff line change
@@ -1 +1,9 @@
11
.DS_Store
2+
tokenflow-results*/**
3+
venv/
4+
__pycache__/
5+
*.pyc
6+
latents/
7+
data/**
8+
!data/wolf.mp4
9+
!data/woman-running.mp4

Diff for: README.md

+12
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,23 @@ For more see the [project webpage](https://diffusion-tokenflow.github.io).
3232
<td><img src="assets/videos.gif"></td>
3333

3434
## Environment
35+
36+
Using conda
3537
```
3638
conda create -n tokenflow python=3.9
3739
conda activate tokenflow
3840
pip install -r requirements.txt
3941
```
42+
43+
Using cpython Windows (powershell)
44+
```powershell
45+
python -m pip install virtualenv
46+
python -m virtualenv venv
47+
venv/Scripts/activate.ps1
48+
pip install -r requirements.txt
49+
pip install -r requirements-torch.txt
50+
```
51+
4052
## Preprocess
4153

4254
Preprocess you video by running using the following command:

Diff for: requirements-torch.txt

+2
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
-i https://download.pytorch.org/whl/cu117
2+
torch==2.0.1+cu117

Diff for: requirements.txt

+3
Original file line numberDiff line numberDiff line change
@@ -9,3 +9,6 @@ pyyaml
99
accelerate
1010
xformers
1111
tensorboard
12+
kornia
13+
av
14+

0 commit comments

Comments
 (0)