Skip to content

Commit faeacd5

Browse files
author
Nathan Lambert
authored
Refactoring RL PR (#1200)
* init file changes * add import utils * finish cleaning files, imports * remove import flags * clean examples * fix imports, tests for merge * update readmes
1 parent d5eedff commit faeacd5

15 files changed

+65
-672
lines changed

docs/source/api/models.mdx

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,15 @@ The models are built on the base class ['ModelMixin'] that is a `torch.nn.module
2222
## UNet2DOutput
2323
[[autodoc]] models.unet_2d.UNet2DOutput
2424

25-
## UNet1DModel
26-
[[autodoc]] UNet1DModel
27-
2825
## UNet2DModel
2926
[[autodoc]] UNet2DModel
3027

28+
## UNet1DOutput
29+
[[autodoc]] models.unet_1d.UNet1DOutput
30+
31+
## UNet1DModel
32+
[[autodoc]] UNet1DModel
33+
3134
## UNet2DConditionOutput
3235
[[autodoc]] models.unet_2d_condition.UNet2DConditionOutput
3336

@@ -37,12 +40,6 @@ The models are built on the base class ['ModelMixin'] that is a `torch.nn.module
3740
## DecoderOutput
3841
[[autodoc]] models.vae.DecoderOutput
3942

40-
## UNet1DModel
41-
[[autodoc]] UNet1DModel
42-
43-
## UNet1DOutput
44-
[[autodoc]] models.unet_1d.UNet1DOutput
45-
4643
## VQEncoderOutput
4744
[[autodoc]] models.vae.VQEncoderOutput
4845

examples/README.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,10 @@ If you feel like another important example should exist, we are more than happy
3636

3737
Training examples show how to pretrain or fine-tune diffusion models for a variety of tasks. Currently we support:
3838

39-
| Task | 🤗 Accelerate | 🤗 Datasets | Colab
40-
|---|---|:---:|:---:|
39+
| Task | 🤗 Accelerate | 🤗 Datasets | Colab
40+
|---------------------------------------------------------------------------------------------------------------------------------------------------------|---|:---:|:---:|
4141
| [**Unconditional Image Generation**](https://github.com/huggingface/diffusers/blob/main/examples/unconditional_image_generation/train_unconditional.py) | ✅ | ✅ | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/training_example.ipynb)
42+
| [**Reinforcement Learning for Control**](https://github.com/huggingface/diffusers/blob/main/examples/rl/run_diffusers_locomotion.py) | | | coming soon.
4243

4344
## Community
4445

examples/community/value_guided_diffuser.py

Lines changed: 0 additions & 108 deletions
This file was deleted.

examples/diffuser/run_diffuser.py

Lines changed: 0 additions & 122 deletions
This file was deleted.

examples/diffuser/run_diffuser_value_guided.py

Lines changed: 0 additions & 69 deletions
This file was deleted.

0 commit comments

Comments
 (0)