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 для всех сред, в которых я программирую (включая виртуальные). Может ли кто-нибудь помочь мне? Заранее большое спасибо!