diffusers/src/diffusers
Pedro Cuenca 9f8c915a75
[Dreambooth] flax fixes (#1765)
* Fail if there are less images than the effective batch size.

* Remove lr-scheduler arg as it's currently ignored.

* Make guidance_scale work for batch_size > 1.
2022-12-19 20:42:25 +01:00
..
commands [Commands] Add env command (#352) 2022-09-04 17:43:51 +02:00
experimental Update RL docs for better sharing / adding models (#1563) 2022-12-07 09:08:12 -08:00
models [Versatile] fix attention mask (#1763) 2022-12-19 15:58:39 +01:00
pipelines [Dreambooth] flax fixes (#1765) 2022-12-19 20:42:25 +01:00
schedulers Bump to 0.12.0.dev0 (#1771) 2022-12-19 18:44:08 +01:00
utils Transformers version req for UnCLIP (#1766) 2022-12-19 17:11:17 +01:00
__init__.py Bump to 0.12.0.dev0 (#1771) 2022-12-19 18:44:08 +01:00
configuration_utils.py Allow saving trained betas (#1468) 2022-11-30 10:05:51 +01:00
dependency_versions_check.py upload some cleaning tools 2022-05-31 10:17:19 +02:00
dependency_versions_table.py StableDiffusionDepth2ImgPipeline (#1531) 2022-12-08 18:25:12 +01:00
dynamic_modules_utils.py [Community pipeline] Add github mechanism (#1680) 2022-12-13 18:01:00 +01:00
hub_utils.py Remove all local telemetry (#1702) 2022-12-14 12:56:35 +01:00
modeling_flax_pytorch_utils.py Add `from_pt` argument in `.from_pretrained` (#527) 2022-09-20 12:39:25 +02:00
modeling_flax_utils.py Remove all local telemetry (#1702) 2022-12-14 12:56:35 +01:00
modeling_utils.py Remove all local telemetry (#1702) 2022-12-14 12:56:35 +01:00
onnx_utils.py [ONNX] Improve ONNXPipeline scheduler compatibility, fix safety_checker (#1173) 2022-11-08 14:39:11 +01:00
optimization.py Revive `make quality` (#203) 2022-08-17 15:22:04 +02:00
pipeline_flax_utils.py Remove all local telemetry (#1702) 2022-12-14 12:56:35 +01:00
pipeline_utils.py Remove all local telemetry (#1702) 2022-12-14 12:56:35 +01:00
training_utils.py [Vae and AutoencoderKL] Final clean of LDM checkpoints (#137) 2022-07-28 10:14:34 +02:00