OSError: Невозможно загрузить веса из файла контрольной точки pytorch при настройке AWS EC2

Я работаю над проектом чатбота, который использует DialoGPT и Django back-end. Я сохранил модель через метод save_pretrained в моей среде colab и решил загрузить их через следующие коды в моем settings.py

tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-medium')
model = AutoModelForCausalLM.from_pretrained("bot/static/friends_model")

Папка friends_model содержит config.json и pytorch_model.bin. Я застрял на том, что когда я запускаю сервер локально на django, чат работает отлично без каких-либо ошибок, но когда я пытаюсь запустить сервер в среде linux ec2, я сталкиваюсь с OSError, упомянутым в заголовке.

Я клонировал git-репозиторий и использовал scp для переноса модели в локальной среде на сервер (EC2). Мое самое смелое предположение - что-то пошло не так во время переноса.

Я использую ОС Windows, Python 3.9 и Pytorch 1.9.0 для всех сред, в которых я программирую (включая виртуальные). Может ли кто-нибудь помочь мне? Заранее большое спасибо!

Вернуться на верх