Нейросеть Stable Diffusion 2 ускорили в 20+ раз при поддержке Nvidia

Не слишком удивимся, если окажется, что скорость растет только на новых видеокартах.

Компания Stability AI, занимающаяся развитием нейросети Stable Diffusion, добилась значительного ускорения работы нейросети. По оценке разработчиков, скорость выросла как минимум в 20 раз, конвергенция достигается за 1-4 шага.

Обработка 50 шагов, раньше занимавшая 5.6 секунды, теперь требует только 0.9 секунды. Как объясняет основатель Stability AI, Эмад Мостак, скорость выросла не в 6+ раз, а больше, потому что для генерации приличного изображения больше не нужно считать 50 шагов.

Помимо сотрудников Stability AI над проектом работали разработчики из Nvidia. Пока Stability AI опубликовала только научную работу, посвященную ускорению нейросети, на идущей сейчас конференции NeurIPS о проекте расскажут подробнее.

Быстрая модель, созданная Stability AI, будет выложена в открытый доступ. О железе, необходимом для ее работы, не сообщается; пользователи Stable Diffusion опасаются, что в ускоренной модели есть привязка к новым видеокартам Nvidia.