rajux75 commited on
Commit
07580f7
·
verified ·
1 Parent(s): deaed92

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -9
requirements.txt CHANGED
@@ -1,19 +1,21 @@
1
  # requirements.txt
2
  fastapi
3
- uvicorn[standard] # ASGI server
4
- python-dotenv # For local .env loading
5
  requests
6
  # Core ML
7
- torch # Or tensorflow if preferred, but diffusers often favors torch
8
  transformers
9
  diffusers[torch]
10
- accelerate # Helps optimize model loading and inference
 
11
  # Image/Video Handling
12
  Pillow
13
- imageio # For potential video frame handling if needed
 
14
  # API Extras
15
  pydantic
16
- slowapi # For rate limiting
17
- # Optional: For better performance on supported hardware
18
- # bitsandbytes # For 8-bit/4-bit quantization (check compatibility)
19
- # xformers # For memory-efficient attention (if GPU available)
 
1
  # requirements.txt
2
  fastapi
3
+ uvicorn[standard]
4
+ python-dotenv
5
  requests
6
  # Core ML
7
+ torch
8
  transformers
9
  diffusers[torch]
10
+ accelerate
11
+ peft #
12
  # Image/Video Handling
13
  Pillow
14
+ imageio[ffmpeg] # Ensure ffmpeg support for MP4
15
+ numpy # Added in previous step, ensure it's here
16
  # API Extras
17
  pydantic
18
+ slowapi
19
+ # Optional: Performance (Uncomment if needed and compatible)
20
+ bitsandbytes
21
+ xformers