modified: requirements.txt

requirements.txt

@@ -1,94 +1,18 @@
-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==7.0.0
-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
-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
+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