ClownRat commited on
Commit
ff899bc
1 Parent(s): 96549f3

Fix dependency.

Browse files
Files changed (2) hide show
  1. pyproject.toml +1 -1
  2. requirements.txt +2 -0
pyproject.toml CHANGED
@@ -21,7 +21,7 @@ dependencies = [
21
  "moviepy==1.0.3", "scenedetect==0.6.3", "numpy", "scikit-learn==1.2.2",
22
  "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",
23
  "bitsandbytes==0.41.0", "pydantic<2,>=1", "markdown2[all]",
24
- "gradio==3.35.2", "gradio_client==0.2.9",
25
  "requests", "httpx==0.24.0", "openai", "uvicorn", "fastapi", "wandb"
26
  ]
27
 
 
21
  "moviepy==1.0.3", "scenedetect==0.6.3", "numpy", "scikit-learn==1.2.2",
22
  "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",
23
  "bitsandbytes==0.41.0", "pydantic<2,>=1", "markdown2[all]",
24
+ "gradio==3.35.2", "gradio_client==0.2.9", opencv-python,
25
  "requests", "httpx==0.24.0", "openai", "uvicorn", "fastapi", "wandb"
26
  ]
27
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  huggingface_hub==0.22.2
2
  torch>=2.0.1
3
  torchvision>=0.15.2
@@ -14,6 +15,7 @@ imageio==2.34.0
14
  imageio-ffmpeg==0.4.9
15
  moviepy==1.0.3
16
  scenedetect==0.6.3
 
17
  numpy
18
  scikit-learn==1.2.2
19
  einops==0.6.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
  huggingface_hub==0.22.2
3
  torch>=2.0.1
4
  torchvision>=0.15.2
 
15
  imageio-ffmpeg==0.4.9
16
  moviepy==1.0.3
17
  scenedetect==0.6.3
18
+ opencv-python
19
  numpy
20
  scikit-learn==1.2.2
21
  einops==0.6.1