Update README.md
Browse files
README.md
CHANGED
|
@@ -4,9 +4,10 @@
|
|
| 4 |
|
| 5 |
This is a fast, minimal port of Boris Dayma's [DALL·E Mega](https://github.com/borisdayma/dalle-mini). It has been stripped down for inference and converted to PyTorch. The only third party dependencies are numpy, requests, pillow and torch.
|
| 6 |
|
| 7 |
-
To generate a
|
|
|
|
| 8 |
- 48 sec with a P100 in Colab
|
| 9 |
-
-
|
| 10 |
|
| 11 |
The flax model and code for converting it to torch can be found [here](https://github.com/kuprel/min-dalle-flax).
|
| 12 |
|
|
|
|
| 4 |
|
| 5 |
This is a fast, minimal port of Boris Dayma's [DALL·E Mega](https://github.com/borisdayma/dalle-mini). It has been stripped down for inference and converted to PyTorch. The only third party dependencies are numpy, requests, pillow and torch.
|
| 6 |
|
| 7 |
+
To generate a 4x4 grid of DALL·E Mega images it takes:
|
| 8 |
+
- 89 sec with a T4 in Colab
|
| 9 |
- 48 sec with a P100 in Colab
|
| 10 |
+
- 13 sec with an A100 on Replicate
|
| 11 |
|
| 12 |
The flax model and code for converting it to torch can be found [here](https://github.com/kuprel/min-dalle-flax).
|
| 13 |
|