OneTrainer icon indicating copy to clipboard operation
OneTrainer copied to clipboard
verified publisher icon Nerogar

[Bug]: ModuleNotFoundError: No module named 'customtkinter'

Open dathide opened this issue 1 year ago • 13 comments

What happened?

Thanks for OneTrainer! I did a fresh install on Arch Linux:

╭─zen at arch in ~/m2a/ai-img 24-05-05 - 13:12:40
╰─○ git clone https://github.com/Nerogar/OneTrainer.git
Cloning into 'OneTrainer'...
remote: Enumerating objects: 7054, done.
remote: Counting objects: 100% (3482/3482), done.
remote: Compressing objects: 100% (616/616), done.
remote: Total 7054 (delta 3149), reused 3033 (delta 2862), pack-reused 3572
Receiving objects: 100% (7054/7054), 2.22 MiB | 8.63 MiB/s, done.
Resolving deltas: 100% (5565/5565), done.
╭─zen at arch in ~/m2a/ai-img 24-05-05 - 13:12:47
╰─○ cd OneTrainer
╭─zen at arch in ~/m2a/ai-img/OneTrainer 24-05-05 - 13:11:19
╰─○ pyenv install 3.10   
Installing Python-3.10.14...
Installed Python-3.10.14 to /home/zen/.pyenv/versions/3.10.14
╭─zen at arch in ~/m2a/ai-img/OneTrainer on master✔ 24-05-05 - 13:12:50
╰─⠠⠵ pyenv virtualenv 3.10 py3.10
╭─zen at arch in ~/m2a/ai-img/OneTrainer on master✔ 24-05-05 - 13:14:22
╰─⠠⠵ pyenv local py3.10
(py3.10) ╭─zen at arch in ~/m2a/ai-img/OneTrainer on master✘✘✘ 24-05-05 - 13:14:29
╰─(py3.10) ⠠⠵ pip install -r requirements.txt
Looking in indexes: https://pypi.org/simple, https://download.pytorch.org/whl/cu118
Obtaining diffusers from git+https://github.com/huggingface/diffusers.git@5d848ec#egg=diffusers (from -r requirements-global.txt (line 17))
  Cloning https://github.com/huggingface/diffusers.git (to revision 5d848ec) to /home/zen/.pyenv/versions/3.10.14/envs/py3.10/src/diffusers
  Running command git clone --filter=blob:none --quiet https://github.com/huggingface/diffusers.git /home/zen/.pyenv/versions/3.10.14/envs/py3.10/src/diffusers
  WARNING: Did not find branch or tag '5d848ec', assuming revision or ref.
  Running command git checkout -q 5d848ec
  Resolved https://github.com/huggingface/diffusers.git to commit 5d848ec
  Installing build dependencies ... done
  Checking if build backend supports build_editable ... done
  Getting requirements to build editable ... done
  Preparing editable metadata (pyproject.toml) ... done
Obtaining mgds from git+https://github.com/Nerogar/mgds.git@1dc3009#egg=mgds (from -r requirements-global.txt (line 27))
  Cloning https://github.com/Nerogar/mgds.git (to revision 1dc3009) to /home/zen/.pyenv/versions/3.10.14/envs/py3.10/src/mgds
  Running command git clone --filter=blob:none --quiet https://github.com/Nerogar/mgds.git /home/zen/.pyenv/versions/3.10.14/envs/py3.10/src/mgds
  WARNING: Did not find branch or tag '1dc3009', assuming revision or ref.
  Running command git checkout -q 1dc3009
  Resolved https://github.com/Nerogar/mgds.git to commit 1dc3009
  Preparing metadata (setup.py) ... done
Collecting torch==2.2.0+cu118
  Downloading https://download.pytorch.org/whl/cu118/torch-2.2.0%2Bcu118-cp310-cp310-linux_x86_64.whl (811.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 811.7/811.7 MB 5.0 MB/s eta 0:00:00
Collecting torchvision==0.17.0+cu118
  Downloading https://download.pytorch.org/whl/cu118/torchvision-0.17.0%2Bcu118-cp310-cp310-linux_x86_64.whl (6.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 6.2/6.2 MB 29.2 MB/s eta 0:00:00
Collecting onnxruntime-gpu==1.16.3
  Downloading onnxruntime_gpu-1.16.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (157.1 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 157.1/157.1 MB 19.3 MB/s eta 0:00:00
Collecting xformers==0.0.24+cu118
  Downloading https://download.pytorch.org/whl/cu118/xformers-0.0.24%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl (218.1 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 218.1/218.1 MB 14.6 MB/s eta 0:00:00
Collecting bitsandbytes==0.43.0
  Downloading bitsandbytes-0.43.0-py3-none-manylinux_2_24_x86_64.whl (102.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 102.2/102.2 MB 21.6 MB/s eta 0:00:00
Collecting numpy==1.26.2
  Downloading numpy-1.26.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (18.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 18.2/18.2 MB 30.4 MB/s eta 0:00:00
Collecting opencv-python==4.8.1.78
  Downloading opencv_python-4.8.1.78-cp37-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (61.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 61.7/61.7 MB 22.7 MB/s eta 0:00:00
Collecting pillow==10.2.0
  Downloading https://download.pytorch.org/whl/pillow-10.2.0-cp310-cp310-manylinux_2_28_x86_64.whl (4.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.5/4.5 MB 55.0 MB/s eta 0:00:00
Collecting tqdm==4.66.1
  Downloading tqdm-4.66.1-py3-none-any.whl (78 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 78.3/78.3 kB 35.4 MB/s eta 0:00:00
Collecting PyYAML==6.0.1
  Downloading PyYAML-6.0.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (705 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 705.5/705.5 kB 43.0 MB/s eta 0:00:00
Collecting huggingface-hub==0.20.3
  Downloading huggingface_hub-0.20.3-py3-none-any.whl (330 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 330.1/330.1 kB 51.0 MB/s eta 0:00:00
Collecting scipy==1.11.4
  Downloading scipy-1.11.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (36.4 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 36.4/36.4 MB 29.6 MB/s eta 0:00:00
Collecting accelerate==0.25.0
  Downloading accelerate-0.25.0-py3-none-any.whl (265 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 265.7/265.7 kB 35.0 MB/s eta 0:00:00
Collecting safetensors==0.4.1
  Downloading safetensors-0.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.3/1.3 MB 49.6 MB/s eta 0:00:00
Collecting tensorboard==2.15.1
  Downloading tensorboard-2.15.1-py3-none-any.whl (5.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.5/5.5 MB 31.4 MB/s eta 0:00:00
Collecting pytorch-lightning==2.1.3
  Downloading pytorch_lightning-2.1.3-py3-none-any.whl (777 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 777.7/777.7 kB 39.4 MB/s eta 0:00:00
Collecting transformers==4.36.2
  Downloading transformers-4.36.2-py3-none-any.whl (8.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 8.2/8.2 MB 35.2 MB/s eta 0:00:00
Collecting omegaconf==2.3.0
  Downloading omegaconf-2.3.0-py3-none-any.whl (79 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 79.5/79.5 kB 5.9 MB/s eta 0:00:00
Collecting invisible-watermark==0.2.0
  Downloading invisible_watermark-0.2.0-py3-none-any.whl (1.6 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.6/1.6 MB 49.2 MB/s eta 0:00:00
Collecting pooch==1.8.0
  Downloading pooch-1.8.0-py3-none-any.whl (62 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 62.7/62.7 kB 13.7 MB/s eta 0:00:00
Collecting open-clip-torch==2.23.0
  Downloading open_clip_torch-2.23.0-py3-none-any.whl (1.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.5/1.5 MB 49.1 MB/s eta 0:00:00
Collecting dadaptation==3.2
  Using cached dadaptation-3.2-py3-none-any.whl
Collecting lion-pytorch==0.1.2
  Downloading lion_pytorch-0.1.2-py3-none-any.whl (4.4 kB)
Collecting prodigyopt==1.0
  Downloading prodigyopt-1.0-py3-none-any.whl (5.5 kB)
Collecting schedulefree==1.2.5
  Using cached schedulefree-1.2.5-py3-none-any.whl
Collecting scalene==1.5.39
  Downloading scalene-1.5.39-cp310-cp310-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl (854 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 854.7/854.7 kB 45.7 MB/s eta 0:00:00
Collecting customtkinter==5.2.1
  Downloading customtkinter-5.2.1-py3-none-any.whl (295 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 296.0/296.0 kB 39.0 MB/s eta 0:00:00
Collecting nvidia-curand-cu11==10.3.0.86
  Downloading https://download.pytorch.org/whl/cu118/nvidia_curand_cu11-10.3.0.86-py3-none-manylinux1_x86_64.whl (58.1 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 58.1/58.1 MB 26.6 MB/s eta 0:00:00
Collecting jinja2
  Downloading https://download.pytorch.org/whl/Jinja2-3.1.3-py3-none-any.whl (133 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 133.2/133.2 kB 9.8 MB/s eta 0:00:00
Collecting nvidia-cusparse-cu11==11.7.5.86
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cusparse_cu11-11.7.5.86-py3-none-manylinux1_x86_64.whl (204.1 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 204.1/204.1 MB 7.5 MB/s eta 0:00:00
Collecting nvidia-nvtx-cu11==11.8.86
  Downloading https://download.pytorch.org/whl/cu118/nvidia_nvtx_cu11-11.8.86-py3-none-manylinux1_x86_64.whl (99 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 99.1/99.1 kB 3.3 MB/s eta 0:00:00
Collecting fsspec
  Downloading fsspec-2024.3.1-py3-none-any.whl (171 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 172.0/172.0 kB 51.1 MB/s eta 0:00:00
Collecting nvidia-cublas-cu11==11.11.3.6
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cublas_cu11-11.11.3.6-py3-none-manylinux1_x86_64.whl (417.9 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 417.9/417.9 MB 9.6 MB/s eta 0:00:00
Collecting nvidia-cuda-cupti-cu11==11.8.87
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cuda_cupti_cu11-11.8.87-py3-none-manylinux1_x86_64.whl (13.1 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 13.1/13.1 MB 33.0 MB/s eta 0:00:00
Collecting nvidia-cuda-runtime-cu11==11.8.89
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cuda_runtime_cu11-11.8.89-py3-none-manylinux1_x86_64.whl (875 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 875.6/875.6 kB 10.1 MB/s eta 0:00:00
Collecting nvidia-cudnn-cu11==8.7.0.84
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cudnn_cu11-8.7.0.84-py3-none-manylinux1_x86_64.whl (728.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 728.5/728.5 MB 6.0 MB/s eta 0:00:00
Collecting typing-extensions>=4.8.0
  Downloading typing_extensions-4.11.0-py3-none-any.whl (34 kB)
Collecting nvidia-cuda-nvrtc-cu11==11.8.89
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cuda_nvrtc_cu11-11.8.89-py3-none-manylinux1_x86_64.whl (23.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 23.2/23.2 MB 26.0 MB/s eta 0:00:00
Collecting nvidia-cusolver-cu11==11.4.1.48
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cusolver_cu11-11.4.1.48-py3-none-manylinux1_x86_64.whl (128.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 128.2/128.2 MB 17.3 MB/s eta 0:00:00
Collecting nvidia-nccl-cu11==2.19.3
  Downloading https://download.pytorch.org/whl/cu118/nvidia_nccl_cu11-2.19.3-py3-none-manylinux1_x86_64.whl (135.3 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 135.3/135.3 MB 16.9 MB/s eta 0:00:00
Collecting networkx
  Downloading networkx-3.3-py3-none-any.whl (1.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.7/1.7 MB 22.2 MB/s eta 0:00:00
Collecting triton==2.2.0
  Downloading https://download.pytorch.org/whl/triton-2.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (167.9 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 167.9/167.9 MB 15.0 MB/s eta 0:00:00
Collecting filelock
  Downloading filelock-3.14.0-py3-none-any.whl (12 kB)
Collecting nvidia-cufft-cu11==10.9.0.58
  Downloading https://download.pytorch.org/whl/cu118/nvidia_cufft_cu11-10.9.0.58-py3-none-manylinux1_x86_64.whl (168.4 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 168.4/168.4 MB 15.0 MB/s eta 0:00:00
Collecting sympy
  Downloading https://download.pytorch.org/whl/sympy-1.12-py3-none-any.whl (5.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.7/5.7 MB 46.9 MB/s eta 0:00:00
Collecting requests
  Downloading requests-2.31.0-py3-none-any.whl (62 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 62.6/62.6 kB 11.8 MB/s eta 0:00:00
Collecting flatbuffers
  Downloading flatbuffers-24.3.25-py2.py3-none-any.whl (26 kB)
Collecting packaging
  Downloading packaging-24.0-py3-none-any.whl (53 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 53.5/53.5 kB 15.4 MB/s eta 0:00:00
Collecting coloredlogs
  Downloading coloredlogs-15.0.1-py2.py3-none-any.whl (46 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 46.0/46.0 kB 15.9 MB/s eta 0:00:00
Collecting protobuf
  Downloading protobuf-5.26.1-cp37-abi3-manylinux2014_x86_64.whl (302 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 302.8/302.8 kB 12.2 MB/s eta 0:00:00
Collecting psutil
  Downloading psutil-5.9.8-cp36-abi3-manylinux_2_12_x86_64.manylinux2010_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (288 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 288.2/288.2 kB 12.6 MB/s eta 0:00:00
Collecting six>1.9
  Downloading six-1.16.0-py2.py3-none-any.whl (11 kB)
Collecting protobuf
  Downloading protobuf-4.23.4-cp37-abi3-manylinux2014_x86_64.whl (304 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 304.5/304.5 kB 13.0 MB/s eta 0:00:00
Collecting absl-py>=0.4
  Downloading absl_py-2.1.0-py3-none-any.whl (133 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 133.7/133.7 kB 12.9 MB/s eta 0:00:00
Collecting grpcio>=1.48.2
  Downloading grpcio-1.63.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (5.6 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 5.6/5.6 MB 14.4 MB/s eta 0:00:00
Requirement already satisfied: setuptools>=41.0.0 in /home/zen/.pyenv/versions/3.10.14/envs/py3.10/lib/python3.10/site-packages (from tensorboard==2.15.1->-r requirements-global.txt (line 13)) (65.5.0)
Collecting google-auth-oauthlib<2,>=0.5
  Downloading google_auth_oauthlib-1.2.0-py2.py3-none-any.whl (24 kB)
Collecting werkzeug>=1.0.1
  Downloading werkzeug-3.0.2-py3-none-any.whl (226 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 226.8/226.8 kB 20.0 MB/s eta 0:00:00
Collecting markdown>=2.6.8
  Downloading Markdown-3.6-py3-none-any.whl (105 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 105.4/105.4 kB 19.8 MB/s eta 0:00:00
Collecting tensorboard-data-server<0.8.0,>=0.7.0
  Downloading tensorboard_data_server-0.7.2-py3-none-manylinux_2_31_x86_64.whl (6.6 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 6.6/6.6 MB 21.0 MB/s eta 0:00:00
Collecting google-auth<3,>=1.6.3
  Downloading google_auth-2.29.0-py2.py3-none-any.whl (189 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 189.2/189.2 kB 24.5 MB/s eta 0:00:00
Collecting torchmetrics>=0.7.0
  Downloading torchmetrics-1.3.2-py3-none-any.whl (841 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 841.5/841.5 kB 22.4 MB/s eta 0:00:00
Collecting lightning-utilities>=0.8.0
  Downloading lightning_utilities-0.11.2-py3-none-any.whl (26 kB)
Collecting tokenizers<0.19,>=0.14
  Downloading tokenizers-0.15.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (3.6 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 3.6/3.6 MB 23.8 MB/s eta 0:00:00
Collecting regex!=2019.12.17
  Downloading regex-2024.4.28-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (774 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 774.1/774.1 kB 26.4 MB/s eta 0:00:00
Collecting antlr4-python3-runtime==4.9.*
  Using cached antlr4_python3_runtime-4.9.3-py3-none-any.whl
Collecting PyWavelets>=1.1.1
  Downloading pywavelets-1.6.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (4.5 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 4.5/4.5 MB 27.2 MB/s eta 0:00:00
Collecting platformdirs>=2.5.0
  Downloading platformdirs-4.2.1-py3-none-any.whl (17 kB)
Collecting ftfy
  Downloading ftfy-6.2.0-py3-none-any.whl (54 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 54.4/54.4 kB 3.1 MB/s eta 0:00:00
Collecting timm
  Downloading timm-0.9.16-py3-none-any.whl (2.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.2/2.2 MB 30.8 MB/s eta 0:00:00
Collecting sentencepiece
  Downloading sentencepiece-0.2.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.3/1.3 MB 28.2 MB/s eta 0:00:00
Collecting cloudpickle>=2.2.1
  Downloading cloudpickle-3.0.0-py3-none-any.whl (20 kB)
Collecting pynvml<=11.5,>=11.0.0
  Downloading pynvml-11.5.0-py3-none-any.whl (53 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 53.1/53.1 kB 16.3 MB/s eta 0:00:00
Collecting rich>=10.7.0
  Downloading rich-13.7.1-py3-none-any.whl (240 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 240.7/240.7 kB 43.5 MB/s eta 0:00:00
Collecting wheel>=0.36.1
  Using cached wheel-0.43.0-py3-none-any.whl (65 kB)
Collecting darkdetect
  Downloading darkdetect-0.8.0-py3-none-any.whl (9.0 kB)
Collecting importlib-metadata
  Downloading importlib_metadata-7.1.0-py3-none-any.whl (24 kB)
Collecting aiohttp!=4.0.0a0,!=4.0.0a1
  Downloading aiohttp-3.9.5-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.2/1.2 MB 33.2 MB/s eta 0:00:00
Collecting pyasn1-modules>=0.2.1
  Downloading pyasn1_modules-0.4.0-py3-none-any.whl (181 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 181.2/181.2 kB 24.6 MB/s eta 0:00:00
Collecting rsa<5,>=3.1.4
  Downloading rsa-4.9-py3-none-any.whl (34 kB)
Collecting cachetools<6.0,>=2.0.0
  Downloading cachetools-5.3.3-py3-none-any.whl (9.3 kB)
Collecting requests-oauthlib>=0.7.0
  Downloading requests_oauthlib-2.0.0-py2.py3-none-any.whl (24 kB)
Collecting MarkupSafe>=2.0
  Downloading https://download.pytorch.org/whl/MarkupSafe-2.1.5-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (25 kB)
Collecting idna<4,>=2.5
  Downloading idna-3.7-py3-none-any.whl (66 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 66.8/66.8 kB 23.0 MB/s eta 0:00:00
Collecting certifi>=2017.4.17
  Downloading certifi-2024.2.2-py3-none-any.whl (163 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 163.8/163.8 kB 39.0 MB/s eta 0:00:00
Collecting charset-normalizer<4,>=2
  Downloading charset_normalizer-3.3.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (142 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 142.1/142.1 kB 10.6 MB/s eta 0:00:00
Collecting urllib3<3,>=1.21.1
  Downloading urllib3-2.2.1-py3-none-any.whl (121 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 121.1/121.1 kB 39.0 MB/s eta 0:00:00
Collecting pygments<3.0.0,>=2.13.0
  Downloading pygments-2.18.0-py3-none-any.whl (1.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.2/1.2 MB 31.9 MB/s eta 0:00:00
Collecting markdown-it-py>=2.2.0
  Downloading markdown_it_py-3.0.0-py3-none-any.whl (87 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 87.5/87.5 kB 29.5 MB/s eta 0:00:00
Collecting humanfriendly>=9.1
  Downloading humanfriendly-10.0-py2.py3-none-any.whl (86 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 86.8/86.8 kB 29.0 MB/s eta 0:00:00
Collecting wcwidth<0.3.0,>=0.2.12
  Downloading wcwidth-0.2.13-py2.py3-none-any.whl (34 kB)
Collecting zipp>=0.5
  Downloading zipp-3.18.1-py3-none-any.whl (8.2 kB)
Collecting mpmath>=0.19
  Downloading https://download.pytorch.org/whl/mpmath-1.3.0-py3-none-any.whl (536 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 536.2/536.2 kB 50.9 MB/s eta 0:00:00
Collecting multidict<7.0,>=4.5
  Downloading multidict-6.0.5-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (124 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 124.3/124.3 kB 36.6 MB/s eta 0:00:00
Collecting async-timeout<5.0,>=4.0
  Downloading async_timeout-4.0.3-py3-none-any.whl (5.7 kB)
Collecting attrs>=17.3.0
  Downloading attrs-23.2.0-py3-none-any.whl (60 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 60.8/60.8 kB 25.6 MB/s eta 0:00:00
Collecting frozenlist>=1.1.1
  Downloading frozenlist-1.4.1-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (239 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 239.5/239.5 kB 33.0 MB/s eta 0:00:00
Collecting aiosignal>=1.1.2
  Downloading aiosignal-1.3.1-py3-none-any.whl (7.6 kB)
Collecting yarl<2.0,>=1.0
  Downloading yarl-1.9.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (301 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 301.6/301.6 kB 15.8 MB/s eta 0:00:00
Collecting mdurl~=0.1
  Downloading mdurl-0.1.2-py3-none-any.whl (10.0 kB)
Collecting pyasn1<0.7.0,>=0.4.6
  Downloading pyasn1-0.6.0-py2.py3-none-any.whl (85 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 85.3/85.3 kB 32.4 MB/s eta 0:00:00
Collecting oauthlib>=3.0.0
  Downloading oauthlib-3.2.2-py3-none-any.whl (151 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 151.7/151.7 kB 32.2 MB/s eta 0:00:00
Building wheels for collected packages: diffusers
  Building editable for diffusers (pyproject.toml) ... done
  Created wheel for diffusers: filename=diffusers-0.27.0.dev0-0.editable-py3-none-any.whl size=11091 sha256=29d54233c6d8098b62892a5a3213c7f6910c937bbff9b82ff8c4f8098c401daa
  Stored in directory: /tmp/pip-ephem-wheel-cache-lsh_08dg/wheels/ae/11/b2/3b3ec057d16e3dfa927bb8b335781a3177a5d4663b8ab52c59
Successfully built diffusers
Installing collected packages: wcwidth, sentencepiece, mpmath, mgds, flatbuffers, antlr4-python3-runtime, zipp, wheel, urllib3, typing-extensions, tqdm, tensorboard-data-server, sympy, six, schedulefree, safetensors, regex, PyYAML, pynvml, pygments, pyasn1, psutil, protobuf, prodigyopt, platformdirs, pillow, packaging, oauthlib, nvidia-nvtx-cu11, nvidia-nccl-cu11, nvidia-cusparse-cu11, nvidia-curand-cu11, nvidia-cufft-cu11, nvidia-cuda-runtime-cu11, nvidia-cuda-nvrtc-cu11, nvidia-cuda-cupti-cu11, nvidia-cublas-cu11, numpy, networkx, multidict, mdurl, MarkupSafe, markdown, idna, humanfriendly, grpcio, ftfy, fsspec, frozenlist, filelock, darkdetect, dadaptation, cloudpickle, charset-normalizer, certifi, cachetools, attrs, async-timeout, absl-py, yarl, werkzeug, triton, scipy, rsa, requests, PyWavelets, pyasn1-modules, opencv-python, omegaconf, nvidia-cusolver-cu11, nvidia-cudnn-cu11, markdown-it-py, lightning-utilities, jinja2, importlib-metadata, customtkinter, coloredlogs, aiosignal, torch, rich, requests-oauthlib, pooch, onnxruntime-gpu, huggingface-hub, google-auth, aiohttp, xformers, torchvision, torchmetrics, tokenizers, scalene, lion-pytorch, invisible-watermark, google-auth-oauthlib, diffusers, bitsandbytes, accelerate, transformers, timm, tensorboard, pytorch-lightning, open-clip-torch
  Running setup.py develop for mgds
Successfully installed MarkupSafe-2.1.5 PyWavelets-1.6.0 PyYAML-6.0.1 absl-py-2.1.0 accelerate-0.25.0 aiohttp-3.9.5 aiosignal-1.3.1 antlr4-python3-runtime-4.9.3 async-timeout-4.0.3 attrs-23.2.0 bitsandbytes-0.43.0 cachetools-5.3.3 certifi-2024.2.2 charset-normalizer-3.3.2 cloudpickle-3.0.0 coloredlogs-15.0.1 customtkinter-5.2.1 dadaptation-3.2 darkdetect-0.8.0 diffusers-0.27.0.dev0 filelock-3.14.0 flatbuffers-24.3.25 frozenlist-1.4.1 fsspec-2024.3.1 ftfy-6.2.0 google-auth-2.29.0 google-auth-oauthlib-1.2.0 grpcio-1.63.0 huggingface-hub-0.20.3 humanfriendly-10.0 idna-3.7 importlib-metadata-7.1.0 invisible-watermark-0.2.0 jinja2-3.1.3 lightning-utilities-0.11.2 lion-pytorch-0.1.2 markdown-3.6 markdown-it-py-3.0.0 mdurl-0.1.2 mgds-0.1 mpmath-1.3.0 multidict-6.0.5 networkx-3.3 numpy-1.26.2 nvidia-cublas-cu11-11.11.3.6 nvidia-cuda-cupti-cu11-11.8.87 nvidia-cuda-nvrtc-cu11-11.8.89 nvidia-cuda-runtime-cu11-11.8.89 nvidia-cudnn-cu11-8.7.0.84 nvidia-cufft-cu11-10.9.0.58 nvidia-curand-cu11-10.3.0.86 nvidia-cusolver-cu11-11.4.1.48 nvidia-cusparse-cu11-11.7.5.86 nvidia-nccl-cu11-2.19.3 nvidia-nvtx-cu11-11.8.86 oauthlib-3.2.2 omegaconf-2.3.0 onnxruntime-gpu-1.16.3 open-clip-torch-2.23.0 opencv-python-4.8.1.78 packaging-24.0 pillow-10.2.0 platformdirs-4.2.1 pooch-1.8.0 prodigyopt-1.0 protobuf-4.23.4 psutil-5.9.8 pyasn1-0.6.0 pyasn1-modules-0.4.0 pygments-2.18.0 pynvml-11.5.0 pytorch-lightning-2.1.3 regex-2024.4.28 requests-2.31.0 requests-oauthlib-2.0.0 rich-13.7.1 rsa-4.9 safetensors-0.4.1 scalene-1.5.39 schedulefree-1.2.5 scipy-1.11.4 sentencepiece-0.2.0 six-1.16.0 sympy-1.12 tensorboard-2.15.1 tensorboard-data-server-0.7.2 timm-0.9.16 tokenizers-0.15.2 torch-2.2.0+cu118 torchmetrics-1.3.2 torchvision-0.17.0+cu118 tqdm-4.66.1 transformers-4.36.2 triton-2.2.0 typing-extensions-4.11.0 urllib3-2.2.1 wcwidth-0.2.13 werkzeug-3.0.2 wheel-0.43.0 xformers-0.0.24+cu118 yarl-1.9.4 zipp-3.18.1

[notice] A new release of pip is available: 23.0.1 -> 24.0
[notice] To update, run: python -m pip install --upgrade pip
(py3.10) ╭─zen at arch in ~/m2a/ai-img/OneTrainer on master✘✘✘ 24-05-05 - 13:18:26
╰─(py3.10) ⠠⠵ ./start-ui.sh 
conda not found; python version correct; use native python
Traceback (most recent call last):
  File "/home/zen/m2a/ai-img/OneTrainer/scripts/train_ui.py", line 5, in <module>
    from modules.ui.TrainUI import TrainUI
  File "/home/zen/m2a/ai-img/OneTrainer/modules/ui/TrainUI.py", line 9, in <module>
    import customtkinter as ctk
ModuleNotFoundError: No module named 'customtkinter'

What did you expect would happen?

I expected the GUI to show up.

Relevant log output

No response

Output of pip freeze

absl-py==2.1.0 accelerate==0.25.0 aiohttp==3.9.5 aiosignal==1.3.1 antlr4-python3-runtime==4.9.3 async-timeout==4.0.3 attrs==23.2.0 bitsandbytes==0.43.0 cachetools==5.3.3 certifi==2024.2.2 charset-normalizer==3.3.2 cloudpickle==3.0.0 coloredlogs==15.0.1 customtkinter==5.2.1 dadaptation==3.2 darkdetect==0.8.0 -e git+https://github.com/huggingface/diffusers.git@5d848ec07c2011d600ce5e5c1aa02a03152aea9b#egg=diffusers filelock==3.14.0 flatbuffers==24.3.25 frozenlist==1.4.1 fsspec==2024.3.1 ftfy==6.2.0 google-auth==2.29.0 google-auth-oauthlib==1.2.0 grpcio==1.63.0 huggingface-hub==0.20.3 humanfriendly==10.0 idna==3.7 importlib_metadata==7.1.0 invisible-watermark==0.2.0 Jinja2==3.1.3 lightning-utilities==0.11.2 lion-pytorch==0.1.2 Markdown==3.6 markdown-it-py==3.0.0 MarkupSafe==2.1.5 mdurl==0.1.2 -e git+https://github.com/Nerogar/mgds.git@1dc300967e75b6fa0fb4b72587f3df08a8278efd#egg=mgds mpmath==1.3.0 multidict==6.0.5 networkx==3.3 numpy==1.26.2 nvidia-cublas-cu11==11.11.3.6 nvidia-cuda-cupti-cu11==11.8.87 nvidia-cuda-nvrtc-cu11==11.8.89 nvidia-cuda-runtime-cu11==11.8.89 nvidia-cudnn-cu11==8.7.0.84 nvidia-cufft-cu11==10.9.0.58 nvidia-curand-cu11==10.3.0.86 nvidia-cusolver-cu11==11.4.1.48 nvidia-cusparse-cu11==11.7.5.86 nvidia-nccl-cu11==2.19.3 nvidia-nvtx-cu11==11.8.86 oauthlib==3.2.2 omegaconf==2.3.0 onnxruntime-gpu==1.16.3 open-clip-torch==2.23.0 opencv-python==4.8.1.78 packaging==24.0 pillow==10.2.0 platformdirs==4.2.1 pooch==1.8.0 prodigyopt==1.0 protobuf==4.23.4 psutil==5.9.8 pyasn1==0.6.0 pyasn1_modules==0.4.0 Pygments==2.18.0 pynvml==11.5.0 pytorch-lightning==2.1.3 PyWavelets==1.6.0 PyYAML==6.0.1 regex==2024.4.28 requests==2.31.0 requests-oauthlib==2.0.0 rich==13.7.1 rsa==4.9 safetensors==0.4.1 scalene==1.5.39 schedulefree==1.2.5 scipy==1.11.4 sentencepiece==0.2.0 six==1.16.0 sympy==1.12 tensorboard==2.15.1 tensorboard-data-server==0.7.2 timm==0.9.16 tokenizers==0.15.2 torch==2.2.0+cu118 torchmetrics==1.3.2 torchvision==0.17.0+cu118 tqdm==4.66.1 transformers==4.36.2 triton==2.2.0 typing_extensions==4.11.0 urllib3==2.2.1 wcwidth==0.2.13 Werkzeug==3.0.2 xformers==0.0.24+cu118 yarl==1.9.4 zipp==3.18.1

dathide avatar May 05 '24 20:05 dathide

I'll start you on the right path conda not found; python version correct; use native python For some reason it seems it's trying to use a conda env instead of the venv -- I don't use linux but I imagine the sh should be automatically switching into the venv environment, and it seems that the installer installed in a venv

SirTrippsalot avatar May 06 '24 00:05 SirTrippsalot

conda not found is unrelated. its just trying anaconda before python venv. it switches automatically to the right option. try installing google-perftools via apt.

yggdrasil75 avatar May 06 '24 13:05 yggdrasil75

This is working for me at the moment. UI starts up. Haven't trained yet so don't know if any conflicts but:

I made a copy of the start up file (called it start-ui-alt.sh) and then replaced:

image (start-ui.sh)

With: image (start-ui-alt.sh the copy I created to keep the original intact)

Here's the code for copy paste (replace lines 43 to 50):

# Check for venv in the current directory
if [ -d "venv" ]; then
    # Activate the local venv
    source venv/bin/activate
    # Run the script
    python scripts/train_ui.py
else 
    echo "venv not found in the current directory."
    echo "Please create and activate the virtual environment before running the script."
fi

Didn't make a pull request cause it's just a quick and dirty solution. May find something more proper later.

C0nsumption avatar May 08 '24 20:05 C0nsumption

Windows 11, the startui.bat gives me this:

activating venv C:\Users\AlexMazurek\Videos\Experimenting\OneTrainer\venv Using Python "C:\Users\AlexMazurek\Videos\Experimenting\OneTrainer\venv\Scripts\python.exe" Traceback (most recent call last): File "C:\Users\AlexMazurek\Videos\Experimenting\OneTrainer\scripts\train_ui.py", line 5, in from modules.ui.TrainUI import TrainUI File "C:\Users\AlexMazurek\Videos\Experimenting\OneTrainer\modules\ui\TrainUI.py", line 9, in import customtkinter as ctk ModuleNotFoundError: No module named 'customtkinter' Press any key to continue . . .

Out of my depth here, I don't know anything about python. Vaguely, it seems to maybe have something to do with my pip installation? I don't know. Would love some guidance.

aurmza avatar May 11 '24 04:05 aurmza

I would guess that you are using python 3.12. OneTrainer should work ok with 3.11, but 3.10 is still the go to Python for most machine learning programs right now.

Calamdor avatar May 11 '24 05:05 Calamdor

If I did update my Python, I didn't mean to. Been running ComfyUI for weeks. The only thing for sure I updated was pip. I'll try reinstalling and see what happens.

aurmza avatar May 11 '24 12:05 aurmza

Reinstalled 3.10.6, same issue. Also re-cloned the OneTrainer just for laughs. I'm not really sure what to even dig into.

aurmza avatar May 11 '24 14:05 aurmza

When you try python --version at the command prompt you get 3.10.6? (Screenshot from my PC, running 3.10.11) You have run install.bat and did not get any errors? If both of these things check out, then I am out of ideas unfortunately. image

Calamdor avatar May 11 '24 14:05 Calamdor

3.10.6 is definitely the python version. Since you mentioned it, I redid the install.bat, and I guess I wasn't paying good enough attention, because it definitely crashes and closes itself. Guess I assumed it was just done.

Not sure where I could find a log to what just happened, but yeah, I'll update my python versioning.

aurmza avatar May 11 '24 14:05 aurmza

Better to run the install.bat from the command prompt if it is closing before you can see the error. Normally, install.bat waits for user input before closing if successful.

Calamdor avatar May 11 '24 14:05 Calamdor

I had the problem as well resulting from not using a venv as set in start-ui.sh line 6. What also didn't work is using a symbolic link from local venv to the correct path. I ended up just commenting out the check whether variable $python_venv is a folder and then it worked.

I would recommend to have the start up script tell the user it is creating a new venv and ask if it should run the installer instead of just creating the venv and then failing.

jkrauss82 avatar Jun 16 '24 14:06 jkrauss82

@jkrauss82 @dathide Please nuke your installations and reinstall. We have significantly improved the install process especially for linux. This should no longer be an issue, if it it please ping me and I will reopen

O-J1 avatar Oct 13 '24 16:10 O-J1