update performance graph

This commit is contained in:
Brett Kuprel 2022-07-11 21:06:15 -04:00
parent 37dee13779
commit a4d544d6c4

3
README.md vendored
View File

@ -13,11 +13,10 @@ To generate a 4x4 grid of DALL·E Mega images it takes:
- 48 sec with a P100 in Colab
- 13 sec with an A100 on Replicate
Here's a more detailed breakdown of performance on an A100:
Here's a more detailed breakdown of performance on an A100. Credit to @technobird22 and his [NeoGen](https://github.com/technobird22/NeoGen) discord bot for the graph.
<br />
<img src="https://github.com/kuprel/min-dalle/raw/main/batch_performance.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).