dreambooth-replicate
v1.3.0
Published
```javascript npm install @runvnc/dreambooth-replicate ```
Downloads
5
Readme
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()