Модель Stable Diffusion Dreambooth отправляется в Hub

Question

Я настроил модель стабильной диффузии с помощью Dreambooth в Colab. Теперь новая Модель с гирями находится здесь./content/my_model/facial-sketch/800Я также получил.ckptфайл. Как мне теперь перенести эту модель в HuggingFace Hub? Я попробовал методомpush_to_hub("Path"), но это не сработало. Также я не мог бы использовать аргумент--push_to_hubво время обучения. Есть ли у них какой-нибудь способ протолкнуть этот путь?

Answer 1

Вы можете использовать библиотеку Python HuggingFace Hub.

Сначала установите его и войдите:

      pip install -U huggingface_hub
huggingface-cli login

Предполагая, что.ckptфайл находится по адресу/content/my_model/facial-sketch/800.ckptЗапустите следующее в новой ячейке:

      from huggingface_hub import HfApi
api = HfApi()
api.upload_file(
    path_or_fileobj="/content/my_model/facial-sketch/800.ckpt",
    path_in_repo="model.ckpt",
    repo_id="username/mymodel",
    repo_type="model",
)

Вы можете создать модель на сайте hf.co/models под своей учетной записью. Затем изменитеusername/mymodelдля пути вашей модели. (бывший.:https://huggingface.co/johndoe/yogurtбыло быjohndoe/yogurt)