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

update notebook

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-07 10:28:07 -04:00
parent 27642edbe6
commit 9a4a839b3b
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
min_dalle.ipynb vendored
View File

@ -130,9 +130,14 @@
"source": [ "source": [
"from IPython.display import display, update_display\n", "from IPython.display import display, update_display\n",
"from math import log2\n", "from math import log2\n",
"import torch\n",
"from min_dalle import MinDalle\n", "from min_dalle import MinDalle\n",
"\n", "\n",
"model = MinDalle(is_mega=True, is_reusable=True)" "model = MinDalle(\n",
" dtype=torch.float32,\n",
" is_mega=True, \n",
" is_reusable=True\n",
")"
] ]
}, },
{ {

2
replicate_predictor.py
View File

@ -19,7 +19,7 @@ class ReplicatePredictor(BasePredictor):
default=True default=True
), ),
grid_size: int = Input( grid_size: int = Input(
description='Size of the image grid. 4x4 takes about 15 seconds, 8x8 takes about 45 seconds', description='Size of the image grid. 4x4 takes about 15 seconds, 8x8 takes about 35 seconds',
ge=1, ge=1,
le=8, le=8,
default=4 default=4