search

hollowstrawberry / kohya-colab

Accessible Google Colab notebooks for Stable Diffusion Lora training, based on the work of kohya-ss and Linaqruf
GNU General Public License v3.0
590 stars 82 forks source link

Asking for a session restart #174

Closed SAC020 closed 3 weeks ago

SAC020 commented 1 month ago

During setup it asks for a runtime restart to update huggingface.

Ignoring the message ("cancel") continues the training without problems.

Restarting the runtime resets everything and then just repeats when the code gets re-run, resulting in an infinite loop

Building wheels for collected packages: dadaptation Building wheel for dadaptation (pyproject.toml) ... done Created wheel for dadaptation: filename=dadaptation-3.1-py3-none-any.whl size=22789 sha256=c7cc146ea1877a02d48e4dd85d4f956bbddcc11b441bdc43b685c3274f2f5c22 Stored in directory: /root/.cache/pip/wheels/34/25/92/b7708738900a69a166f41de56c776039b5a05599b36075c34f Successfully built dadaptation Installing collected packages: smmap, setproctitle, sentry-sdk, prodigyopt, docker-pycreds, dadaptation, huggingface-hub, gitdb, gitpython, wandb, lion-pytorch, invisible-watermark, timm, open-clip-torch Attempting uninstall: huggingface-hub Found existing installation: huggingface-hub 0.23.4 Uninstalling huggingface-hub-0.23.4: Successfully uninstalled huggingface-hub-0.23.4 Successfully installed dadaptation-3.1 docker-pycreds-0.4.0 gitdb-4.0.11 gitpython-3.1.43 huggingface-hub-0.23.1 invisible-watermark-0.2.0 lion-pytorch-0.1.2 open-clip-torch-2.20.0 prodigyopt-1.0 sentry-sdk-2.10.0 setproctitle-1.3.3 smmap-5.0.1 timm-1.0.7 wandb-0.17.4 WARNING: The following packages were previously imported in this runtime: [huggingface_hub] You must restart the runtime in order to use newly installed versions.

hollowstrawberry commented 1 month ago

I've been avoiding this issue as you can cancel the window instead of restarting, but it's quite annoying and confusing, so I just pushed a simple fix. Let me know if that worked.