StableVITON Deployer commited on
Commit
48eeaef
·
1 Parent(s): 6d8f43a

Fix: Update dependencies to resolve MT5Tokenizer import error

Browse files
Files changed (1) hide show
  1. requirements.txt +18 -3
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
  # Core Machine Learning (Latest compatible versions)
2
  torch>=2.2.0
3
  torchvision>=0.17.0
4
- diffusers>=0.27.0
5
- transformers[sentencepiece]>=4.39.0
6
- accelerate>=0.28.0
7
  xformers>=0.0.25 # Memory efficient attention
8
 
9
  # Image Processing
@@ -24,3 +24,18 @@ python-dotenv>=1.0.1
24
  huggingface-hub>=0.22.0
25
  sentencepiece>=0.1.99
26
  protobuf==3.20.3 # Pinned to 3.20.x for compatibility
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Core Machine Learning (Latest compatible versions)
2
  torch>=2.2.0
3
  torchvision>=0.17.0
4
+ diffusers>=0.28.0
5
+ transformers>=4.41.0
6
+ accelerate>=0.29.0
7
  xformers>=0.0.25 # Memory efficient attention
8
 
9
  # Image Processing
 
24
  huggingface-hub>=0.22.0
25
  sentencepiece>=0.1.99
26
  protobuf==3.20.3 # Pinned to 3.20.x for compatibility
27
+
28
+
29
+ pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117
30
+ pip install pytorch-lightning==1.5.0
31
+ pip install einops
32
+ pip install opencv-python==4.7.0.72
33
+ pip install matplotlib
34
+ pip install omegaconf
35
+ pip install albumentations
36
+ pip install transformers==4.33.2
37
+ pip install xformers==0.0.19
38
+ pip install triton==2.0.0
39
+ pip install open-clip-torch==2.19.0
40
+ pip install diffusers==0.20.2
41
+ pip install scipy==1.10.1