Imag / requirements.txt
Baraaqasem's picture
Update requirements.txt
8473106 verified
deepspeed; platform_system != "Windows"
bitsandbytes
flask
flask_restful
flask_cors
faiss-cpu
fire
h5py
numpy>=1.23.5
pandas<2.0.0
peft
torch
torchvision
torchaudio
xformers~=0.0.20
jupyterlab>=4.0.2
notebook>=6.5.4
albumentations>=1.1.0
opencv-python>=4.2.0
pudb~=2019.2
imageio>=2.14.1
imageio-ffmpeg>=0.4.7
pytorch-lightning>=1.5.9
omegaconf~=2.1.1
gradio==5.0.0
pillow~=9.5.0
einops>=0.4.1
torch-fidelity>=0.3.0
setuptools>=59.5.0
transformers==4.37.2
torchmetrics>=0.6.0
lpips
dreamsim
image-reward
kornia>=0.6
diffusers>=0.18.0
accelerate>=0.20.3
safetensors
datasets
tqdm>=4.64.1
matplotlib>=3.7.1
taming-transformers-rom1504~=0.0.6
madgrad>=1.1
-e git+https://github.com/openai/CLIP.git@main#egg=clip
dominate>=2.8.0
-e git+https://github.com/CompVis/latent-diffusion.git#egg=latent-diffusion #ldm
jsonargparse
openai
nltk~=3.8.1
krippendorff
statsmodels
plotly
fal_client
open_clip_torch
decord
huggingface_hub
open-clip-torch-any-py3
modelscope
protobuf==3.20.*
rotary_embedding_torch
av
natsort
tensorboard
wandb