update readme

This commit is contained in:
Kohya S
2023-07-03 21:06:58 +09:00
parent 2febbfe4b0
commit 6aa62b9b66

View File

@@ -32,6 +32,7 @@ Summary of the feature:
- `--no_half_vae`: Disable the half-precision (mixed-precision) VAE. VAE for SDXL seems to produce NaNs in some cases. This option is useful to avoid the NaNs.
- The image generation during training is now available. However, the VAE for SDXL seems to produce NaNs in some cases when using `fp16`. The images will be black. Currently, the NaNs cannot be avoided even with `--no_half_vae` option. It works with `bf16` or without mixed precision.
- `--weighted_captions` option is not supported yet.
- `--min_timestep` and `--max_timestep` options are added to each training script. These options can be used to train U-Net with different timesteps. The default values are 0 and 1000.
`requirements.txt` is updated to support SDXL training.