tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

log2_temperature -> temperature

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-11 14:09:50 -04:00
parent 22b3fd1961
commit 887706c9c9
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
min_dalle.ipynb vendored
View File

@ -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",
"log2_temperature = 0 #@param {type:\"slider\", min:-5, max:5, step:0.1}\n",
"temperature = 1 #@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",
@ -204,7 +204,7 @@
" seed=-1,\n",
" grid_size=grid_size,\n",
" log2_mid_count=log2_mid_count,\n",
" temperature=2 ** log2_temperature,\n",
" temperature=temperature,\n",
" top_k=int(top_k),\n",
" supercondition_factor=float(supercondition_factor)\n",
")\n",

8
replicate_predictor.py
View File

@ -27,11 +27,11 @@ class ReplicatePredictor(BasePredictor):
le=8,
default=4
),
log2_temperature: float = Input(
temperature: float = Input(
description='A higher temperature results in more variety.',
ge=-3,
ge=0.01,
le=3,
default=0
default=1
),
) -> Iterator[Path]:
try:
@ -40,7 +40,7 @@ class ReplicatePredictor(BasePredictor):
seed = -1,
grid_size = grid_size,
log2_mid_count = 3 if intermediate_outputs else 0,
temperature = 2 ** log2_temperature,
temperature = temperature,
supercondition_factor = 2 ** 4,
top_k = 2 ** 8,
is_verbose = True