Ответ на пост «Dreambooth. Делаем аватарки через нейросеть Stable Diffusion»⁠⁠1

Пожалуйста, помогите.. Кто-нибудь проверял работу 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: всё работает, я был просто невнимателен. Сам нашёл ошибку.