Diffusers
GMDiTPipeline
Lakonik commited on
Commit
29eeed7
·
verified ·
1 Parent(s): d39f898

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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