Update README.md
Browse files
README.md
CHANGED
|
@@ -23,10 +23,25 @@ Among its main features are:
|
|
| 23 |
|
| 24 |
## Speed
|
| 25 |
|
| 26 |
-
### test
|
| 27 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|:-:|:-:|:-:|
|
| 29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
|
| 32 |
|
|
@@ -73,7 +88,7 @@ docker pull bigmoyan/lyra_aigc:v0.1
|
|
| 73 |
```
|
| 74 |
|
| 75 |
## citation
|
| 76 |
-
```
|
| 77 |
@Misc{xFormers2022,
|
| 78 |
author = {Bin Wu, Kangjian Wu, Zhengtao Wang},
|
| 79 |
title = {LyraSD},
|
|
|
|
| 23 |
|
| 24 |
## Speed
|
| 25 |
|
| 26 |
+
### test environment
|
| 27 |
+
|
| 28 |
+
- device: Nvidia A100 40G
|
| 29 |
+
- img size: 512x512
|
| 30 |
+
- percision:fp16
|
| 31 |
+
- steps: 30
|
| 32 |
+
- solver: LMSD
|
| 33 |
+
|
| 34 |
+
### text2img
|
| 35 |
+
|model|time cost(ms)|memory(MB)|
|
| 36 |
|:-:|:-:|:-:|
|
| 37 |
+
|Pytorch|~5000ms|~10240|
|
| 38 |
+
|LyraSD|~435ms|~4026|
|
| 39 |
+
|
| 40 |
+
### superResolution
|
| 41 |
+
|model|time cost(ms)|memory(MB)|
|
| 42 |
+
|:-:|:-:|:-:|
|
| 43 |
+
|Pytorch|720ms|~6650|
|
| 44 |
+
|LyraSD|~26ms|~1600|
|
| 45 |
|
| 46 |
|
| 47 |
|
|
|
|
| 88 |
```
|
| 89 |
|
| 90 |
## citation
|
| 91 |
+
```bibtex
|
| 92 |
@Misc{xFormers2022,
|
| 93 |
author = {Bin Wu, Kangjian Wu, Zhengtao Wang},
|
| 94 |
title = {LyraSD},
|