Paints-UNDO

• Paints-UNDO

Ubuntu 20.04 安装 conda

mkdir -p ~/miniconda3
cd ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
~/miniconda3/bin/conda init bash
source ~/.bashrc
conda --version

git clone https://github.com/lllyasviel/Paints-UNDO.git
cd Paints-UNDO
conda create -n paints_undo python=3.10
conda activate paints_undo
pip install xformers
pip install -r requirements.txt
python gradio_app.py

conda deactivate

conda activate paints_undo

# python gradio_app.py
Traceback (most recent call last):
  File "/root/Paints-UNDO/gradio_app.py", line 13, in <module>
    import torch
  File "/root/miniconda3/envs/paints_undo/lib/python3.10/site-packages/torch/__init__.py", line 239, in <module>
    from torch._C import *  # noqa: F403
ImportError: /root/miniconda3/envs/paints_undo/lib/python3.10/site-packages/torch/lib/../../nvidia/cusparse/lib/libcusparse.so.12: undefined symbol: __nvJitLinkAddData_12_1, version libnvJitLink.so.12

# nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Mon_Oct_24_19:12:58_PDT_2022
Cuda compilation tools, release 12.0, V12.0.76
Build cuda_12.0.r12.0/compiler.31968024_0

pip uninstall torch torchvision torchaudio

cu117 不支持

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

cu118 支持 cuda 12.0

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

conda deactivate

conda activate paints_undo
cd Paints-UNDO
python gradio_app.py

# python gradio_app.py
Traceback (most recent call last):
  File "/root/Paints-UNDO/gradio_app.py", line 39, in <module>
    tokenizer = CLIPTokenizer.from_pretrained(model_name, subfolder="tokenizer")
  File "/root/miniconda3/envs/paints_undo/lib/python3.10/site-packages/transformers/tokenization_utils_base.py", line 2094, in from_pretrained
    raise EnvironmentError(
OSError: Can't load tokenizer for 'lllyasviel/paints_undo_single_frame'. If you were trying to load it from 'https://huggingface.co/models', make sure you don't have a local directory with the same name. Otherwise, make sure 'lllyasviel/paints_undo_single_frame' is the correct path to a directory containing all relevant files for a CLIPTokenizer tokenizer.

pip install httpx[socks]

Running on local URL:  http://0.0.0.0:7860

conda activate paints_undo
cd Paints-UNDO
python gradio_app.py