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
617 stars 93 forks source link

Error when using custom models #238

Open AIuser0101 opened 1 day ago

AIuser0101 commented 1 day ago

Hello, I'm trying to use custom models via google drive, I'm linking it correctly with:

/content/drive/MyDrive/checkpoints/checkpoint.safetensors (the checkpoint folder is where I store all the custom checkpoint files)

I'm also on colab pro using a A100.

The notebook works great with the default checkpoints (except for Illustrious, it technically completes the training without error but it just doesn't show up in forge UI when I'm testing, so I figure I can just use some custom model )

Anyway, I add in the custom model in the field and it seems to run okay but crashes after the log says:

" (star icon) Starting Trainer..."

this is the error code I get:

Traceback (most recent call last): File "/content/kohya-trainer/train_network_xl_wrapper.py", line 2, in from sdxl_train_network import setup_parser, SdxlNetworkTrainer File "/content/kohya-trainer/sdxl_train_network.py", line 3, in from library import sdxl_model_util, sdxl_train_util, train_util File "/content/kohya-trainer/library/sdxl_model_util.py", line 7, in from diffusers import AutoencoderKL, EulerDiscreteScheduler, UNet2DConditionModel File "/usr/local/lib/python3.10/dist-packages/diffusers/init.py", line 3, in from .configuration_utils import ConfigMixin File "/usr/local/lib/python3.10/dist-packages/diffusers/configuration_utils.py", line 34, in from .utils import ( File "/usr/local/lib/python3.10/dist-packages/diffusers/utils/init.py", line 38, in from .dynamic_modules_utils import get_class_from_dynamic_module File "/usr/local/lib/python3.10/dist-packages/diffusers/utils/dynamic_modules_utils.py", line 28, in from huggingface_hub import HfFolder, cached_download, hf_hub_download, model_info

ImportError: cannot import name 'cached_download' from 'huggingface_hub' (/usr/local/lib/python3.10/dist-packages/huggingface_hub/init.py)

=== If anyone knows please let me know. I think it's some dependency that broke maybe? Thanks for the help,

eniiaflannery commented 1 day ago

not only with custom models, even with model default shows the same error :(