Manjeet9812 commited on
Commit
90b8e78
·
verified ·
1 Parent(s): dfec546

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -5
requirements.txt CHANGED
@@ -2,23 +2,22 @@
2
  fastapi==0.111.0
3
  uvicorn[standard]==0.30.0
4
 
5
- # Pin to TF 2.15-compatible NumPy stack
6
  numpy==1.26.4
7
- ml-dtypes==0.2.0
8
  tensorflow==2.15.1
9
  keras==2.15.0
10
 
11
- # Diffusion stack (compatible set; avoids cached_download errors)
12
  diffusers==0.21.2
13
  transformers==4.30.2
14
  huggingface-hub==0.16.4
15
  safetensors==0.3.1
16
 
17
- # PyTorch (CPU wheels)
18
  torch==2.8.0
19
  torchvision==0.23.0
20
 
21
- # OpenAI CLIP
22
  git+https://github.com/openai/CLIP.git
23
 
24
  # Utils
 
2
  fastapi==0.111.0
3
  uvicorn[standard]==0.30.0
4
 
5
+ # TF stack (let TF pull correct ml-dtypes)
6
  numpy==1.26.4
 
7
  tensorflow==2.15.1
8
  keras==2.15.0
9
 
10
+ # Diffusion stack (older, stable)
11
  diffusers==0.21.2
12
  transformers==4.30.2
13
  huggingface-hub==0.16.4
14
  safetensors==0.3.1
15
 
16
+ # PyTorch CPU (fine even on GPU Space; we don’t do heavy torch ops)
17
  torch==2.8.0
18
  torchvision==0.23.0
19
 
20
+ # CLIP from Git
21
  git+https://github.com/openai/CLIP.git
22
 
23
  # Utils