search

dreamingtulpa / replicate-ruby

Replicate Ruby client
60 stars 6 forks source link

SDXL training support? #10

Open dtbaker opened 5 months ago

dtbaker commented 5 months ago

I'm wondering if the SDXL training follows a similar pattern to the dreambooth training?

Would it be easy to add in?

https://replicate.com/stability-ai/sdxl/train

dreamingtulpa commented 5 months ago

I haven't tested it, but based on their guide, this should work:

upload = Replicate.client.upload_zip('tmp/data.zip') # replace with the path to your zip file

training = Replicate.client.create_training(
  version: "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b",
  input: {
    input_images: upload.serving_url,
  },
  destination: 'yourusername/yourmodel'
)
felixding commented 1 month ago

I tried to create a new training with the above code. Unfortunately it didn't work:

400 Bad Request: {"detail"=>"model not specified"}