Skip to content

Refactoring RL PR #1200

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 7 commits into from
Nov 8, 2022
Merged

Refactoring RL PR #1200

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
640ce26
init file changes
natolambert Nov 8, 2022
a0172cf
add import utils
natolambert Nov 8, 2022
69a0d5d
finish cleaning files, imports
natolambert Nov 8, 2022
28031ef
remove import flags
natolambert Nov 8, 2022
42bc177
clean examples
natolambert Nov 8, 2022
1df51a6
fix imports, tests for merge
natolambert Nov 8, 2022
c99cfa8
update readmes
natolambert Nov 8, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 6 additions & 9 deletions docs/source/api/models.mdx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,12 +22,15 @@ The models are built on the base class ['ModelMixin'] that is a `torch.nn.module
## UNet2DOutput
[[autodoc]] models.unet_2d.UNet2DOutput

## UNet1DModel
[[autodoc]] UNet1DModel

## UNet2DModel
[[autodoc]] UNet2DModel

## UNet1DOutput
[[autodoc]] models.unet_1d.UNet1DOutput

## UNet1DModel
[[autodoc]] UNet1DModel

## UNet2DConditionOutput
[[autodoc]] models.unet_2d_condition.UNet2DConditionOutput

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

## UNet1DModel
[[autodoc]] UNet1DModel

## UNet1DOutput
[[autodoc]] models.unet_1d.UNet1DOutput

## VQEncoderOutput
[[autodoc]] models.vae.VQEncoderOutput

Expand Down
5 changes: 3 additions & 2 deletions examples/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,9 +36,10 @@ If you feel like another important example should exist, we are more than happy

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

| Task | 🤗 Accelerate | 🤗 Datasets | Colab
|---|---|:---:|:---:|
| Task | 🤗 Accelerate | 🤗 Datasets | Colab
|---------------------------------------------------------------------------------------------------------------------------------------------------------|---|:---:|:---:|
| [**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)
| [**Reinforcement Learning for Control**](https://github.com/huggingface/diffusers/blob/main/examples/rl/run_diffusers_locomotion.py) | | | coming soon.

## Community

Expand Down
108 changes: 0 additions & 108 deletions examples/community/value_guided_diffuser.py
View file
Open in desktop

This file was deleted.

122 changes: 0 additions & 122 deletions examples/diffuser/run_diffuser.py
View file
Open in desktop

This file was deleted.

69 changes: 0 additions & 69 deletions examples/diffuser/run_diffuser_value_guided.py
View file
Open in desktop

This file was deleted.

Loading