faster decoder self attention

This commit is contained in:
Brett Kuprel
2022-07-04 08:05:55 -04:00
parent a79d30f718
commit 377d15cb16
4 changed files with 21 additions and 23 deletions
+1 -1
View File
@@ -178,8 +178,8 @@
"%%time\n",
"\n",
"text = \"Dali painting of WALL·E\" #@param {type:\"string\"}\n",
"seed = 0 #@param {type:\"integer\"}\n",
"grid_size = 2 #@param {type:\"integer\"}\n",
"seed = -1 #@param {type:\"integer\"}\n",
"\n",
"display(model.generate_image(text, seed, grid_size))"
]