update model

README.md

@@ -1,3 +1,40 @@
----
-license: mit
----
+---
+base_model: stabilityai/stable-diffusion-3.5-medium
+library_name: peft
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+This model is trained using Flow-GRPO with LoRA. We provide only the LoRA weights here, so you will need to download the SD 3.5 Medium base model first.
+
+## Model Details
+
+### Model Sources
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** https://github.com/yifan123/flow_grpo
+- **Paper:** https://www.arxiv.org/pdf/2505.05470
+
+## Uses
+```python
+import torch
+from diffusers import StableDiffusion3Pipeline
+from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
+from peft import PeftModel
+
+model_id = "stabilityai/stable-diffusion-3.5-medium"
+lora_ckpt_path = "jieliu/SD3.5M-FlowGRPO-GenEval"
+device = "cuda"
+
+
+pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.transformer = PeftModel.from_pretrained(pipe.transformer, lora_ckpt_path)
+pipe.transformer = pipe.transformer.merge_and_unload()
+pipe = pipe.to(device)
+
+prompt = 'a photo of a black kite and a green bear'
+image = pipe(prompt, height=512, width=512, num_inference_steps=40,guidance_scale=4.5).images[0]  
+image.save(f"flow_grpo.png")
+```

adapter_config.json

@@ -0,0 +1,38 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": {
+    "base_model_class": "SD3Transformer2DModel",
+    "parent_library": "diffusers.models.transformers.transformer_sd3"
+  },
+  "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": 64,
+  "lora_dropout": 0.0,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 32,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "attn.add_v_proj",
+    "attn.to_add_out",
+    "attn.to_k",
+    "attn.add_q_proj",
+    "attn.to_v",
+    "attn.to_q",
+    "attn.add_k_proj",
+    "attn.to_out.0"
+  ],
+  "task_type": null,
+  "use_dora": false,
+  "use_rslora": false
+}

adapter_model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5540d6bb1ebea944a6efa26ff529bbbeb303b2112ab43d090317229e615a3f04
+size 75156544