You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: examples/community/README.md
+30Lines changed: 30 additions & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -62,6 +62,7 @@ If a community doesn't work as expected, please open an issue and ping the autho
62
62
| AnimateDiff Image-To-Video Pipeline | Experimental Image-To-Video support for AnimateDiff (open to improvements) |[AnimateDiff Image To Video Pipeline](#animatediff-image-to-video-pipeline)|[](https://drive.google.com/file/d/1TvzCDPHhfFtdcJZe4RLloAwyoLKuttWK/view?usp=sharing)|[Aryan V S](https://github.com/a-r-r-o-w)|
63
63
| IP Adapter FaceID Stable Diffusion | Stable Diffusion Pipeline that supports IP Adapter Face ID |[IP Adapter Face ID](#ip-adapter-face-id)| - |[Fabio Rigano](https://github.com/fabiorigano)|
64
64
| InstantID Pipeline | Stable Diffusion XL Pipeline that supports InstantID |[InstantID Pipeline](#instantid-pipeline)|[](https://huggingface.co/spaces/InstantX/InstantID)|[Haofan Wang](https://github.com/haofanwang)|
65
+
| UFOGen Scheduler | Scheduler for UFOGen Model (compatible with Stable Diffusion pipelines) |[UFOGen Scheduler](#ufogen-scheduler)| - |[dg845](https://github.com/dg845)|
65
66
66
67
To load a custom pipeline you just need to pass the `custom_pipeline` argument to `DiffusionPipeline`, as one of the files in `diffusers/examples/community`. Feel free to send a PR with your own pipelines, we will merge them quickly.
67
68
@@ -3605,3 +3606,32 @@ image = pipe(
3605
3606
controlnet_conditioning_scale=0.8,
3606
3607
).images[0]
3607
3608
```
3609
+
3610
+
### UFOGen Scheduler
3611
+
3612
+
[UFOGen](https://arxiv.org/abs/2311.09257) is a generative model designed for fast one-step text-to-image generation, trained via adversarial training starting from an initial pretrained diffusion model such as Stable Diffusion. `scheduling_ufogen.py` implements a onestep and multistep sampling algorithm for UFOGen models compatible with pipelines like `StableDiffusionPipeline`. A usage example is as follows:
3613
+
3614
+
```py
3615
+
import torch
3616
+
from diffusers import StableDiffusionPipeline
3617
+
3618
+
from scheduling_ufogen import UFOGenScheduler
3619
+
3620
+
#NOTE: currently, I am not aware of any publicly available UFOGen model checkpoints trained from SD v1.5.
3621
+
ufogen_model_id_or_path ="/path/to/ufogen/model"
3622
+
pipe = StableDiffusionPipeline(
3623
+
ufogen_model_id_or_path,
3624
+
torch_dtype=torch.float16,
3625
+
)
3626
+
3627
+
# You can initialize a UFOGenScheduler as follows:
0 commit comments