update performance graph

This commit is contained in:
Brett Kuprel 2022-07-11 20:58:00 -04:00
parent 887706c9c9
commit 59b6a66958
4 changed files with 5 additions and 5 deletions

2
README.md vendored
View File

@ -13,7 +13,7 @@ To generate a 4x4 grid of DALL·E Mega images it takes:
- 48 sec with a P100 in Colab - 48 sec with a P100 in Colab
- 13 sec with an A100 on Replicate - 13 sec with an A100 on Replicate
Here's a more detailed breakdown of total inference time vs number of generated images on an A100: Here's a more detailed breakdown of performance on an A100:
<br /> <br />
<img src="https://github.com/kuprel/min-dalle/raw/main/performance.png" alt="min-dalle" width="450"/> <img src="https://github.com/kuprel/min-dalle/raw/main/performance.png" alt="min-dalle" width="450"/>
<br /> <br />

2
min_dalle.ipynb vendored
View File

@ -194,7 +194,7 @@
"text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n", "text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n",
"intermediate_outputs = True #@param {type:\"boolean\"}\n", "intermediate_outputs = True #@param {type:\"boolean\"}\n",
"grid_size = 5 #@param {type:\"integer\"}\n", "grid_size = 5 #@param {type:\"integer\"}\n",
"temperature = 1 #@param {type:\"slider\", min:0.01, max:3, step:0.01}\n", "temperature = 2 #@param {type:\"slider\", min:0.01, max:3, step:0.01}\n",
"supercondition_factor = 16 #@param {type:\"number\"}\n", "supercondition_factor = 16 #@param {type:\"number\"}\n",
"top_k = 256 #@param {type:\"integer\"}\n", "top_k = 256 #@param {type:\"integer\"}\n",
"log2_mid_count = 3 if intermediate_outputs else 0\n", "log2_mid_count = 3 if intermediate_outputs else 0\n",

BIN
performance.png vendored

Binary file not shown.

Before

Width:  |  Height:  |  Size: 96 KiB

After

Width:  |  Height:  |  Size: 16 KiB

View File

@ -30,8 +30,8 @@ class ReplicatePredictor(BasePredictor):
temperature: float = Input( temperature: float = Input(
description='A higher temperature results in more variety.', description='A higher temperature results in more variety.',
ge=0.01, ge=0.01,
le=3, le=10,
default=1 default=2
), ),
) -> Iterator[Path]: ) -> Iterator[Path]:
try: try: