Update README.md
Browse files
README.md
CHANGED
|
@@ -18,7 +18,7 @@ library_name: bagel-mot
|
|
| 18 |
This repository hosts the model weights for **BAGEL**, fine-tuned on the **NHR-Edit** dataset. For installation, usage instructions, and further documentation, please visit the [official BAGEL GitHub repository](https://github.com/bytedance-seed/BAGEL).
|
| 19 |
|
| 20 |
|
| 21 |
-
### Training
|
| 22 |
|
| 23 |
We performed parameter-efficient adaptation on the generation expert’s attention and FFN projection layers using LoRA.
|
| 24 |
|
|
@@ -39,7 +39,7 @@ target_modules = [
|
|
| 39 |
]
|
| 40 |
```
|
| 41 |
|
| 42 |
-
### Image Editing Metrics
|
| 43 |
|
| 44 |
#### Metrics for GEdit-Bench-EN:
|
| 45 |
|
|
@@ -48,7 +48,7 @@ target_modules = [
|
|
| 48 |
| BAGEL-7B-MoT | 7.983 | 6.570 | 6.921 |
|
| 49 |
| **BAGEL-NHR-Edit** | 8.067 | 6.881 | 7.115 |
|
| 50 |
|
| 51 |
-
|
| 52 |
|
| 53 |
#### Metrics for ImgEdit-Bench:
|
| 54 |
|
|
@@ -57,9 +57,9 @@ __NOTE__: Scoring model: __gpt-4.1-2025-04-14__ (with default temperature)
|
|
| 57 |
| BAGEL-7B-MoT | 4.22| 1.53| 3.04| 3.3| 4.07| 3.67| 3.98| 3.5 | 3.0 | 3.3 |
|
| 58 |
| **BAGEL-NHR-Edit** | 4.3| 1.62| 3.18| 3.42| 3.95| 3.55| 4.19| 3.77| 2.94| 3.39|
|
| 59 |
|
| 60 |
-
|
| 61 |
|
| 62 |
-
### Image Editing Results
|
| 63 |
|
| 64 |
Generated images for ImgEdit-Bench and GEdit-Bench are included in this repository.
|
| 65 |
|
|
|
|
| 18 |
This repository hosts the model weights for **BAGEL**, fine-tuned on the **NHR-Edit** dataset. For installation, usage instructions, and further documentation, please visit the [official BAGEL GitHub repository](https://github.com/bytedance-seed/BAGEL).
|
| 19 |
|
| 20 |
|
| 21 |
+
### 🛠️ Training Setup
|
| 22 |
|
| 23 |
We performed parameter-efficient adaptation on the generation expert’s attention and FFN projection layers using LoRA.
|
| 24 |
|
|
|
|
| 39 |
]
|
| 40 |
```
|
| 41 |
|
| 42 |
+
### 📊 Image Editing Metrics
|
| 43 |
|
| 44 |
#### Metrics for GEdit-Bench-EN:
|
| 45 |
|
|
|
|
| 48 |
| BAGEL-7B-MoT | 7.983 | 6.570 | 6.921 |
|
| 49 |
| **BAGEL-NHR-Edit** | 8.067 | 6.881 | 7.115 |
|
| 50 |
|
| 51 |
+
> **Note:** Scoring model: `gpt-4.1-2025-04-14` (with default temperature)
|
| 52 |
|
| 53 |
#### Metrics for ImgEdit-Bench:
|
| 54 |
|
|
|
|
| 57 |
| BAGEL-7B-MoT | 4.22| 1.53| 3.04| 3.3| 4.07| 3.67| 3.98| 3.5 | 3.0 | 3.3 |
|
| 58 |
| **BAGEL-NHR-Edit** | 4.3| 1.62| 3.18| 3.42| 3.95| 3.55| 4.19| 3.77| 2.94| 3.39|
|
| 59 |
|
| 60 |
+
> **Note:** Scoring model: `gpt-4o-2024-11-20` (with temperature = 0.0)
|
| 61 |
|
| 62 |
+
### 🖼️ Image Editing Results
|
| 63 |
|
| 64 |
Generated images for ImgEdit-Bench and GEdit-Bench are included in this repository.
|
| 65 |
|