expo-stable-diffusion

expo-stable-diffusion

Read Full Detailed Guide

❗️ expo-stable-diffusion currently only works on iOS due to the platform's ability to run Stable Diffusion models on Apple Neural Engine!

Introduction

Start by installing the expo-stable-diffusion module into your Expo managed project:

npx expo install expo-stable-diffusion

Configuration

Update iOS Deployment Target

In order for the project to build successfully, you have to set the iOS Deployment Target to 16.2. You can achieve this by installing the expo-build-properties plugin:

npx expo install expo-build-properties

Configure the plugin by adding the following to your app.json:

{
  "expo": {
    "plugins": [
      [
        "expo-build-properties",
        {
          "ios": {
            "deploymentTarget": "16.2"
          }
        }
      ]
    ]
  }
}

Enable Increased Memory Limit

To prevent memory issues, add the Increased Memory Limit capability to your iOS project. Add the following to your app.json:

{
  "expo": {
    "ios": {
      "entitlements": {
        "com.apple.developer.kernel.increased-memory-limit": true
      }
    }
  }
}

Build Your iOS App

npx expo prebuild --clean --platform ios
npx expo run:ios

Usage

After installation and configuration, you can start generating images using expo-stable-diffusion. Here's a basic example:

import * as FileSystem from 'expo-file-system';
import * as ExpoStableDiffusion from 'expo-stable-diffusion';

const MODEL_PATH = FileSystem.documentDirectory + 'Model/stable-diffusion-2-1';
const SAVE_PATH = FileSystem.cacheDirectory + 'GeneratedImages/image.jpeg';

await ExpoStableDiffusion.loadModel(MODEL_PATH);

await ExpoStableDiffusion.generateImage({
  prompt: 'A cat coding at night',
  stepCount: 25,
  savePath: SAVE_PATH,
});

Obtaining Stable Diffusion Models

To use the expo-stable-diffusion module, you need a converted Core ML Stable Diffusion model. You can convert your own model using Apple's official guide or download pre-converted models from Apple's Hugging Face repository or my Hugging Face repository.

Troubleshooting

❗️ The model load time and image generation duration take some time, especially on devices with lower RAM than 6GB! Find more information in Q6 in the FAQ section of the ml-stable-diffusion repo.

Running Stable Diffusion on Lower-End Devices

failed to load ANE model