search

lllyasviel / Paints-UNDO

Understand Human Behavior to Align True Needs
Apache License 2.0
3.49k stars 310 forks source link

Could anyone tell me what exact versions of torch, xformers and other dependencies they managed to run Paints-UNDO with? #95

Open nezdot opened 4 weeks ago

nezdot commented 4 weeks ago

I'm getting this error when trying to run python gradio_app.py

AssertionError: Torch not compiled with CUDA enabled

I did try the solution in #6 (pip install torch==2.3.0 torchvision torchaudio xformers --index-url https://download.pytorch.org/whl/cu121) but still no luck so I want to try to copy someone's setup to see if I can try to hard force it and if that would help.

This is what I have installed right now apparently btw

torch                     2.3.0+cu121
torchaudio                2.3.0+cu121
torchvision               0.18.0
diffusers                 0.28.0
xformers                  0.0.26.post1
wheel                     0.43.0
transformers              4.41.1

Please let me know if I should provide some more info!

ReloadProcz commented 3 weeks ago

Well I successfully ran it in a venv and the pip in the venv told this list to me:

Package                   Version
------------------------- ------------
accelerate                0.30.1
aiofiles                  23.2.1
altair                    5.4.1
annotated-types           0.7.0
anyio                     4.6.2.post1
attrs                     24.2.0
av                        13.1.0
bitsandbytes              0.43.1
certifi                   2024.8.30
charset-normalizer        3.4.0
click                     8.1.7
colorama                  0.4.6
coloredlogs               15.0.1
contourpy                 1.3.0
cycler                    0.12.1
diffusers                 0.28.0
einops                    0.8.0
exceptiongroup            1.2.2
fastapi                   0.112.4
ffmpy                     0.4.0
filelock                  3.13.1
flatbuffers               24.3.25
fonttools                 4.54.1
fsspec                    2024.2.0
gradio                    4.43.0
gradio_client             1.3.0
h11                       0.14.0
httpcore                  1.0.6
httpx                     0.27.2
huggingface-hub           0.24.0
humanfriendly             10.0
idna                      3.10
importlib_metadata        8.5.0
importlib_resources       6.4.5
intel-openmp              2021.4.0
Jinja2                    3.1.3
jsonschema                4.23.0
jsonschema-specifications 2024.10.1
kiwisolver                1.4.7
markdown-it-py            3.0.0
MarkupSafe                2.1.5
matplotlib                3.9.2
mdurl                     0.1.2
mkl                       2021.4.0
mpmath                    1.3.0
narwhals                  1.9.4
networkx                  3.2.1
numpy                     1.26.3
onnxruntime               1.19.2
opencv-python             4.10.0.84
orjson                    3.10.9
packaging                 24.1
pandas                    2.2.3
peft                      0.13.2
pillow                    10.2.0
pip                       24.2
protobuf                  3.20.0
psutil                    6.1.0
pydantic                  2.9.2
pydantic_core             2.23.4
pydub                     0.25.1
Pygments                  2.18.0
pyparsing                 3.2.0
pyreadline3               3.5.4
python-dateutil           2.9.0.post0
python-multipart          0.0.12
pytz                      2024.2
PyYAML                    6.0.2
referencing               0.35.1
regex                     2024.9.11
requests                  2.32.3
rich                      13.9.2
rpds-py                   0.20.0
ruff                      0.7.0
safetensors               0.4.5
semantic-version          2.10.0
setuptools                65.5.0
shellingham               1.5.4
six                       1.16.0
sniffio                   1.3.1
starlette                 0.38.6
sympy                     1.13.1
tbb                       2021.11.0
tensorboardX              2.6.2.2
tokenizers                0.19.1
tomlkit                   0.12.0
torch                     2.3.1+cu121
torchaudio                2.3.1+cu121
torchvision               0.18.1+cu121
tqdm                      4.66.5
transformers              4.41.1
typer                     0.12.5
typing_extensions         4.12.2
tzdata                    2024.2
urllib3                   2.2.3
uvicorn                   0.32.0
websockets                11.0.3
xformers                  0.0.27
zipp                      3.20.2

Can this help u?

Append: Forgot to say, if conflicts exists while setting up the environment, then create a venv inside the repository and use the pip in the venv to run the setup.