Update README.md
Browse files
README.md
CHANGED
@@ -1,7 +1,8 @@
|
|
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,7 +30,7 @@ Model used in the paper:
|
|
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 |
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
3 |
library_name: diffusers
|
4 |
+
datasets:
|
5 |
+
- ILSVRC/imagenet-1k
|
6 |
---
|
7 |
|
8 |
# Gaussian Mixture Flow Matching Models (GMFlow)
|
|
|
30 |
|
31 |
## Usage
|
32 |
|
33 |
+
Please first install the [official code repository](https://github.com/Lakonik/GMFlow).
|
34 |
|
35 |
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.
|
36 |
|