modified: requirements.txt

requirements.txt

@@ -1,18 +1,95 @@
-torch
-einops
-accelerate==0.34.2
-deepspeed==0.14.4
-einops==0.8.0
-transformers==4.41.2
-huggingface-hub==0.24.6
-datasets
-omegaconf
-diffusers
-sentencepiece
-opencv-python
-matplotlib
-onnxruntime
-torchvision
-timm
-invisible-watermark
-fire
+accelerate==1.4.0
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.8.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+diffusers==0.30.0
+einops==0.8.1
+fastapi==0.115.11
+ffmpy==0.5.0
+filelock==3.17.0
+fire==0.7.0
+-e git+https://github.com/HolmesShuan/FireFlow-Fast-Inversion-of-Rectified-Flow-for-Image-Semantic-Editing@df4eab3f73eed5efa175438532962c42bf033cf2#egg=FireFlow
+flux==1.3.5
+fsspec==2025.2.0
+gradio==5.20.0
+gradio_client==1.7.2
+groovy==0.1.2
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.29.1
+idna==3.10
+importlib_metadata==8.6.1
+invisible-watermark==0.2.0
+Jinja2==3.1.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.4.2
+numpy==2.2.3
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-cusparselt-cu12==0.6.2
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+opencv-python==4.11.0.86
+orjson==3.10.15
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+protobuf==5.29.3
+psutil==5.9.8
+pydantic==2.10.6
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2025.1
+PyWavelets==1.8.0
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+ruff==0.9.9
+safehttpx==0.1.6
+safetensors==0.5.3
+scipy==1.15.2
+semantic-version==2.10.0
+sentencepiece==0.2.0
+setuptools==75.8.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+spaces==0.40.0
+starlette==0.46.0
+sympy==1.13.1
+termcolor==2.5.0
+tokenizers==0.21.0
+tomlkit==0.13.2
+torch==2.6.0
+torch-fidelity==0.3.0
+torchvision==0.21.0
+tqdm==4.67.1
+transformers==4.49.0
+triton==3.2.0
+typer==0.15.2
+typing_extensions==4.12.2
+tzdata==2025.1
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==15.0
+wheel==0.45.1
+zipp==3.21.0