lucataco / diffusers-dreambooth-lora

Huggingface Diffusers FLUX.1 [dev] DreamBooth Training (WIP)

  • Public
  • 80 runs
  • GitHub
  • License

Input

Output

Run time and cost

This model runs on Nvidia A100 (80GB) GPU hardware. We don't yet have enough runs of this model to provide performance information.

Readme

Huggingface - Diffusers Dreambooth LoRA Trainer (Work in progress)

About

Cog implementation of the Diffusers Dreambooth LoRA Trainer

Follow me on Twitter/X @lucataco93

How to use

Upload images and train a LoRA

If you want to upload your trained LoRA to Huggingface, be sure to add a HF_token that has write access and choose a model location.

An example of an uploaded LoRA can be found here

Run your LoRA

Copy the URL from the trained_model.tar from the prediction output, and use the lucataco/flux-dev-lora Explorer model to run your LoRA

An example of a prediction run on a LoRA can be found here

Licensing and commercial use

If you generate images on Replicate with FLUX.1 models and their fine-tunes, then you can use the images commercially.

If you download the weights off Replicate and generate images on your own computer, you can’t use the images commercially.