Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
pipeline_tag: any-to-any
|
4 |
+
---
|
5 |
+
|
6 |
+
This repository contains the model described in [OmniFlow: Any-to-Any Generation with Multi-Modal Rectified Flows](https://huggingface.co/papers/2412.01169).
|
7 |
+
|
8 |
+
Code: [https://github.com/jacklishufan/OmniFlows](https://github.com/jacklishufan/OmniFlows)
|
9 |
+
|
10 |
+
Usage:
|
11 |
+
|
12 |
+
```
|
13 |
+
from omniflow import OmniFlowPipeline
|
14 |
+
|
15 |
+
pipeline = OmniFlowPipeline.load_pretrained('ckpts/v0.5',device='cuda')
|
16 |
+
|
17 |
+
pipeline.cfg_mode = 'new'
|
18 |
+
imgs = pipeline("portrait of a cyberpunk girl with neon tattoos and a visor,staring intensely. Standing on top of a building",height=512,width=512,add_token_embed=0,task='t2i')
|
19 |
+
|
20 |
+
```
|
21 |
+
|
22 |
+
See [Notebook](https://github.com/jacklishufan/OmniFlows/blob/main/scripts/Demo.ipynb) for more examples
|
23 |
+
|