Superprompt

Узел Superprompt, основанный на модели roborovski/superprompt-v1, которая представляет собой модель T5, дообученную на наборе данных SuperPrompt.

Узел Superprompt — это инструмент для расширения текста, который автоматически преобразует короткие и простые текстовые промпты в более подробные и описательные версии. Его основная функция заключается в использовании дообученной языковой модели T5 для предоставления более богатого описания на вход последующим задачам генерации изображений по тексту.

Функциональность узла

Этот узел основан на модели roborovski/superprompt-v1, которая представляет собой модель T5, дообученную на наборе данных SuperPrompt. Он принимает базовый промпт и инструктивный промпт, объединяет их и передает в модель, которая генерирует вывод заданной длины — более подробный текст — в соответствии с инструкцией. При первом использовании узел автоматически загрузит необходимые файлы модели с Hugging Face.

Описание параметров узла - Superprompt

Параметры элементов управления (Parameters)

Название параметра	Тип данных	Обязательный	Значение по умолчанию	Диапазон значений/Опции	Описание
`instruction_prompt`	STRING	Да	Expand the following prompt to add more detail	-	Текст инструкции, указывающий модели, как расширить промпт. Инструкция по умолчанию: "Expand the following prompt to add more detail".
`prompt`	STRING	Да	-	-	Исходный текстовый промпт, который необходимо расширить и улучшить. Этот параметр обычно получает текстовый ввод от других узлов.
`max_new_tokens`	INT	Да	128	1 - 4096 (шаг: 1)	Управляет максимальной длиной нового текста, генерируемого моделью. Диапазон значений от 1 до 4096 с шагом 1.

Выходные данные (Output)

Название параметра	Тип данных	Описание
STRING	STRING	Расширенный и улучшенный подробный текстовый промпт.

Сценарии использования

В типичном рабочем процессе AI-рисования вы можете ввести краткий промпт с первоначальной идеей (например, "кот") в этот узел, и он выведет более подробное описание (например, "пушистый рыжий кот, лениво лежащий на подоконнике в солнечном свете"). Этот улучшенный промпт можно напрямую подключить к входному порту для промптов модели генерации изображений по тексту, чтобы сгенерировать изображение с более богатыми деталями, более соответствующее ожиданиям.

Важные замечания

Обратите внимание, что степень детализации и качество сгенерированного текста зависят от параметра max_new_tokens и самого входного промпта. Слишком длинная генерация может привести к избыточности или несвязности содержания. Для загрузки модели этому узлу требуется подключение к интернету, а процесс генерации потребляет определенные вычислительные ресурсы.

Ссылка на исходный код узла Superprompt

Узел Superprompt взят из пакета узлов ComfyUI-KJNodes.