Skip to content

Commit

Permalink
Merge pull request #1212 from MrTuanDao/patch-2
Browse files Browse the repository at this point in the history
add vae in example
  • Loading branch information
bghira authored Dec 12, 2024
2 parents e5c1cfd + e1b48bc commit 636cd7f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions documentation/quickstart/FLUX.md
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,7 @@ There, you will possibly need to modify the following variables:
- `model_type` - Set this to `lora`.
- `model_family` - Set this to `flux`.
- `pretrained_model_name_or_path` - Set this to `black-forest-labs/FLUX.1-dev`.
- `pretrained_vae_model_name_or_path` - Set this to `black-forest-labs/FLUX.1-dev`.
- Note that you will need to log in to Huggingface and be granted access to download this model. We will go over logging in to Huggingface later in this tutorial.
- `output_dir` - Set this to the directory where you want to store your checkpoints and validation images. It's recommended to use a full path here.
- `train_batch_size` - this should be kept at 1, especially if you have a very small dataset.
Expand Down

0 comments on commit 636cd7f

Please sign in to comment.