Text-to-Image
Diffusers
Safetensors
PEFT
image-to-image
image-editing
lora
reinforcement-learning
rubric-policy-optimization
auto-rubric
Instructions to use OpenEnvisionLab/Auto-Rubric-as-Reward with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use OpenEnvisionLab/Auto-Rubric-as-Reward with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev,Qwen/Qwen-Image-Edit", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("OpenEnvisionLab/Auto-Rubric-as-Reward") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - PEFT
How to use OpenEnvisionLab/Auto-Rubric-as-Reward with PEFT:
Task type is invalid.
- Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
- DiffusionBee
File size: 1,056 Bytes
cecbc3b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | {
"alpha_pattern": {},
"auto_mapping": {
"base_model_class": "FluxTransformer2DModel",
"parent_library": "diffusers.models.transformers.transformer_flux"
},
"base_model_name_or_path": null,
"bias": "none",
"fan_in_fan_out": false,
"inference_mode": true,
"init_lora_weights": "gaussian",
"layer_replication": null,
"layers_pattern": null,
"layers_to_transform": null,
"loftq_config": {},
"lora_alpha": 32,
"lora_dropout": 0.0,
"megatron_config": null,
"megatron_core": "megatron.core",
"modules_to_save": null,
"peft_type": "LORA",
"r": 16,
"rank_pattern": {},
"revision": null,
"target_modules": [
"ff.net.0.proj",
"ff_context.net.2",
"attn.to_add_out",
"ff.net.2",
"attn.add_v_proj",
"attn.to_k",
"ff_context.net.0.proj",
"attn.add_k_proj",
"attn.add_q_proj",
"attn.to_out.0",
"attn.to_v",
"attn.to_q"
],
"task_type": null,
"use_dora": false,
"use_rslora": false
} |