requirements?

[moming133]

Create a python 3.10.13 conda env (you could also use virtualenv)

#conda create -n py310 python=3.10.13
#conda activate py310

Install pytorch with your CUDA version, e.g.

#pip install torch==2.5.9+cu118 torchvision --extra-index-url https://download.pytorch.org/whl/cu118

torch==2.5.0
transformers==4.45.2
datasets==2.20.0
accelerate==0.26.1
jupyter==1.0.0
numpy==1.26.3
pillow==10.2.0
peft==0.13.2
diffusers==0.30.3
timm==0.9.16
opencv-python
gradio
spaces
insightface
onnxruntime

[RedAIGC]

Create a python 3.10.13 conda env (you could also use virtualenv)

#conda create -n py310 python=3.10.13 #conda activate py310

Install pytorch with your CUDA version, e.g.

#pip install torch==2.5.9+cu118 torchvision --extra-index-url https://download.pytorch.org/whl/cu118

torch==2.5.0 transformers==4.45.2 datasets==2.20.0 accelerate==0.26.1 jupyter==1.0.0 numpy==1.26.3 pillow==10.2.0 peft==0.13.2 diffusers==0.30.3 timm==0.9.16 opencv-python gradio spaces insightface onnxruntime

FYI

absl-py==2.0.0
accelerate==0.30.1
aiofiles==23.2.1
albucore==0.0.13
albumentations==0.2.0
annotated-types==0.7.0
antlr4-python3-runtime==4.9.3
anyio==4.6.2.post1
argon2-cffi==23.1.0
argon2-cffi-bindings==21.2.0
asttokens==2.4.0
astunparse==1.6.3
asyncio==3.4.3
attrs==23.1.0
audioread==3.0.1
backcall==0.2.0
beautifulsoup4==4.12.2
bitsandbytes==0.41.1
bleach==6.0.0
blis==0.7.11
boto3==1.28.78
botocore==1.31.78
braceexpand==0.1.7
cachetools==5.3.1
catalogue==2.0.10
certifi==2023.7.22
cffi==1.16.0
cloudpathlib==0.15.1
cmake==3.27.6
coloredlogs==15.0.1
comm==0.1.4
confection==0.1.3
contourpy==1.1.1
cos-python-sdk-v5==1.9.26
coscmd==1.8.6.33
cpm-kernels==1.0.11
crcmod==1.7
cycler==0.12.1
cymem==2.0.8
Cython==3.0.11
datasets==2.14.6
DateTime==5.2
debugpy==1.8.0
decorator==5.1.1
deepspeed==0.14.4
defusedxml==0.7.1
diffusers==0.27.1
dill==0.3.7
dm-tree==0.1.8
easydict==1.13
einops==0.8.0
eval_type_backport==0.2.0
exceptiongroup==1.1.3
execnet==2.0.2
executing==2.0.0
expecttest==0.1.3
facexlib==0.3.0
fairscale==0.4.13
fakeredis==2.25.1
fastapi==0.115.2
fastChat==0.1.1
fastjsonschema==2.18.1
ffmpy==0.4.0
filelock==3.12.4
filterpy==1.4.5
fire==0.7.0
flash-attn==2.5.8
flatbuffers==24.3.25
fonttools==4.43.1
ftfy==6.1.1
gast==0.5.4
google-auth==2.23.2
google-auth-oauthlib==0.4.6
gradio==5.1.0
gradio_client==1.4.0
grpcio==1.59.0
gunicorn==22.0.0
h11==0.14.0
hjson==3.1.0
httpcore==1.0.6
httpx==0.27.2
huggingface-hub==0.25.2
humanfriendly==10.0
hypothesis==5.35.1
idna==3.4
imageio==2.35.1
imagesize==1.4.1
imgaug==0.2.6
iniconfig==2.0.0
insightface==0.7.3
intel-openmp==2021.4.0
ipdb==0.13.13
ipykernel==6.25.2
ipython==8.16.1
ipython-genutils==0.2.0
jedi==0.19.1
Jinja2==3.1.2
jmespath==1.0.1
joblib==1.3.2
json5==0.9.14
jsonlines==4.0.0
jsonschema==4.19.1
jsonschema-specifications==2023.7.1
kiwisolver==1.4.5
kornia==0.5.0
kornia_rs==0.1.5
langcodes==3.3.0
lazy_loader==0.4
librosa==0.9.2
lightning-utilities==0.11.8
loralib==0.1.2
lxml==5.3.0
Markdown==3.4.4
markdown-it-py==3.0.0
MarkupSafe==2.1.3
matplotlib==3.8.0
matplotlib-inline==0.1.6
mdit-py-plugins==0.4.0
mdurl==0.1.2
mistune==3.0.2
mkl==2021.1.1
mkl-devel==2021.1.1
mkl-include==2021.1.1
mock==5.1.0
mpmath==1.3.0
multiprocess==0.70.15
murmurhash==1.0.10
nbclient==0.8.0
nbconvert==7.9.2
nbformat==5.9.2
nest-asyncio==1.5.8
networkx==3.3
ninja==1.11.1.1
notebook==6.4.10
numpy==1.24.4
nvfuser==0.0.20+gitunknown
nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-dali-cuda120==1.30.0
nvidia-ml-py==12.560.30
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.1.105
nvidia-nvtx-cu12==12.1.105
nvidia-pyindex==1.0.9
oauthlib==3.2.2
omegaconf==2.3.0
onnx-graphsurgeon==0.3.27
onnxruntime==1.19.2
onnxruntime-gpu==1.19.2
open-clip-torch==2.23.0
opencv-contrib-python==4.5.5.64
opencv-python==4.5.5.64
opencv-python-headless==4.5.5.64
optimum==1.13.3
optimum-quanto==0.2.1
orjson==3.10.7
paddleocr==2.9.1
pandocfilters==1.5.0
parso==0.8.3
pathy==0.10.2
peft==0.13.0
pexpect==4.8.0
pickleshare==0.7.5
pillow==10.3.0
pillow-heif==0.13.1
platformdirs==3.11.0
pluggy==1.3.0
polygraphy==0.49.0
pooch==1.7.0
preshed==3.0.9
prettytable==3.9.0
prometheus-async==22.2.0
prometheus_client==0.20.0
prompt-toolkit==3.0.39
protobuf==4.24.4
ptyprocess==0.7.0
pure-eval==0.2.2
py-cpuinfo==9.0.0
pyasn1==0.5.0
pyasn1-modules==0.3.0
pybind11==2.11.1
pybind11-global==2.11.1
pyclipper==1.3.0.post6
pycparser==2.21
pycryptodome==3.19.0
pydantic==2.9.2
pydantic_core==2.23.4
pydub==0.25.1
Pygments==2.16.1
pyparsing==3.1.1
pyPEG2==2.15.2
pytest==7.4.2
pytest-flakefinder==1.1.0
pytest-rerunfailures==12.0
pytest-shard==0.1.2
pytest-xdist==3.3.1
python-dateutil==2.8.2
python-docx==1.1.2
python-hostlist==1.23.0
python-multipart==0.0.12
pytorch-lightning==2.3.0
pytorch-quantization==2.1.2
PyYAML==6.0.1
pyzmq==26.2.0
RapidFuzz==3.10.1
redenv==0.0.5
redeuler==0.20.5
redis==5.1.1
referencing==0.30.2
regex==2023.10.3
requests==2.31.0
requests-oauthlib==1.3.1
resampy==0.4.2
rich==13.9.2
rpds-py==0.10.4
rsa==4.9
ruff==0.6.9
s3transfer==0.7.0
safetensors==0.4.5
scikit-image==0.24.0
scikit-learn==1.5.2
seaborn==0.13.0
semantic-version==2.10.0
Send2Trash==1.8.2
sentencepiece==0.1.99
shapely==2.0.6
shellingham==1.5.4
six==1.16.0
smart-open==6.4.0
sniffio==1.3.1
sortedcontainers==2.4.0
soundfile==0.12.1
soupsieve==2.5
spacy==3.7.1
spacy-legacy==3.0.12
spacy-loggers==1.0.5
sphinx-glpi-theme==0.3
srsly==2.4.8
stack-data==0.6.3
starlette==0.40.0
SwissArmyTransformer==0.4.8
sympy==1.12
tabulate==0.9.0
tbb==2021.10.0
tensorboard==2.9.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.1
tensorboardX==2.6.2.2
termcolor==2.5.0
terminado==0.17.1
thinc==8.2.1
threadpoolctl==3.2.0
thriftpy2==0.5.2
tifffile==2024.8.30
timm==0.9.10
tinycss2==1.2.1
tokenizers==0.20.1
toml==0.10.2
tomli==2.0.1
tomlkit==0.12.0
torch==2.3.0+cu121
torchmetrics==1.6.0
torchvision==0.18.0
tornado==6.3.3
tqdm==4.66.1
traitlets==5.9.0
transformers==4.45.2
triton==2.3.0
typer==0.12.5
types-dataclasses==0.6.6
typing_extensions==4.9.0
uvicorn==0.31.1
voluptuous==0.14.2
wasabi==1.1.2
wcwidth==0.2.8
weasel==0.3.2
webdataset==0.2.73
webencodings==0.5.1
websockets==12.0
Werkzeug==3.0.0
wrapt==1.16.0
xdoctest==1.0.2
xformers==0.0.26.post1
xlrd==1.2.0
xmltodict==0.13.0
xxhash==3.4.1
zope.interface==6.1