Ответ на пост «Dreambooth. Делаем аватарки через нейросеть Stable Diffusion»
Пожалуйста, помогите.. Кто-нибудь проверял работу Dreambooth? Мне нужно знать, я тупой, или не всё так безнадёжно https://colab.research.google.com/github/ShivamShrirao/diffu...
Русскоязычная версия https://colab.research.google.com/drive/1ewXM9rkmYMoerXWw8QY...
У меня успешно проходит тренировка нейросети по фотографиям -- это видно в сэмплах.
Но уже в самом конце -- в пункте Inference возникает ошибка в строчке
pipe = StableDiffusionPipeline.from_pretrained(model_path, scheduler=scheduler, safety_checker=None, torch_dtype=torch.float16).to("cuda")
Вид ошибки -- Error no file named model_index.json found in directory
Понятно, что он не находит изначальный файл model_index.json
Поэтому мне так и нужно знать, что у кого-то это работает, чтобы я знал, что дело во мне.
Конечно, тогда бы очень помогло, если бы вы сказали, что надо изменить. Потому что тут не так уж много надо менять -- добавить token с сайта https://huggingface.co , и заменить класс dog на нужные класс моделей -- cat, woman, car или другой необходимый.
Можно ещё заменить zwx на свой идентификатор, но можно и не менять. Пожалуйста, если у вас всё работает, скажите, что вы изменили, чтобы оно работало...
UPD: всё работает, я был просто невнимателен. Сам нашёл ошибку.






