Add pipeline tag and library name to model card
#1
by
nielsr
HF Staff
- opened
README.md
CHANGED
|
@@ -1,5 +1,7 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
|
| 5 |
# Gaussian Mixture Flow Matching Models (GMFlow)
|
|
@@ -27,7 +29,7 @@ Model used in the paper:
|
|
| 27 |
|
| 28 |
## Usage
|
| 29 |
|
| 30 |
-
Please first install the [official code repository](https://github.com/Lakonik/GMFlow)
|
| 31 |
|
| 32 |
We provide a Diffusers pipeline for easy inference. The following code demonstrates how to sample images from the pretrained GM-DiT model using the GM-ODE 2 solver and the GM-SDE 2 solver.
|
| 33 |
|
|
@@ -86,4 +88,4 @@ for i, (word, image) in enumerate(zip(words, output.images)):
|
|
| 86 |
primaryClass={cs.LG},
|
| 87 |
url={https://arxiv.org/abs/2504.05304},
|
| 88 |
}
|
| 89 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
+
pipeline_tag: unconditional-image-generation
|
| 4 |
+
library_name: diffusers
|
| 5 |
---
|
| 6 |
|
| 7 |
# Gaussian Mixture Flow Matching Models (GMFlow)
|
|
|
|
| 29 |
|
| 30 |
## Usage
|
| 31 |
|
| 32 |
+
Please first install the [official code repository](https://github.com/Lakonik/GMFlow).\
|
| 33 |
|
| 34 |
We provide a Diffusers pipeline for easy inference. The following code demonstrates how to sample images from the pretrained GM-DiT model using the GM-ODE 2 solver and the GM-SDE 2 solver.
|
| 35 |
|
|
|
|
| 88 |
primaryClass={cs.LG},
|
| 89 |
url={https://arxiv.org/abs/2504.05304},
|
| 90 |
}
|
| 91 |
+
```
|