apple/ml-stable-diffusion

Stable Diffusion с Core ML на Apple Silicon

Vision⭐ 17 969Pythonпоследний релиз: 1.1.1
Открыть на GitHub ↗

Что это за инструмент

Инструмент для запуска моделей Stable Diffusion на устройствах Apple Silicon с использованием Core ML, обеспечивающий локальную генерацию изображений.

Зачем нужен

Позволяет конвертировать PyTorch-модели в формат Core ML для эффективного выполнения на чипах Apple (M1, M2, A14 и новее). Полезен для интеграции генерации изображений в нативные iOS/iPadOS/macOS приложения без необходимости использования облачных API.

Что можно реализовать

  • Создание нативных мобильных приложений с функцией генерации изображений по текстовому описанию
  • Локальная обработка медиа-контента на устройствах Apple без отправки данных в облако
  • Прототипирование AI-функций в Swift-проектах с использованием Xcode
  • Офлайн-генерация графики для творческих инструментов на базе macOS

Ключевые возможности

  • Поддержка конвертации моделей из Hugging Face diffusers в Core ML
  • Готовый Swift-пакет для интеграции в Xcode-проекты
  • Оптимизация производительности через сжатие весов до 6 бит и выбор вычислительных блоков (CPU/Neural Engine)
  • Работа на устройствах с чипами M1, M2, A14 и новее
  • Доступные бенчмарки производительности для различных моделей (SD 2.1, SDXL) на iPhone и iPad

👤 Кому подойдёт: iOS/macOS разработчики, специалисты по машинному обучению, работающие с экосистемой Apple, создатели мобильных приложений с AI-функциями

Релизы

1.1.1minorbreaking
🕐 29 мес назад · релиз на GitHub ↗

Релиз 1.1.1: смена лицензии на MIT, исправления совместимости с Python 3.8 и типов данных, а также поддержка смешанной битовой компрессии.

  • Breaking: Лицензия проекта изменена на MIT.
  • Added: Добавлена поддержка скомпилированных моделей в Python и применение смешанной битовой компрессии через новый API оптимизации.
  • Fixed: Исправлена ошибка с типом данных (dtype) в safety_checker.
  • Fixed: Устранены проблемы с отсутствующими параметрами и импортами, связанными с анализом смешанной битовой компрессии.
  • Fixed: Исправлена совместимость pipeline.py с Python 3.8.