Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# OneDiffusion
|
| 2 |
|
| 3 |
The inference code provided for the text-to-image workflow. The modified code is not a requirement, it's for demo purposes only. It has less heavy requirements than the [original repo](https://github.com/lehduong/OneDiffusion/tree/b6024589cc56b5af36268761828878b25af5e2fb). The inference speed is 7 s/it atm with the flash attention module removed.
|
|
@@ -31,4 +36,4 @@ if __name__ == '__main__':
|
|
| 31 |
height=1024,
|
| 32 |
width=1024).images[0]
|
| 33 |
image.save('cat.png')
|
| 34 |
-
```
|
|
|
|
| 1 |
+
---
|
| 2 |
+
base_model:
|
| 3 |
+
- lehduong/OneDiffusion
|
| 4 |
+
---
|
| 5 |
+
|
| 6 |
# OneDiffusion
|
| 7 |
|
| 8 |
The inference code provided for the text-to-image workflow. The modified code is not a requirement, it's for demo purposes only. It has less heavy requirements than the [original repo](https://github.com/lehduong/OneDiffusion/tree/b6024589cc56b5af36268761828878b25af5e2fb). The inference speed is 7 s/it atm with the flash attention module removed.
|
|
|
|
| 36 |
height=1024,
|
| 37 |
width=1024).images[0]
|
| 38 |
image.save('cat.png')
|
| 39 |
+
```
|