Убедитесь, что _init_weights реализован для <класса donut.model.DonutModel>.

Я столкнулся с проблемой с репозиторием пончиков. Я пытаюсь загрузить предварительно обученную модель naver-clova-ix/donut-base-finetuned-cord-v2, но каждый раз получаю одну и ту же ошибку:

Убедитесь, что _init_weights реализован для <класса donut.model.DonutModel>.

P.S. Я уже пробовал клонировать официальную ветку репозитория черезhuggingface.co.

Может ли кто-нибудь объяснить, почему эта ошибка сохраняется или что могло быть не так при клонировании официальной ветки? Предложения будут оценены

введите сюда описание изображения

          model = DonutModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
if torch.cuda.is_available():
    model.half()
    device = torch.device("cuda")
    model.to(device)
else:
    model.encoder.to(torch.bfloat16)
model.eval()
image = Image.open("train/images/0000ae6cbdb1.jpg").convert("RGB")
output = model.inference(image=image, prompt="<s_sroie-donut>")
output
Источник

0 ответов

Другие вопросы по тегам