Как заменить устаревший «разговорный» конвейер в Hugging Face Transformers? ⇐ Python

[Anonymous]

Я работаю над чат-ботом, используя библиотеку Трансформеров Hugging Face. Раньше я использовал диалоговый конвейер, но, похоже, он устарел. Вот код, который я использовал:

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

# Initialize the conversational pipeline
conversation = pipeline("conversational", model=model, tokenizer=tokenizer)

# Define the chat history
chat_history = [
{"role": "system", "content": "You are a smart chatbot"},
{"role": "user", "content": "What is the capital of France?"},
]

# Generate a response
response = conversation(chat_history)
print(response)

вызовите эту ошибку:

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

KeyError: "Unknown task conversational, available tasks are ['audio-classification', 'automatic-speech-recognition', 'depth-estimation', 'document-question-answering', 'feature-extraction', 'fill-mask', 'image-classification', 'image-feature-extraction', 'image-segmentation', 'image-to-image', 'image-to-text', 'mask-generation', 'ner', 'object-detection', 'question-answering', 'sentiment-analysis', 'summarization', 'table-question-answering', 'text-classification', 'text-generation', 'text-to-audio', 'text-to-speech', 'text2text-generation', 'token-classification', 'translation', 'video-classification', 'visual-question-answering', 'vqa', 'zero-shot-audio-classification', 'zero-shot-classification', 'zero-shot-image-classification', 'zero-shot-object-detection', 'translation_XX_to_YY']"

Теперь, когда диалоговый конвейер устарел, я не знаю, как добиться той же функциональности. Я читал, что мне может потребоваться напрямую использовать конвейер генерации текста или метод генерации модели, но я не уверен, как реализовать это с моей существующей структурой истории чата.
Итак, каков рекомендуемый способ создания диалогового чат-бота с использованием последней версии Transformers?
Будем очень признательны за любые рекомендации или примеры кода!

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