add performance graph to readme

This commit is contained in:
Brett Kuprel 2022-07-11 08:40:31 -04:00
parent 2ab4365059
commit 6c78a6073e

4
README.md vendored
View File

@ -14,7 +14,9 @@ To generate a 4x4 grid of DALL·E Mega images it takes:
- 13 sec with an A100 on Replicate
Here's a more detailed breakdown on the A100 of total inference time vs number of generated images:
<img src="https://github.com/kuprel/min-dalle/raw/main/performance_a100.png" alt="min-dalle" width="400"/>
<br />
<img src="https://github.com/kuprel/min-dalle/raw/main/performance_a100.png" alt="min-dalle" width="450"/>
<br />
Credit to @technobird22 and his [NeoGen](https://github.com/technobird22/NeoGen) discord bot for the graph
The flax model and code for converting it to torch can be found [here](https://github.com/kuprel/min-dalle-flax).