dreambooth-replicate v1.3.0

Install

npm install @runvnc/dreambooth-replicate

Usage

Short example

import DreamboothModel from '@runvnc/dreambooth-replicate'
import delay from 'delay'

async function test() {
  const model = new DreamboothModel({}) // env var: REPLICATE_API_TOKEN

  await model.startTraining({imagesZipUrl:'https://imagehost.com/trainimages.zip',
                             modelName: 'user1-woman1'})
  await delay(60 * 1000 * 30) // wait ~ 20+ minutes

  let imageUrl = await model.generateImage('a painting of a woman named sksmz')
}

test()

Generate image

import DreamboothModel from '@runvnc/dreambooth-replicate'
import delay from 'delay'

async function test() {
  const model = new DreamboothModel({version: 'xxxxxxxxxxxxxxxxxxxxxxxxxxxxx})

  let imageUrl = await model.generateImage('a painting of a woman named sksmz', {}, true)
  console.log(imageUrl)
}

test()

Training status / Options

import DreamboothModel from '@runvnc/dreambooth-replicate'
import delay from 'delay'

async function test() {
  const imagesZipUrl = 'https://imagehost.com/trainimages.zip'

  // optional: trainingId, modelName, instance, className, version, trainingVersion
  const model = new DreamboothModel({apiToken: 'XXXXXXXXXXX'})

  // optional: instance ('a photo of a woman named sksmz')
  // optional: className ('a photo of a woman')
  // optional: maxSteps (2000)
  await model.startTraining({imagesZipUrl, modelName: 'user1-woman1'})
  console.log('Model training Id: '+model.trainingId)

  let status
  do {
    await delay(15 * 1000)
    status = await model.getTrainingStatus()   
    console.log(status)
  } while (status != 'succeeded' && status != 'failed')

  console.log('Training complete. Generating image..')

  // optional: ,waitForCompletion
  const wait = true
  const opts = { steps: 50, seed: 123456, negative_prompt: 'blurry', num_outputs: 4 } 
  let imageUrl = await model.generateImage('a painting of a woman named sksmz', opts, wait)
  console.log('Output image URL: ' + imageUrl)
}


test()