ssoxye commited on
Commit
a628acd
·
1 Parent(s): 313389b

Add torchvision dependency

Browse files
.gitignore CHANGED
@@ -14,3 +14,5 @@ preprocess/mhp_extension/demo/*.jpg
14
  !diffusers3/models/**
15
  !diffusers3/pipelines/
16
  !diffusers3/pipelines/**
 
 
 
14
  !diffusers3/models/**
15
  !diffusers3/pipelines/
16
  !diffusers3/pipelines/**
17
+
18
+ **/.ipynb_checkpoints/
.ipynb_checkpoints/requirements-checkpoint.txt CHANGED
@@ -12,3 +12,4 @@ imageio
12
  opencv-python-headless==4.10.0.84
13
 
14
  torch>=2.3,<2.4
 
 
12
  opencv-python-headless==4.10.0.84
13
 
14
  torch>=2.3,<2.4
15
+ torchvision>=0.18,<0.19
requirements.txt CHANGED
@@ -12,3 +12,4 @@ imageio
12
  opencv-python-headless==4.10.0.84
13
 
14
  torch>=2.3,<2.4
 
 
12
  opencv-python-headless==4.10.0.84
13
 
14
  torch>=2.3,<2.4
15
+ torchvision>=0.18,<0.19