File size: 913 Bytes
c31750c
 
b897866
 
 
 
 
 
 
 
 
 
 
 
 
c31750c
b897866
3f9c4a9
b897866
df3d7ff
b897866
ad21b83
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
---
license: creativeml-openrail-m
language:
- en
library_name: diffusers
pipeline_tag: text-to-image
tags:
- stable-diffusion
- stable-diffusion-diffusers
- text-to-image
inference:
  parameters:
    num_inference_steps: 50
    guidance_scale: 5.0
    eta: 1.0
---

# ddpo-compressibility

This model was finetuned from [Stable Diffusion v1-4](https:/CompVis/stable-diffusion-v1-4) using [DDPO](https://arxiv.org/abs/2305.13301) and a reward function encouraging images that are JPEG-compressible. See [the project website](https://rl-diffusion.github.io/) for more details.

The model was finetuned for 60 iterations with a batch size of 256 samples per iteration. During finetuning, it was prompted with all of the animals in the [Imagenet-1000](https://deeplearning.cms.waikato.ac.nz/user-guide/class-maps/IMAGENET/) categories (the first 398 categories), but it exhibits some generalization to other prompts.