Как создать чат-бот LLaMa 2 с использованием Gradio и Hugging Face в бесплатной среде Colab.
Создание чат-бота LLaMa 2 с помощью Gradio и Hugging Face в Colab.
Благодаря Gradio вы можете создать пользовательский интерфейс чатбота в одной строке кода!
![Изображение, созданное автором с помощью Leonardo.ai. Подсказка: «Человек в традиционной перуанской одежде, в глубоком разговоре с умной ламой, в фоне Перуанские Анды»](https://miro.medium.com/v2/resize:fit:640/format:webp/1*vDeolFWlmG1SPABTrxIfwQ.jpeg)
Ландшафт открытых моделей больших языков очень интересен.
На момент написания этой статьи модели LLaMa 2 получают больше всего внимания. Поэтому стоит научиться общаться с этими моделями.
В этой статье вы узнаете, как создать чатбота с помощью Gradio ChatInterface
и Hugging Face. И вы сможете запустить код в бесплатной записной книжке Colab!
Поскольку мы рассмотрим каждый шаг, этот учебник подойдет для начинающих. Однако опытные инженеры также многое узнают, так как мы говорим о передовых технологиях, таких как Llama 2.
- Время 100 ИИ Самые влиятельные?
- Понимание алгоритмического предвзятости типы, причины и примеры случаев
- Могут ли трансформеры научиться стратегировать?
Рабочий процесс
В этом пошаговом руководстве мы пройдем следующие шаги:
- Установка: Начнем с настройки нашей среды с необходимыми библиотеками.
- Предварительные требования: Убедитесь, что у вас есть доступ к модели Llama-2 7B на Hugging Face.
- Загрузка модели и токенизатора: Получите модель и токенизатор для нашей сессии.
- Создание конвейера Llama: Подготовьте нашу модель для генерации ответов.
- Форматирование подсказки для Llama 2: Подготовьте сообщения, чтобы следовать правильной структуре подсказок.
- Генерация ответов Llama: Создайте функцию для получения ответов на основе истории вашего чата.
- Взаимодействие с Llama через ChatInterface Gradio: Запросите модель для получения ответов и исследуйте ее возможности.
Примечание: В конце этой статьи вы найдете все полезные ссылки, включая готовую записную книжку Colab с описанием проекта (и видеоверсию этого руководства).
Погрузимся и повеселимся!
Предварительные требования.
Прежде чем перейти к коду, вам потребуется потратить 2 минуты на выполнение этих 3 необходимых шагов:
- Убедитесь, что вы переключили свою среду выполнения Colab на GPU для оптимальной производительности. В верхнем меню перейдите к
Среда выполнения -> Изменить тип среды выполнения
и выберите «T4 GPU». - Создайте учетную запись Hugging Face, чтобы получить доступ…