Text-to-Image
Diffusers
Safetensors
diffusion
image-generation
reinforcement-learning
self-distillation
lora
Instructions to use WeiChow/DiffusionOPSD with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use WeiChow/DiffusionOPSD with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("WeiChow/DiffusionOPSD") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Inference
- Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- Draw Things
- DiffusionBee
Release SD3.5-M HPSv2.1, PickScore, and CLIPScore adapters
Browse files
README.md
CHANGED
|
@@ -41,10 +41,13 @@ By turning image-level rewards into explicit, continually refreshed intermediate
|
|
| 41 |
|
| 42 |
## Released Checkpoints
|
| 43 |
|
| 44 |
-
This repository provides
|
| 45 |
|
| 46 |
| Checkpoint | Backbone | Training objective |
|
| 47 |
|---|---|---|
|
|
|
|
|
|
|
|
|
|
| 48 |
| [`sd35-m-hpsv3`](./sd35-m-hpsv3) | [Stable Diffusion 3.5 Medium](https://huggingface.co/stabilityai/stable-diffusion-3.5-medium) | HPSv3 |
|
| 49 |
| [`z-image-turbo-hpsv3`](./z-image-turbo-hpsv3) | [Z-Image-Turbo](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo) | HPSv3 |
|
| 50 |
| [`z-image-turbo-pointwise`](./z-image-turbo-pointwise) | [Z-Image-Turbo](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo) | Pointwise reward |
|
|
|
|
| 41 |
|
| 42 |
## Released Checkpoints
|
| 43 |
|
| 44 |
+
This repository provides six rank-32 LoRA adapters:
|
| 45 |
|
| 46 |
| Checkpoint | Backbone | Training objective |
|
| 47 |
|---|---|---|
|
| 48 |
+
| [`sd35-m-hpsv2`](./sd35-m-hpsv2) | [Stable Diffusion 3.5 Medium](https://huggingface.co/stabilityai/stable-diffusion-3.5-medium) | HPSv2.1 |
|
| 49 |
+
| [`sd35-m-pickscore`](./sd35-m-pickscore) | [Stable Diffusion 3.5 Medium](https://huggingface.co/stabilityai/stable-diffusion-3.5-medium) | PickScore |
|
| 50 |
+
| [`sd35-m-clipscore`](./sd35-m-clipscore) | [Stable Diffusion 3.5 Medium](https://huggingface.co/stabilityai/stable-diffusion-3.5-medium) | CLIPScore |
|
| 51 |
| [`sd35-m-hpsv3`](./sd35-m-hpsv3) | [Stable Diffusion 3.5 Medium](https://huggingface.co/stabilityai/stable-diffusion-3.5-medium) | HPSv3 |
|
| 52 |
| [`z-image-turbo-hpsv3`](./z-image-turbo-hpsv3) | [Z-Image-Turbo](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo) | HPSv3 |
|
| 53 |
| [`z-image-turbo-pointwise`](./z-image-turbo-pointwise) | [Z-Image-Turbo](https://huggingface.co/Tongyi-MAI/Z-Image-Turbo) | Pointwise reward |
|
sd35-m-clipscore/adapter_config.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"alpha_pattern": {},
|
| 3 |
+
"auto_mapping": {
|
| 4 |
+
"base_model_class": "SD3Transformer2DModel",
|
| 5 |
+
"parent_library": "diffusers.models.transformers.transformer_sd3"
|
| 6 |
+
},
|
| 7 |
+
"base_model_name_or_path": null,
|
| 8 |
+
"bias": "none",
|
| 9 |
+
"eva_config": null,
|
| 10 |
+
"exclude_modules": null,
|
| 11 |
+
"fan_in_fan_out": false,
|
| 12 |
+
"inference_mode": true,
|
| 13 |
+
"init_lora_weights": "gaussian",
|
| 14 |
+
"layer_replication": null,
|
| 15 |
+
"layers_pattern": null,
|
| 16 |
+
"layers_to_transform": null,
|
| 17 |
+
"loftq_config": {},
|
| 18 |
+
"lora_alpha": 64,
|
| 19 |
+
"lora_bias": false,
|
| 20 |
+
"lora_dropout": 0.0,
|
| 21 |
+
"megatron_config": null,
|
| 22 |
+
"megatron_core": "megatron.core",
|
| 23 |
+
"modules_to_save": null,
|
| 24 |
+
"peft_type": "LORA",
|
| 25 |
+
"r": 32,
|
| 26 |
+
"rank_pattern": {},
|
| 27 |
+
"revision": null,
|
| 28 |
+
"target_modules": [
|
| 29 |
+
"attn.to_v",
|
| 30 |
+
"attn.add_q_proj",
|
| 31 |
+
"attn.add_k_proj",
|
| 32 |
+
"attn.add_v_proj",
|
| 33 |
+
"attn.to_k",
|
| 34 |
+
"attn.to_add_out",
|
| 35 |
+
"attn.to_out.0",
|
| 36 |
+
"attn.to_q"
|
| 37 |
+
],
|
| 38 |
+
"task_type": null,
|
| 39 |
+
"use_dora": false,
|
| 40 |
+
"use_rslora": false
|
| 41 |
+
}
|
sd35-m-clipscore/adapter_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a15abf9327a65b02be8cb88307f7db6cefb519005f7173a432dca68b37296cec
|
| 3 |
+
size 75156544
|
sd35-m-hpsv2/adapter_config.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"alpha_pattern": {},
|
| 3 |
+
"auto_mapping": {
|
| 4 |
+
"base_model_class": "SD3Transformer2DModel",
|
| 5 |
+
"parent_library": "diffusers.models.transformers.transformer_sd3"
|
| 6 |
+
},
|
| 7 |
+
"base_model_name_or_path": null,
|
| 8 |
+
"bias": "none",
|
| 9 |
+
"eva_config": null,
|
| 10 |
+
"exclude_modules": null,
|
| 11 |
+
"fan_in_fan_out": false,
|
| 12 |
+
"inference_mode": true,
|
| 13 |
+
"init_lora_weights": "gaussian",
|
| 14 |
+
"layer_replication": null,
|
| 15 |
+
"layers_pattern": null,
|
| 16 |
+
"layers_to_transform": null,
|
| 17 |
+
"loftq_config": {},
|
| 18 |
+
"lora_alpha": 64,
|
| 19 |
+
"lora_bias": false,
|
| 20 |
+
"lora_dropout": 0.0,
|
| 21 |
+
"megatron_config": null,
|
| 22 |
+
"megatron_core": "megatron.core",
|
| 23 |
+
"modules_to_save": null,
|
| 24 |
+
"peft_type": "LORA",
|
| 25 |
+
"r": 32,
|
| 26 |
+
"rank_pattern": {},
|
| 27 |
+
"revision": null,
|
| 28 |
+
"target_modules": [
|
| 29 |
+
"attn.to_k",
|
| 30 |
+
"attn.add_v_proj",
|
| 31 |
+
"attn.add_q_proj",
|
| 32 |
+
"attn.to_add_out",
|
| 33 |
+
"attn.to_v",
|
| 34 |
+
"attn.to_q",
|
| 35 |
+
"attn.add_k_proj",
|
| 36 |
+
"attn.to_out.0"
|
| 37 |
+
],
|
| 38 |
+
"task_type": null,
|
| 39 |
+
"use_dora": false,
|
| 40 |
+
"use_rslora": false
|
| 41 |
+
}
|
sd35-m-hpsv2/adapter_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9bb9cb8bcab44338684733622cb88825c38a5b33e1f6d4e68adb36cd784ee957
|
| 3 |
+
size 75156544
|
sd35-m-pickscore/adapter_config.json
ADDED
|
@@ -0,0 +1,41 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"alpha_pattern": {},
|
| 3 |
+
"auto_mapping": {
|
| 4 |
+
"base_model_class": "SD3Transformer2DModel",
|
| 5 |
+
"parent_library": "diffusers.models.transformers.transformer_sd3"
|
| 6 |
+
},
|
| 7 |
+
"base_model_name_or_path": null,
|
| 8 |
+
"bias": "none",
|
| 9 |
+
"eva_config": null,
|
| 10 |
+
"exclude_modules": null,
|
| 11 |
+
"fan_in_fan_out": false,
|
| 12 |
+
"inference_mode": true,
|
| 13 |
+
"init_lora_weights": "gaussian",
|
| 14 |
+
"layer_replication": null,
|
| 15 |
+
"layers_pattern": null,
|
| 16 |
+
"layers_to_transform": null,
|
| 17 |
+
"loftq_config": {},
|
| 18 |
+
"lora_alpha": 64,
|
| 19 |
+
"lora_bias": false,
|
| 20 |
+
"lora_dropout": 0.0,
|
| 21 |
+
"megatron_config": null,
|
| 22 |
+
"megatron_core": "megatron.core",
|
| 23 |
+
"modules_to_save": null,
|
| 24 |
+
"peft_type": "LORA",
|
| 25 |
+
"r": 32,
|
| 26 |
+
"rank_pattern": {},
|
| 27 |
+
"revision": null,
|
| 28 |
+
"target_modules": [
|
| 29 |
+
"attn.to_out.0",
|
| 30 |
+
"attn.to_v",
|
| 31 |
+
"attn.to_k",
|
| 32 |
+
"attn.add_q_proj",
|
| 33 |
+
"attn.add_k_proj",
|
| 34 |
+
"attn.to_add_out",
|
| 35 |
+
"attn.add_v_proj",
|
| 36 |
+
"attn.to_q"
|
| 37 |
+
],
|
| 38 |
+
"task_type": null,
|
| 39 |
+
"use_dora": false,
|
| 40 |
+
"use_rslora": false
|
| 41 |
+
}
|
sd35-m-pickscore/adapter_model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:18bb4fbe81bd3753c05c587900d2a21fe7c0c4f84a43cd1dd870257951362729
|
| 3 |
+
size 75156544
|