PrunaVAED: более быстрый VAE-декодер для LTX-2.3 от Pruna AI
PrunaVAED — это заменяемый VAE-декодер для LTX-2.3, обеспечивающий в 1.7 раза более быстрое декодирование и снижение пикового потребления VRAM примерно на 50%.
Модель PrunaVAED напрямую заменяет видеодекодер VAE в diffusers/LTX-2.3-Diffusers. Кодировщик и формат латентного представления остаются неизменными, что делает ее обновлением без изменения конфигурации: никаких изменений в рабочем процессе не требуется.
Бенчмарк
Метрики сравнивают видео, декодированные из одинаковых латентных представлений с помощью VAED LTX-2.3 и PrunaVAED на NVIDIA H100 80GB (bfloat16, размер пакета 1, полное декодирование без тайлинга или torch.compile):
| Длина клипа | VAED LTX-2.3 | PrunaVAED | Ускорение |
|---|---|---|---|
| 5 с @ 720p | 841,3 мс | 500,2 мс | 1.68× |
| 10 с @ 720p | 1670,7 мс | 998,0 мс | 1.67× |
Сравнение бок о бок
На следующих видео сравнивается оригинальный декодер LTX-2.3 (слева) с PrunaVAED (справа) на одних и тех же латентных представлениях:
| Декодер LTX-2.3 | PrunaVAED |
|---|---|
Поддержка в ComfyUI
PrunaVAED поддерживается в ComfyUI начиная с коммита c38171d (Support Pruna LTX VAE). Узлы-обертки ComfyUI-LTXVideo от Lightricks автоматически определяют и загружают декодер PrunaVAED, если файл модели помещен в правильную директорию.
Использование
Скачайте safetensors PrunaVAED с HuggingFace и поместите его в папку models/vae/ вашего ComfyUI. Узлы LTXVideo в ComfyUI автоматически будут использовать его в качестве декодера — никакой дополнительной настройки не требуется.
Ссылки
- Модель на HuggingFace: suryatmodulus/PrunaVAED (также доступна по адресу PrunaAI/PrunaVAED)
- GitHub Pruna AI: github.com/PrunaAI/pruna
- Базовая модель: diffusers/LTX-2.3-Diffusers
- Лицензия: LTX-2 Community License
Комментарии
Войдите через GitHub, чтобы участвовать в обсуждении.