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

[nezdot]

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]

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.

[zArvin6808]

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

Thank you very much for sharing, it makes things much easier for me, I'll send you a flying kiss

[BaiZe1998]

thx, you give me a big favor!