Проблема с развертыванием проекта MGIE — KeyError: 'emb' ⇐ Python

[Anonymous]

Мы успешно достигли последнего шага в файле demo.ipynd, но столкнулись с препятствием из-за ошибки KeyError: 'emb'. В частности, после обучения с использованием предоставленного вами набора данных и предварительно обученных моделей мы импортировали обученный файл mllm.pt. Однако похоже, что в параметрах обученной модели отсутствует поле «emb», что приводит к сбою следующей строки:

Код: Выделить всё

EMB = ckpt['emb'].cuda()

Отслеживание ошибок выглядит следующим образом:

Код: Выделить всё

KeyError                                  Traceback (most recent call last)
Cell In[3], line 31
     28 image_token_len = (vision_config.image_size//vision_config.patch_size)**2
     30 _ = model.eval()
---> 31 EMB = ckpt['emb'].cuda()
     32 with T.inference_mode(): NULL = model.edit_head(T.zeros(1, 8, 4096).half().to('cuda'), EMB)
     33 print('NULL:', NULL.shape)

KeyError: 'emb'

Мы стремимся решить две проблемы: настройку среды и любые ошибки в процессе. Мы будем очень признательны за любые советы от тех, кто успешно развернул проект MGIE.

Подробнее здесь: https://stackoverflow.com/questions/791 ... yerror-emb