Update README.md
Browse files
README.md
CHANGED
|
@@ -5,4 +5,19 @@ pipeline_tag: any-to-any
|
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
|