phantom8015/stable-diffusion-cjs

Stable diffusion CJS package for Node JS

stable-diffusion-cjs

This is a CJS version of this ES module for Node.js that provides uses Stable Diffusion to generate images from a given prompt using AI.

Installation

npm install stable-diffusion-cjs

Usage

const AI = require("stable-diffusion-cjs")
const fs = require("fs")

let prompt = "A cat"

 AI.generate(prompt, async (result) => {
    if (result.error) {
        console.log(result.error)
        return;
    }
    try {
        for (let i = 0; i < result.results.length; i++) {
            let data = result.results[i].split(",")[1]
            const buffer = Buffer.from(data, "base64")
            const filename = `image_${i + 1}.png`
            fs.writeFileSync(filename, buffer)
        }
    } catch (e) {
        console.log(e)
    }
})

A trick to get better images is to add:

{enhanced}

at the end of your prompt.

Output

stable-diffusion-cjs's Issues

cb is not a function