Это точно настроенная модель DialoGPT (small) в наборе данных AfriWOZ. Он предназначен для использования в качестве разговорной системы в нигерийском английском языке Pidgin. Набор данных, на котором он обучен, ограничен по объему, поскольку он охватывает только определенные области, такие как рестораны, гостиницы, такси и бронирование. Пожалуйста, найдите информацию о предварительной обработке, обучении и полной информации о DialoGPT в оригинальном репозитории DialoGPT. Документ для этой работы можно найти на arXiv: https://arxiv.org/pdf/2204.08083.pdf Теперь мы готовы попробовать, как работает модель в качестве чат-партнера! python from transformers import AutoModelForCausalLM, AutoTokenizer import torch tokenizer = AutoTokenizer.frompretrained («tosin/dialogptafriwozpidgin») model = AutoModelForCausalLM.frompretrained («tosin/dialogptafriwoz_pidgin») newuserinputids = tokenizer.encode (input(«>> User:») tokenizer.eostoken, return_tensors=’pt’) botinputids = torch.cat ([chathistoryids, newuserinputids], dim=-1) if step > 0 else newuserinputids chathistoryids = model.generate (botinputids, maxlength
Модальности:
Генерация текста
Области применения:
Диалог / чат
Задача: Генерация текста
Автор: tosin
Теги: gpt2, conversational, en, text-generation-inference, endpoints_compatible
Лайков: 2 | Загрузок: 82
Описание основано на материалах HuggingFace. Перевод выполнен автоматически.
