Instructions to use kvablack/ddpo-alignment with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use kvablack/ddpo-alignment with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("kvablack/ddpo-alignment", dtype=torch.bfloat16, device_map="cuda") prompt = "a horse playing chess" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
Kevin Black commited on
Commit ·
9ee26ab
1
Parent(s): 4849d80
update checkpoint
Browse files- model_index.json +1 -1
- unet/diffusion_pytorch_model.bin +1 -1
model_index.json
CHANGED
|
@@ -12,7 +12,7 @@
|
|
| 12 |
],
|
| 13 |
"scheduler": [
|
| 14 |
"diffusers",
|
| 15 |
-
"
|
| 16 |
],
|
| 17 |
"text_encoder": [
|
| 18 |
"transformers",
|
|
|
|
| 12 |
],
|
| 13 |
"scheduler": [
|
| 14 |
"diffusers",
|
| 15 |
+
"DDIMScheduler"
|
| 16 |
],
|
| 17 |
"text_encoder": [
|
| 18 |
"transformers",
|
unet/diffusion_pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3438366373
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:510278605ca9242b6a3d20d1520c6894e06f6c7e92f37a4a58fb6b8cbeff306f
|
| 3 |
size 3438366373
|