diff --git a/README.md b/README.md
index 0334b97..3e149fb 100644
--- a/README.md
+++ b/README.md
@@ -13,7 +13,7 @@ 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 total inference time vs number of generated images on an A100:
+Here's a more detailed breakdown of performance on an A100:
diff --git a/min_dalle.ipynb b/min_dalle.ipynb
index d7b50d8..4e922a0 100644
--- a/min_dalle.ipynb
+++ b/min_dalle.ipynb
@@ -194,7 +194,7 @@
"text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n",
"intermediate_outputs = True #@param {type:\"boolean\"}\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",
"top_k = 256 #@param {type:\"integer\"}\n",
"log2_mid_count = 3 if intermediate_outputs else 0\n",
diff --git a/performance.png b/performance.png
index 42cd6e7..bfceb4d 100644
Binary files a/performance.png and b/performance.png differ
diff --git a/replicate_predictor.py b/replicate_predictor.py
index 51f4afd..aa79916 100644
--- a/replicate_predictor.py
+++ b/replicate_predictor.py
@@ -30,11 +30,11 @@ class ReplicatePredictor(BasePredictor):
temperature: float = Input(
description='A higher temperature results in more variety.',
ge=0.01,
- le=3,
- default=1
+ le=10,
+ default=2
),
) -> Iterator[Path]:
- try:
+ try:
image_stream = self.model.generate_image_stream(
text = text,
seed = -1,