Spaces:
Running
Running
Merge pull request #27 from tmabraham/fix-forced-bos-token-on-demo
Browse files- demo/demo_notebook.ipynb +156 -68
demo/demo_notebook.ipynb
CHANGED
|
@@ -11,7 +11,7 @@
|
|
| 11 |
},
|
| 12 |
{
|
| 13 |
"cell_type": "code",
|
| 14 |
-
"execution_count":
|
| 15 |
"metadata": {
|
| 16 |
"colab": {
|
| 17 |
"base_uri": "https://localhost:8080/"
|
|
@@ -27,7 +27,7 @@
|
|
| 27 |
},
|
| 28 |
{
|
| 29 |
"cell_type": "code",
|
| 30 |
-
"execution_count":
|
| 31 |
"metadata": {
|
| 32 |
"id": "M1wVkrpjU6zO"
|
| 33 |
},
|
|
@@ -39,7 +39,7 @@
|
|
| 39 |
},
|
| 40 |
{
|
| 41 |
"cell_type": "code",
|
| 42 |
-
"execution_count":
|
| 43 |
"metadata": {},
|
| 44 |
"outputs": [
|
| 45 |
{
|
|
@@ -65,7 +65,7 @@
|
|
| 65 |
},
|
| 66 |
{
|
| 67 |
"cell_type": "code",
|
| 68 |
-
"execution_count":
|
| 69 |
"metadata": {
|
| 70 |
"id": "9jQnM6S2vCpn"
|
| 71 |
},
|
|
@@ -80,7 +80,7 @@
|
|
| 80 |
},
|
| 81 |
{
|
| 82 |
"cell_type": "code",
|
| 83 |
-
"execution_count":
|
| 84 |
"metadata": {
|
| 85 |
"id": "_eEaJVxAKpV5"
|
| 86 |
},
|
|
@@ -133,7 +133,7 @@
|
|
| 133 |
},
|
| 134 |
{
|
| 135 |
"cell_type": "code",
|
| 136 |
-
"execution_count":
|
| 137 |
"metadata": {
|
| 138 |
"scrolled": true
|
| 139 |
},
|
|
@@ -150,10 +150,10 @@
|
|
| 150 |
"text/html": [
|
| 151 |
"\n",
|
| 152 |
" Tracking run with wandb version 0.10.33<br/>\n",
|
| 153 |
-
" Syncing run <strong style=\"color:#cdcd00\">
|
| 154 |
" Project page: <a href=\"https://wandb.ai/tmabraham/vqgan-jax\" target=\"_blank\">https://wandb.ai/tmabraham/vqgan-jax</a><br/>\n",
|
| 155 |
-
" Run page: <a href=\"https://wandb.ai/tmabraham/vqgan-jax/runs/
|
| 156 |
-
" Run data is saved locally in <code>/home/tmabraham/vqgan-jax/wandb/run-
|
| 157 |
" "
|
| 158 |
],
|
| 159 |
"text/plain": [
|
|
@@ -167,20 +167,20 @@
|
|
| 167 |
"name": "stderr",
|
| 168 |
"output_type": "stream",
|
| 169 |
"text": [
|
| 170 |
-
"\u001b[34m\u001b[1mwandb\u001b[0m: Downloading large artifact model-1ef8yxby:
|
| 171 |
]
|
| 172 |
}
|
| 173 |
],
|
| 174 |
"source": [
|
| 175 |
"import wandb\n",
|
| 176 |
"run = wandb.init()\n",
|
| 177 |
-
"artifact = run.use_artifact('wandb/hf-flax-dalle-mini/model-1ef8yxby:
|
| 178 |
"artifact_dir = artifact.download()"
|
| 179 |
]
|
| 180 |
},
|
| 181 |
{
|
| 182 |
"cell_type": "code",
|
| 183 |
-
"execution_count":
|
| 184 |
"metadata": {
|
| 185 |
"id": "_6-XKK40oEfP",
|
| 186 |
"scrolled": true
|
|
@@ -203,6 +203,15 @@
|
|
| 203 |
"model = CustomFlaxBartForConditionalGeneration.from_pretrained(artifact_dir)"
|
| 204 |
]
|
| 205 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 206 |
{
|
| 207 |
"cell_type": "code",
|
| 208 |
"execution_count": 8,
|
|
@@ -253,27 +262,57 @@
|
|
| 253 |
{
|
| 254 |
"cell_type": "code",
|
| 255 |
"execution_count": 10,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 256 |
"metadata": {
|
| 257 |
"id": "P32mJJSbrU1F"
|
| 258 |
},
|
| 259 |
"outputs": [],
|
| 260 |
"source": [
|
| 261 |
-
"input_ids_test = tokenizer
|
| 262 |
]
|
| 263 |
},
|
| 264 |
{
|
| 265 |
"cell_type": "code",
|
| 266 |
-
"execution_count":
|
| 267 |
"metadata": {},
|
| 268 |
"outputs": [
|
| 269 |
{
|
| 270 |
"data": {
|
| 271 |
"text/plain": [
|
| 272 |
-
"DeviceArray([[ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 273 |
-
" 2]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 274 |
]
|
| 275 |
},
|
| 276 |
-
"execution_count":
|
| 277 |
"metadata": {},
|
| 278 |
"output_type": "execute_result"
|
| 279 |
}
|
|
@@ -284,18 +323,38 @@
|
|
| 284 |
},
|
| 285 |
{
|
| 286 |
"cell_type": "code",
|
| 287 |
-
"execution_count":
|
| 288 |
"metadata": {
|
| 289 |
"id": "C7cHbIHruELT"
|
| 290 |
},
|
| 291 |
"outputs": [],
|
| 292 |
"source": [
|
| 293 |
-
"greedy_output = model.generate(input_ids_test, max_length=257)"
|
| 294 |
]
|
| 295 |
},
|
| 296 |
{
|
| 297 |
"cell_type": "code",
|
| 298 |
-
"execution_count":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 299 |
"metadata": {
|
| 300 |
"colab": {
|
| 301 |
"base_uri": "https://localhost:8080/"
|
|
@@ -307,42 +366,16 @@
|
|
| 307 |
{
|
| 308 |
"data": {
|
| 309 |
"text/plain": [
|
| 310 |
-
"DeviceArray([[16384,
|
| 311 |
-
"
|
| 312 |
-
"
|
| 313 |
-
"
|
| 314 |
-
"
|
| 315 |
-
"
|
| 316 |
-
"
|
| 317 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 318 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 319 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 320 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 321 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 322 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 323 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 324 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 325 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 326 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 327 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 328 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 329 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 330 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 331 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 332 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 333 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 334 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 335 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 336 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 337 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 338 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 339 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 340 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 341 |
-
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 342 |
-
" 10042]], dtype=int32)"
|
| 343 |
]
|
| 344 |
},
|
| 345 |
-
"execution_count":
|
| 346 |
"metadata": {},
|
| 347 |
"output_type": "execute_result"
|
| 348 |
}
|
|
@@ -351,6 +384,58 @@
|
|
| 351 |
"greedy_output[0]"
|
| 352 |
]
|
| 353 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 354 |
{
|
| 355 |
"cell_type": "markdown",
|
| 356 |
"metadata": {},
|
|
@@ -360,7 +445,7 @@
|
|
| 360 |
},
|
| 361 |
{
|
| 362 |
"cell_type": "code",
|
| 363 |
-
"execution_count":
|
| 364 |
"metadata": {},
|
| 365 |
"outputs": [],
|
| 366 |
"source": [
|
|
@@ -378,7 +463,7 @@
|
|
| 378 |
},
|
| 379 |
{
|
| 380 |
"cell_type": "code",
|
| 381 |
-
"execution_count":
|
| 382 |
"metadata": {},
|
| 383 |
"outputs": [],
|
| 384 |
"source": [
|
|
@@ -387,7 +472,7 @@
|
|
| 387 |
},
|
| 388 |
{
|
| 389 |
"cell_type": "code",
|
| 390 |
-
"execution_count":
|
| 391 |
"metadata": {},
|
| 392 |
"outputs": [],
|
| 393 |
"source": [
|
|
@@ -402,13 +487,14 @@
|
|
| 402 |
},
|
| 403 |
{
|
| 404 |
"cell_type": "code",
|
| 405 |
-
"execution_count":
|
| 406 |
"metadata": {
|
| 407 |
"colab": {
|
| 408 |
"base_uri": "https://localhost:8080/"
|
| 409 |
},
|
| 410 |
"id": "Jz032w73nHEf",
|
| 411 |
-
"outputId": "994d8e85-bff7-480b-8b69-f69dedc15c49"
|
|
|
|
| 412 |
},
|
| 413 |
"outputs": [
|
| 414 |
{
|
|
@@ -420,47 +506,49 @@
|
|
| 420 |
}
|
| 421 |
],
|
| 422 |
"source": [
|
| 423 |
-
"model = VQModel.from_pretrained(\"
|
| 424 |
]
|
| 425 |
},
|
| 426 |
{
|
| 427 |
"cell_type": "code",
|
| 428 |
-
"execution_count":
|
| 429 |
"metadata": {},
|
| 430 |
"outputs": [],
|
| 431 |
"source": [
|
| 432 |
"def get_images(indices, model):\n",
|
| 433 |
" indices = indices[:, 1:]\n",
|
| 434 |
-
"
|
| 435 |
-
"
|
|
|
|
| 436 |
]
|
| 437 |
},
|
| 438 |
{
|
| 439 |
"cell_type": "code",
|
| 440 |
-
"execution_count":
|
| 441 |
"metadata": {},
|
| 442 |
"outputs": [
|
| 443 |
{
|
| 444 |
"name": "stdout",
|
| 445 |
"output_type": "stream",
|
| 446 |
"text": [
|
|
|
|
| 447 |
"Working with z of shape (1, 256, 16, 16) = 65536 dimensions.\n"
|
| 448 |
]
|
| 449 |
},
|
| 450 |
{
|
| 451 |
"data": {
|
| 452 |
-
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAAEAAAEACAIAAAD9XIvPAAAAF0lEQVR4nGP4//8/EwMDwygexaN45GEA7ucE/J1FRrMAAAAASUVORK5CYII=\n",
|
| 453 |
"text/plain": [
|
| 454 |
-
"<PIL.Image.Image image mode=RGB size=
|
| 455 |
]
|
| 456 |
},
|
| 457 |
-
"execution_count":
|
| 458 |
"metadata": {},
|
| 459 |
"output_type": "execute_result"
|
| 460 |
}
|
| 461 |
],
|
| 462 |
"source": [
|
| 463 |
-
"custom_to_pil(np.asarray(get_images(greedy_output[0], model)[0]))"
|
| 464 |
]
|
| 465 |
}
|
| 466 |
],
|
|
|
|
| 11 |
},
|
| 12 |
{
|
| 13 |
"cell_type": "code",
|
| 14 |
+
"execution_count": null,
|
| 15 |
"metadata": {
|
| 16 |
"colab": {
|
| 17 |
"base_uri": "https://localhost:8080/"
|
|
|
|
| 27 |
},
|
| 28 |
{
|
| 29 |
"cell_type": "code",
|
| 30 |
+
"execution_count": 1,
|
| 31 |
"metadata": {
|
| 32 |
"id": "M1wVkrpjU6zO"
|
| 33 |
},
|
|
|
|
| 39 |
},
|
| 40 |
{
|
| 41 |
"cell_type": "code",
|
| 42 |
+
"execution_count": 2,
|
| 43 |
"metadata": {},
|
| 44 |
"outputs": [
|
| 45 |
{
|
|
|
|
| 65 |
},
|
| 66 |
{
|
| 67 |
"cell_type": "code",
|
| 68 |
+
"execution_count": 3,
|
| 69 |
"metadata": {
|
| 70 |
"id": "9jQnM6S2vCpn"
|
| 71 |
},
|
|
|
|
| 80 |
},
|
| 81 |
{
|
| 82 |
"cell_type": "code",
|
| 83 |
+
"execution_count": 4,
|
| 84 |
"metadata": {
|
| 85 |
"id": "_eEaJVxAKpV5"
|
| 86 |
},
|
|
|
|
| 133 |
},
|
| 134 |
{
|
| 135 |
"cell_type": "code",
|
| 136 |
+
"execution_count": 5,
|
| 137 |
"metadata": {
|
| 138 |
"scrolled": true
|
| 139 |
},
|
|
|
|
| 150 |
"text/html": [
|
| 151 |
"\n",
|
| 152 |
" Tracking run with wandb version 0.10.33<br/>\n",
|
| 153 |
+
" Syncing run <strong style=\"color:#cdcd00\">rare-night-7</strong> to <a href=\"https://wandb.ai\" target=\"_blank\">Weights & Biases</a> <a href=\"https://docs.wandb.com/integrations/jupyter.html\" target=\"_blank\">(Documentation)</a>.<br/>\n",
|
| 154 |
" Project page: <a href=\"https://wandb.ai/tmabraham/vqgan-jax\" target=\"_blank\">https://wandb.ai/tmabraham/vqgan-jax</a><br/>\n",
|
| 155 |
+
" Run page: <a href=\"https://wandb.ai/tmabraham/vqgan-jax/runs/qzxavce8\" target=\"_blank\">https://wandb.ai/tmabraham/vqgan-jax/runs/qzxavce8</a><br/>\n",
|
| 156 |
+
" Run data is saved locally in <code>/home/tmabraham/vqgan-jax/wandb/run-20210715_075019-qzxavce8</code><br/><br/>\n",
|
| 157 |
" "
|
| 158 |
],
|
| 159 |
"text/plain": [
|
|
|
|
| 167 |
"name": "stderr",
|
| 168 |
"output_type": "stream",
|
| 169 |
"text": [
|
| 170 |
+
"\u001b[34m\u001b[1mwandb\u001b[0m: Downloading large artifact model-1ef8yxby:latest, 1674.97MB. 2 files... Done. 0:0:0\n"
|
| 171 |
]
|
| 172 |
}
|
| 173 |
],
|
| 174 |
"source": [
|
| 175 |
"import wandb\n",
|
| 176 |
"run = wandb.init()\n",
|
| 177 |
+
"artifact = run.use_artifact('wandb/hf-flax-dalle-mini/model-1ef8yxby:latest', type='bart_model')\n",
|
| 178 |
"artifact_dir = artifact.download()"
|
| 179 |
]
|
| 180 |
},
|
| 181 |
{
|
| 182 |
"cell_type": "code",
|
| 183 |
+
"execution_count": 6,
|
| 184 |
"metadata": {
|
| 185 |
"id": "_6-XKK40oEfP",
|
| 186 |
"scrolled": true
|
|
|
|
| 203 |
"model = CustomFlaxBartForConditionalGeneration.from_pretrained(artifact_dir)"
|
| 204 |
]
|
| 205 |
},
|
| 206 |
+
{
|
| 207 |
+
"cell_type": "code",
|
| 208 |
+
"execution_count": 7,
|
| 209 |
+
"metadata": {},
|
| 210 |
+
"outputs": [],
|
| 211 |
+
"source": [
|
| 212 |
+
"model.config.forced_bos_token_id = None"
|
| 213 |
+
]
|
| 214 |
+
},
|
| 215 |
{
|
| 216 |
"cell_type": "code",
|
| 217 |
"execution_count": 8,
|
|
|
|
| 262 |
{
|
| 263 |
"cell_type": "code",
|
| 264 |
"execution_count": 10,
|
| 265 |
+
"metadata": {},
|
| 266 |
+
"outputs": [],
|
| 267 |
+
"source": [
|
| 268 |
+
"input_text = ['I enjoy walking with my cute dog']*8"
|
| 269 |
+
]
|
| 270 |
+
},
|
| 271 |
+
{
|
| 272 |
+
"cell_type": "code",
|
| 273 |
+
"execution_count": 11,
|
| 274 |
"metadata": {
|
| 275 |
"id": "P32mJJSbrU1F"
|
| 276 |
},
|
| 277 |
"outputs": [],
|
| 278 |
"source": [
|
| 279 |
+
"input_ids_test = tokenizer(input_text, return_tensors='jax')"
|
| 280 |
]
|
| 281 |
},
|
| 282 |
{
|
| 283 |
"cell_type": "code",
|
| 284 |
+
"execution_count": 12,
|
| 285 |
"metadata": {},
|
| 286 |
"outputs": [
|
| 287 |
{
|
| 288 |
"data": {
|
| 289 |
"text/plain": [
|
| 290 |
+
"{'input_ids': DeviceArray([[ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 291 |
+
" 2],\n",
|
| 292 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 293 |
+
" 2],\n",
|
| 294 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 295 |
+
" 2],\n",
|
| 296 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 297 |
+
" 2],\n",
|
| 298 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 299 |
+
" 2],\n",
|
| 300 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 301 |
+
" 2],\n",
|
| 302 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 303 |
+
" 2],\n",
|
| 304 |
+
" [ 0, 100, 2254, 3051, 19, 127, 11962, 2335,\n",
|
| 305 |
+
" 2]], dtype=int32), 'attention_mask': DeviceArray([[1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 306 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 307 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 308 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 309 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 310 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 311 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1],\n",
|
| 312 |
+
" [1, 1, 1, 1, 1, 1, 1, 1, 1]], dtype=int32)}"
|
| 313 |
]
|
| 314 |
},
|
| 315 |
+
"execution_count": 12,
|
| 316 |
"metadata": {},
|
| 317 |
"output_type": "execute_result"
|
| 318 |
}
|
|
|
|
| 323 |
},
|
| 324 |
{
|
| 325 |
"cell_type": "code",
|
| 326 |
+
"execution_count": 13,
|
| 327 |
"metadata": {
|
| 328 |
"id": "C7cHbIHruELT"
|
| 329 |
},
|
| 330 |
"outputs": [],
|
| 331 |
"source": [
|
| 332 |
+
"greedy_output = model.generate(input_ids_test['input_ids'], max_length=257)"
|
| 333 |
]
|
| 334 |
},
|
| 335 |
{
|
| 336 |
"cell_type": "code",
|
| 337 |
+
"execution_count": 14,
|
| 338 |
+
"metadata": {},
|
| 339 |
+
"outputs": [
|
| 340 |
+
{
|
| 341 |
+
"data": {
|
| 342 |
+
"text/plain": [
|
| 343 |
+
"(8, 257)"
|
| 344 |
+
]
|
| 345 |
+
},
|
| 346 |
+
"execution_count": 14,
|
| 347 |
+
"metadata": {},
|
| 348 |
+
"output_type": "execute_result"
|
| 349 |
+
}
|
| 350 |
+
],
|
| 351 |
+
"source": [
|
| 352 |
+
"greedy_output[0].shape"
|
| 353 |
+
]
|
| 354 |
+
},
|
| 355 |
+
{
|
| 356 |
+
"cell_type": "code",
|
| 357 |
+
"execution_count": 15,
|
| 358 |
"metadata": {
|
| 359 |
"colab": {
|
| 360 |
"base_uri": "https://localhost:8080/"
|
|
|
|
| 366 |
{
|
| 367 |
"data": {
|
| 368 |
"text/plain": [
|
| 369 |
+
"DeviceArray([[16384, 10042, 10042, ..., 10042, 10042, 9570],\n",
|
| 370 |
+
" [16384, 10042, 10042, ..., 10042, 10042, 9570],\n",
|
| 371 |
+
" [16384, 10042, 10042, ..., 10042, 10042, 9570],\n",
|
| 372 |
+
" ...,\n",
|
| 373 |
+
" [16384, 10042, 10042, ..., 10042, 10042, 9570],\n",
|
| 374 |
+
" [16384, 10042, 10042, ..., 10042, 10042, 9570],\n",
|
| 375 |
+
" [16384, 10042, 10042, ..., 10042, 10042, 9570]], dtype=int32)"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 376 |
]
|
| 377 |
},
|
| 378 |
+
"execution_count": 15,
|
| 379 |
"metadata": {},
|
| 380 |
"output_type": "execute_result"
|
| 381 |
}
|
|
|
|
| 384 |
"greedy_output[0]"
|
| 385 |
]
|
| 386 |
},
|
| 387 |
+
{
|
| 388 |
+
"cell_type": "code",
|
| 389 |
+
"execution_count": 16,
|
| 390 |
+
"metadata": {},
|
| 391 |
+
"outputs": [
|
| 392 |
+
{
|
| 393 |
+
"data": {
|
| 394 |
+
"text/plain": [
|
| 395 |
+
"DeviceArray([16384, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 396 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 397 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 398 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 399 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 400 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 401 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 402 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 403 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 404 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 405 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 406 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 407 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 408 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 409 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 410 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 411 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 412 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 413 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 414 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 415 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 416 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 417 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 418 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 419 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 420 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 421 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 422 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 423 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 424 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 425 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 426 |
+
" 10042, 10042, 10042, 10042, 10042, 10042, 10042, 10042,\n",
|
| 427 |
+
" 9570], dtype=int32)"
|
| 428 |
+
]
|
| 429 |
+
},
|
| 430 |
+
"execution_count": 16,
|
| 431 |
+
"metadata": {},
|
| 432 |
+
"output_type": "execute_result"
|
| 433 |
+
}
|
| 434 |
+
],
|
| 435 |
+
"source": [
|
| 436 |
+
"greedy_output[0][0]"
|
| 437 |
+
]
|
| 438 |
+
},
|
| 439 |
{
|
| 440 |
"cell_type": "markdown",
|
| 441 |
"metadata": {},
|
|
|
|
| 445 |
},
|
| 446 |
{
|
| 447 |
"cell_type": "code",
|
| 448 |
+
"execution_count": 17,
|
| 449 |
"metadata": {},
|
| 450 |
"outputs": [],
|
| 451 |
"source": [
|
|
|
|
| 463 |
},
|
| 464 |
{
|
| 465 |
"cell_type": "code",
|
| 466 |
+
"execution_count": 18,
|
| 467 |
"metadata": {},
|
| 468 |
"outputs": [],
|
| 469 |
"source": [
|
|
|
|
| 472 |
},
|
| 473 |
{
|
| 474 |
"cell_type": "code",
|
| 475 |
+
"execution_count": 19,
|
| 476 |
"metadata": {},
|
| 477 |
"outputs": [],
|
| 478 |
"source": [
|
|
|
|
| 487 |
},
|
| 488 |
{
|
| 489 |
"cell_type": "code",
|
| 490 |
+
"execution_count": 20,
|
| 491 |
"metadata": {
|
| 492 |
"colab": {
|
| 493 |
"base_uri": "https://localhost:8080/"
|
| 494 |
},
|
| 495 |
"id": "Jz032w73nHEf",
|
| 496 |
+
"outputId": "994d8e85-bff7-480b-8b69-f69dedc15c49",
|
| 497 |
+
"scrolled": true
|
| 498 |
},
|
| 499 |
"outputs": [
|
| 500 |
{
|
|
|
|
| 506 |
}
|
| 507 |
],
|
| 508 |
"source": [
|
| 509 |
+
"model = VQModel.from_pretrained(\"flax-community/vqgan_f16_16384\")"
|
| 510 |
]
|
| 511 |
},
|
| 512 |
{
|
| 513 |
"cell_type": "code",
|
| 514 |
+
"execution_count": 21,
|
| 515 |
"metadata": {},
|
| 516 |
"outputs": [],
|
| 517 |
"source": [
|
| 518 |
"def get_images(indices, model):\n",
|
| 519 |
" indices = indices[:, 1:]\n",
|
| 520 |
+
" print(indices.shape)\n",
|
| 521 |
+
" img = model.decode_code(indices)\n",
|
| 522 |
+
" return img"
|
| 523 |
]
|
| 524 |
},
|
| 525 |
{
|
| 526 |
"cell_type": "code",
|
| 527 |
+
"execution_count": 22,
|
| 528 |
"metadata": {},
|
| 529 |
"outputs": [
|
| 530 |
{
|
| 531 |
"name": "stdout",
|
| 532 |
"output_type": "stream",
|
| 533 |
"text": [
|
| 534 |
+
"(1, 256)\n",
|
| 535 |
"Working with z of shape (1, 256, 16, 16) = 65536 dimensions.\n"
|
| 536 |
]
|
| 537 |
},
|
| 538 |
{
|
| 539 |
"data": {
|
| 540 |
+
"image/png": "iVBORw0KGgoAAAANSUhEUgAAAQAAAAEACAIAAADTED8xAAAtSElEQVR4nO1d6Y4jOXKOyFJ3zwzWXgP2An4Dv/8D+QF8ADvrxczsTFcp/CMzmWRcDDJPqfShu0pi8giScZNS4cevf/3tx28/wjcAJKABBoI7ASEMCEgAAAQAAEhwB0CYgfOD9A4BaHqxVHvhk2NkBZpfU1auvs5bRTqn7GcOEkMDwB1gALgDfcD9HT4Qvt6+f337AkBw/wAkGIaJiREABxhZHgGIgAAw52xBMc7l+GL/F9YjwkWyTmL6USpywRhf3wEGQIK3G+AXgBvcfxre7gQEAG9wJ4Bh4nhEoAEGhDsBIgwA91lSJx1Pmvjh/JOJIAhSoFQM6oRJSDmJOu5KoTG+fCQ7fhqwfUiFcmUS1NWwtghKxgN3YTvVo08uq5kqD2Jgmn8O8A7wFf/4+A6j34MfAIjwls2EAIaxyegOZQPT6DLh/DMrL5ZEEM/oeeEFSPoNAQJakbcUqq14w7Q2ACLQO9Ad7l/hdgP4ABgAhpn1px5nR2jxbUpalkelx4+i5oghLMDPBib90gzmkDXjqtjqp1qfUaJ6z6xnx2JKslXrLUhCwfTyhT6WVTBTMqppoMmrJwQaAEfjcAOCAd5mnwcBhjvcZ84GmmLfRL0VBBclQvcjL9bhro9SOaokzgXZb32vy3na6q5V67cS2UQbaeWqS9AaQEa2HzP2G3mYZneGAG7DXAeBAIjgPpoh4n4cCyqIFiEpplRqOHXuwakF6/T57q06VO2B6TUWgMXjCqZ/QbS1HGAW48k+I1Y34nJU1Y0fxIXgbKdPvYwnncqzWz/mdeBGBDCp/OTxJ7ee8faIez6KJe5ob9TW6Ot4fcgr94vsR63ExDWyNUqwGqtQ1ewdZqlnqYW/nWvXgpuSXNoOU6GSCIhgSLVuOCABEgHNrj5NgewyYen9V9mORJ31HOcgomz9OlKLOEo4SBIYTVRvWHUJZLml8GXnV3cNXdCyCSPy2FLhK8sbz9/SlBTKtTzcgAaEpPAJAHE+85p9MlWXBFwvm6DII9TYsRqQ+XY6TrQq5R0sJZtYy9dKs0PMQ7P+jFz/4qyfze0tuUipM5sIRKCx5hgH3wgIgBCG+7LphHP+NBs170oOzdldUtGxYT47ygrZ66rLq/YktYbqfDNxa4PLnR2Komm0oEnz1QhDt3tbiSvKPbUdHIU9FnORPJEUXtN00gUEcAe4IWJ2wwHnJWEuV04xez2KZrGUpS98ij5q5RXLC3ec745ROshYOVZfh+tjmKaGk5qfi9KdmjocP1MSl7EvzYoebvPYlHqcT3lhdoFSJ6pCV9X9RcCVXJ8WvR4cm9w5P6mKrZgkAtkV639+rQQsKQVv0RlE6nooRqTZIEyHAzcWaNBUCWA6GaiOW6r7uQ+cgumoHO/Dl7zXp+B+EPNYOy3VEWHKVd1Ia4Ols5jblKwJiRdOZw2Q4geFDCxvbwCkTgOLFchluLom468GM6pVsqLH9PQJUh07oXllIltFGlurrapuSc5SYeomsxBsFVcJt5belig53H8dc3cyXFOj7ZyME0ExITzFM9xmxI6khV/HSaTEO8M5SFi/7GONm6/Ip6CA24EtQex38UQdLpLGkd6s5Td3TypivtnoHayZzyhOZHU4KbrcUd/VXWQ5lhq5VPyaHfX4x07U8IaABtsCOOHQfpEk16llNkD4cgttaQF9j9R5ai1lN9dqTmUP1MRDa8PIU2UdrPyGZfjUHXLKzYGVKuR8FkVvUH+CAHAjItTC9CxOzyHdvw1NPNepYg6W0jVCrKKCX265NL5uskasrknVHfDXVlWa3e6WOZZv4/xClgKRXbXQSqlDXMJLtY9pUMttyE91xxW8ZX2g075K3worvy1ynk4oLEbJ7HJrcjcgP+WwrKJ0sZrsieqhOSDxU+2feRkJ0qKCqMNqWjSoU5Yro3erjldjoCQDy1uDNrN9/noAuGX0jBc8qYv7odwSx+IdAC9sYIxjuNi8fv5SZbHAxvuP4krWr8eI5D0sd77q/VdHD07ZaYXygW8cSWyNMZ1CF6SjLUnKTSUqN2SWrXW5+wqmYILB4ma1eIf5W1XTegowNlDVpY60lY8kGV2eyXqYwwY1KPLTKt11txYKAYZR4ZcRBvnNwriEDARirA1AhmCoxAQ7zLulcgg5XIQkB8eskj2ybkwisZS0AMQ7ROXVXHOQn8BBW3p88yQerpagLsiNR/HiLLDFlDxNNcbddU2bZIY1XDEmQ8MupVs85ImNGi9NNdNViNy1IlYplUgBMoiVPnZ6wyQu6iwlw50cMHDlrDk+EoiHpVZzdcRIyeaQc9lcxlY7UVVW0VsRABaBjQUeBoxNbtlbfeLOcgSnSkXvy82LmYl5WMNsljRBunhp41r9SDSxSBX+cCtFqxW5e3PAuNV13gpsXq1e3KjWh1n9V8nuSI8ujWUYR6nTsmb2nLpHlIh4zEFEljgumR2dByGXN95wP7RsKPl82aiw9OfjXVFkypK9SI27l0bVedLf1WhUHbhT/IgjRlrZudQyrM/qDh5gnVoijUpMVA2ZMuhTv6WHVi/yUbdesYYg8ymJn+yRWoKNmx4BCUvGnF454lY0qP2oe0La50gIWq6m28xkRVU+72yC6XPqkcs/xBU36k9mjAKQFs5ZJoKSC6IrmoHtnuTTuBDWBukjMNItK7HeqiWtcDSP2rlTWCEmsMIyA5K/UGVjmy0gl7RqbOMILpSX4aS48LbVHYggX7AgZzc5jrV2lhKFcspHhqlBbEtS4S6R8sjS7pY/q8LUaZbcOLBYuTqY1slUV94GTWth9rZSqKXmzN045sU63BoYxNdAah0ZATkajhGbGq735CNxsj8Km521wLKQ9eCznDWoLNdVaocYtALL3+ztbaQpW/Lq/lV5tZvEDUHiRaRya514YROC6x/spBo7xfuMD6qWs1hujQ+ht03KNNfxaquUfB9yAvlLTinN35mVavLKrYYw9ZL/kw0v6I68sBrLbiucZDTxP2pLS7Von4MsksEOmQ8rbRNNaPA32IZ2LDwgK/fCeWBuSaRBs93wLaAiAJkvNP4qwgyDHXlEJW0eCg8xFwkWylD54oUnxWIBwnyd+yDhAexixwIQrwsA9esZBAuvE2Z8nzpnYSaW/0C8wBYN8cJDoc0ClH+qoiIFbp/Lye/g9KRF6FYIpXtm0vqo+t7qzhnyhaeAqmSd2swBqSQqI2clN6jIk/kBsVk80hzyj84sR5Iqf4PB4mqKziHuhcfHxDL5H9ryPaJSL+t5VCepY8UA1ojM58kTSXr+nCyiXFz28OmF/UGzZz9+ef8BHq8bA5h1y3c1R6Xy2WImPIdM+4UHAO6e+yg6H9TSmZTlpwrTaiwP9ZnITOsr3flCLyzHmZeq+VDvqxGdZulnUt6CCipb5ISaf1/serDo2slcbWIIH96a0mQHgrNgDKaqcpO/vBigQ/dbFcuv2i1infP2Sppayv4B8/eK93sJ7CbcD+VEHuw0hQoPorWpXu4kcvK/BNMJQ+6WYgSwbrSetzPyOI6xe0Fa1Ru8BpasoiaxDyMGcv0Z+jZCuNkE2V2g+upYuR1jaRk/obIpU+x/MAiyuyKWDxagSq2i8d4RoHwxk+7RGOUxbALVeLJjAtmCTAoNY1+PvjRSR41lMJ2mh0XAyyCSM9SUbjtQvNgJOXuTLDKGZ9rn4ibNJG81t0xKiiJBcBCta4mZuqKyvHXok9DHPXnigOlrv39ZgXOwmlsLEMOGuIpUtMTBnSNg5BxghBNG+BWaOkT37eoRdg25WfTpEMN8JMshqVbwHwWnajmDzn5Wt3pDF2t3TahbAOmT+P5PB5Lek1a5W6jKnpyRVy6rdcfE0Vgd67SNY5i6UPfUHd15ZJkOOdqOKrzDQBBnN8MCSJ8EaoO1kELFL3voXpCm7KncHgfVqUhlr8a+rFqrXmxbCb93FD9BkB4e2vLW5NxXgdjv7YHQfhVibbU0MGWN/JBUGyHCTFLfWZobS96ITMXZbCYYB8GfXv420LCuzUjxmoyKa7Gv1+oJgDNyNWTzUY3vGsmJ1GSaWO57Jdff59iu2X8SP3NKWm0K6we1QqMuuAOq7oJTcg4sd6NqAfQVXnkwNHtvfXmuuO/nq+FQP/m29znyKk3WKNKagOBEtcOqxUGjptEkf06yGvJqe0ORvXadyxeycg4gQxtZAbXXEWhaJ9JB64qbtFORaCuq5ROXQbmabWRrxVP0Lk1Be2iVS84OSp1fQXBMhWyjm9zktPKI1WewlWqO+IzqB2G+UlFfdyHYQV9uhDfJGL/4Zm2pa6u77qdufea2DkNkWxKFlgp3UlRNWKPdDIp6+siG5uSso2pCVQAwWUC5H9utkQN1ZGgRmKItc/+xEIYoNUytSeKqpLA+1WokeMeyRYyGDgtQJUwV3ZhWb90yc9B2LNSR0V9zDLCappViYmU7ojWdANl6CoYIQuNqpCEwJnJs0MRwudGwvDIqX0SGaEI15NgB+mgtNCjTbbgLVA0JoEWVnoGF9lzrR2jecEaOM2MVqmRYLlC8gvPI2mW13NcdZcVmiXMatGdieixAW69VlVPui793cmEje21xzvKIRBsIi27cAEUgyaXAGsoww/L7g4UqYU3lENr6cK0ZNrXLVoa70ys2C4DcsEaoIRGbCNo8bSkmKDk5b1s0UR3u+DqqNftEQurDiAMG7hJY1XzD0orVWqBNDLSqRRC8ZlJrLUC7mpH7QvaSSnZ1tlLthJdg9i/fB4zta5DKIILapOpzshL2D9wlzvupVlBfgNiYqv8WRMTrhuiC67WqX4uyPPZrCGUWVGcMvm5gHCufyrdmb8xeRLZH3YnEOpLbNnSWLEhHX65CxMpVg1ppka2AJLbxobWpVlpv0FIQ3BoOFTXcljmPrfFL49AJlqYnz+WZzWr9Btuuh7UTVlSav908OeEMuu1SEH+3hluUJVSyQGLIZpRkJlWLykO9kT8o66FOIWUcP5dgXpJ9v9022EMe1vS5Bz05N6lBmGOsg2BGBnlZT3+UKT5QBaCu80WvVXfELCuGrSJiEnkdNuGsI07PAVr8uSFj+gsuKVOF1SB4mVRVLW+BDs9ZhnycZiEWyhAX3KorQA15R+T63vd/TlpbyxMsMFS+wTBDUxAcznDsCyMY5YnRF/c78HWSFQys93/iiG2fbpBu8gv/O5JxUA9PsFQZR0BGvfLhi/V9tDrD+QvNg2d1j2MG0wKs7zg2iW5O63CKKLWEzEa/NP1OqCY33HZ9IPGzE6sEoHEG3Znx1hkizOyeJ+lTXy8x2BbBBOgaFeg+aWMqRsYQpGsjplEPT5w6/YkEJm0yInlJwinoWHNDwDrVPxOXqAVotG4qWKa/emLYbS6mdG9e9MJK+Oa7a6vW8m7j6KiOOsjWmxxbOBQEEDm4tBQA5s9ean5DVFfyyJA2P8Fv2uJOCyBbdlSI1SXhBckK3tPj8wufAZtwf66YdoLds5732+bzABtE4w1wQh9SXr2wP3o9+5X3GqKntA6G1uaS77bWtexAt3bAMEFx/V84EvGVLy8LrUkObbDbw9xXFNbJX1dTB/n06vmil7t/BQTXP/mo61i4OGvu7ipqATZir3g31WqjvicAovlP0r/wcFixa8G7w/rjVNhsAVag59iibEK5rpep/RceCFvs18JRTm9+FiX6iTALjdOIW6rKLUMqa7y4/xFB7HdzaxIcFeoq55x0EtyZP2mRGys/6ZzSKodiwYP3Fx4C8053b2Y9R2J1PTYbagakPv4O1cVnV7LXL1//GRHaVY17eqJf1QU6BkHuzbW8Mr1gZvSFB8GabNBa3X2wAERQ+fRFcvpfMvAsWOPMovu2jmHl+FtA3tpk09AD3bPJfuEEiE1fmwscAIB9h8+emrX7RsmL258bJ5jzOQgW2I3X2OdSWtu+8MQ4TcEpArAbr/m3OK2aVskLz4Qo023OnMpViCvx2kvxfwqsYLm1HKJchdiN6Viw65xoKZ7SlcTyhY2xguXW8oXyvUC7sVqezMm/qyF/bZIQ+ZDYCw8K6t/Y+jVI//Ewfy/Qkc4GO96Sb81mL5foWdErA2uvKacg+Bjd6ng7ITJeFuApEddrggHWXoQfzmYpNr5+ZfrTfNjlc8xSgMRhVAey9g3LOJzkVFjX3ZRD38/EFJ/Xxeu4Jjm3K+4zo6jl93yFu0AehdYV6mfEp7FzOvrmXg9h/csFDX8mdVPUZ0tqNvSZURXztAzPphCk2rYgEiFUlj/SZbh6+PJ0fL/y5hauj/meFZUvybExWI13RmjMp9tqdvH96ebXC+q/CuFlESOdnhsDPJ2Kr+PF9ArWnQQ3hLwSaz8UvwIVUrf53qOHgRoBqwugfoT6M+ORDsJyeCInd/XZNafl3Fv78nKlRvB1Q4C2c4BtyWlB9VT4M8O3Bms/BnU1HDyBnM/OygJVjPvnFgX11iDU7lA9LuTfqWtsn72OrEhe5wanZYHaNu85tjpDxMVjV2CdRXvaI4IaCADGP3uehcP1RUg1LmUBHg4bzoI0jR69J2vUUft8bGiX4Zy3OnJFcoWrEA+KzW9nb3XIlZj+s1kDH6h+sdpLALohP6JT+UxPDRFtjeKF39uznRxrk1lxmfq8u0A6SGOrC4MZgaadsBx6Z+r5XfFqtc8DylW7O/lizcdXwzU/ZnVBkgSUm9vikZPN9GXcOhOouvWPsHIlWih2UgUT3GVVLkpEPw9w2Lo6bHU24kSpH/9v8o66F+CSK7cLdO+uZgE4lA/Fq/g86+oi7rSol7Q6tH4THtHjbyI459bFHmIoftKfD4gh5t51XS9/rO/nVSySO6Zi9c/+rRzlWlitWzGcKiZZIWoBdkV8Dw/fbcn6Ulod7dMnAzmXOyKH2QtJ26OAoOUvXWvL0TZrLM3xgLFPIx+5tM5Yx+4w0yzlaSNP41jxru+WOIwebCgHfVAxyH61oU3LUKk2rvD16Auuas4tXnRyoMzpV70Xv/MmqKOzkuvss4Lxe4HaFwLZq2oPD3AQdhkxkFyrpnfS21RYvbEje9sQjxgKB4FiZYlZj1Ypv4QASKIfYQMljaQ92jA88OEYmQdA0BOPnAQ3ycAlBEDa70tb6wmRUCXnyDyYXsOmcQvT0fNpCH41olaJH2/5i8tyGpcQgEdDh4Sy3NEaVrO2+MJHiNthKwOXlukSAvBom7bq9tVGPTwf+nibq4PqerFhzvpATAF5tEPli0eGGhi09hBp69dZ38Nl0XydrTgH2JycNei4NHN5RDxSv/nKgOFR0EFqEVAFl0k5B7gaLq+IWg8u0TZme8/18mu5Cj3irZwDXEdLPNd2pYsS7O111vsJMK1n7jZH2iSc9dWIE57R50mI78jeqK7rgy78srbdLKT8lcjjcQUeaYHjl7PTKKnvr3lWdUGSdBB/10y5uA0KcOL81au9D6qOxEVDuAy7X4GGbYDKu+bZ5Q3O+gsxjw5V90juPwwPqzQaISzA2pjqilmga4PKTJrD5f5l6by8aQvVm54vPTai+T7FFQXgIn6DAedCqFqNXAbtMOLqJweeAXRIikw5Cb4arm3O1Vs9KXhR706fxaCO4F3RaKz/Q5EduKIAXDgUZtk2LIlVb0E3Zai3hTPi5bh/E0T+0gwT/eGCBzMXJElA3nOGGuFHTuvy67cC7tzqE+cnwRd0uK9GTwbVZc8DYhRRMmRP4RDWfFTuj9C9njeo3InT/1K8giuRJGlRt0B+6oVK7k8vLuzfnYwQb28Yu0yX4U5Rt5EPU10DHcuDtW3aNQB93FsPBKuJ67jzfU4QfGEnZyWkjyQvwx3D/eSScUFY6YQGROSHB8F9I+2Ey2qnETZ5zjmX6h3tRA6KCo+QUJiwDZ2tonMtAbgeppPa2ke21WW3Tnk30cErD4+fFpR+xHChc4BL7dJMDMJimwnK76WpMTJT9pv7Hn3xSY7rXNjeBvOKR2eEp1iA6uWBE6HRpn4jMSu3gNm/bYGae0PKK73CpbSNhEee49XVZrWos4QTBMA5nT8XyWUpKfGu8ZxO8wjpoQmimRBuGIqcAO1SdAQ80USXigHOtQDhAJer85P4SFJlATN2R1F+BbsrUaGKnHcB6LdBT1cIpxMAAEJBSKIUMg+nXMlpajSoN/Ouj0NJXQTg9BU6nQAAiFGxlQx05ChJiCi436t8jUUNYXTQQwTbLlDbfPEifyAj4Urb5a/LEgqnf+FLWjJy7TE7rMJj5fw1qF98q0BUc9a2NiTADTF1+8irtw3YBZ7WawVEWp5h7i1/bXnwZlbHP0/ALA4uc1aAc+q2ekPjbGxAWgcT3+ZFuQT3n03ExCQ5PzViyjMYpjxxoHpJAQSX1nliTHs7pz/zRPDa3A+u+thnPAC42ofiTyQGl98beIWG/Pg3SZ066hAwf4Yw4vtEHYyzEfLjNlSUy0nwFTTE2RZgxDbeYLaeuXOyeJwdHW6CRz/+ddZBfeQzdvUqxDV48lBsxmnqufJsZOTl0MIXyqQwT3qm+lQm+APUGMcBD4dtb9UmAfjsQfCBk7e8/HwLcuHJ+Z41jFON6pCXROgUbKtZXOgk+DLwT1X9CkE0iRvLt67ChbnfSoXtiwvdBj0RpL40K26yQ92ddDTE3F5cVQaSU3cclNug57pBZ+9N2oMr35NpxXRpjx4jERSycltNhDIB+NQBAAAYYdVWPs+ZmEX50Cz7CgRPP0IQiQSOJABXWZnz6HAOp05E9+g833qVDfawnxY2Z3+tvxBzFsTyXIdbVoUKp3zZ4GpsyY+OI4spBrjmIh1MVW24ay6Sj4NpXs+4eDzFl7MAJxKjrb5HzoXDZJTMdKldtnHcoo4LsqRB5XodL4646iLaBgQIqAdViEYN4+hX6XSLGS5ePrEio/acCNp8Yw/dr9bVszh5LDT/UPYx3K8disprvddBJZGiPUJNnhFXXblrDXApo/yqRqvlFuAmmG6DbnC62Akr53h1tC6XiERTarVpvrxyS+MDUkGtVzPMLoLYimn5QZhwhPZbMjaFZZPOkoPguB3koTY7nMpzeVBlA3FphfF0vqHaNld3G56OR2EpD1lYJYtfhWAOE+6bmpWdE9nP9saaEau3DNCYU+6WpPvSuKgewuWAuln05oAK56yoT2MfcNy1I1Ouhlvf41VWvxVidwtg3eG+LBSpjdavL6awAq3Ht4ven/vB1M9uq4pQfmfeAahlLKw6HOdfhrtOZBY2O6OmKZyZ0g8gyC70+3bZSiiJEiLNOUzqHU29mPrZa5E3jEp7Sewn4WQBmLdtmfspzk9ODxgEZHvDP18rspBosb7oaukBi9dEnGvN0GiWvXTXbTNGb709enzqfD0qArDnFcLcMb0Wcl6EUiYlT1AmNt2iy+xAZk8qmftMYjf+2Lszl32uVZ8jPsuH4o3Bj8kCoSy6CDD7OYNkhezFxL2thzVln4iT9x/Klu7KOGS83Tr70zYJUZsvVJCwShC8rUYxwkGT7tONg0gKT8UkCmn6GwKJK1LQGdoII6nXsABY8x434VRLGLKSVeOsI5LyV8G1q3414jYSXgsHaf5XeLqXMghy72d+z/98Rp7OJ9lqb7h52A26UpOP+WKsSIZuuVQ1v2aBeRt0/51LJn4p2fPYYQNkVlYNCpY4fvw9Ju/dKRFt8zVE9YUj8aIK5lVooxQG+zyLrY8cocd0gbadjKE5uCN0PPszZ4b9y4HKu3EKS6I9fxxRQhvOt7pfOcVxhD3SYvtOUmPKOUCVkuPuAok40s+AHwprERx5SAXlNNrcHlzkpwl8hHgP2JityvPCtvAsxYZB25vF+vs306D7J6UW54HmowA24gFnAh1DqOn2+SijmZu7FnnVzqxZVa2hf6qR3pJdfz30OeXSa8H8btDNT1VUzApjWR2ZYDkAQWPkWImDjZhrmngdiQ5arTSGNo50RZJIVEeO7jnxd53rz4PgI7eR2O8247x2XEvSIhTQ0smY+lATgtHeujB1j/ZJZUpIUbnE60ctB6mWxFNDa6jrbLvEABF7sRKSz0751Db6+korR01NjIofDf2TSjYXa1rWDZ376oFQvhlZc0uNEPAH8ZxkHyWpoKf/W2OjVW65lLH83OQwhydXh4yt1ZL8rXhE2bYv88Ppg2CLFFhX2coOF12Uyw9mP2G+IJ2KpcdddlgM2+ErYLlibsUC5T2a+LC2t7UDWi/DbcalaEz0gMA3h8XxKEpkw1Qx14uJ0bG4Eoc+D+EsNtqj9HO57oYK7bzzPDJJItTHVpp3YC2MOsL23LxVkoZ/K8RhHolYTbV8d0j1prqCqhuDCkvxz8ur5h8Bku8+15HMWR0wElBSOltcv7NysmI47/GBCDHRSN5x3w3K0mDZZuImp6FBGiRnWxV8krotVZ5ec1nEe+gapbyHaY2tGN3qKjyuERnxSKB1qeqy0710CSNNrV+P3i/VdpBNKJTb5rB2wB9RFZj0z/fpd0XVz8jXU11Yw46YcmGH8jLaX34irxPEBgIT7KLyofjNofVfbNAeBFhZmsy9jvYDbqvzjP5CQFWPuLrf9K/sbk1Oa1cH2CswysjBLlq/GrF5UhEfI89+7KFE1Qn6utzBwVGKNv4qEuIxhBxYy37qvXXZ86084YaR9/0LMQ5vzY+qrvCWxDAkXb65i7qWSSuDn2NpcHFWFfsA2YauGKJvdn5k50GmQX1+aOMW1fEo0uP8k6wrvzKtQgyVE6Bi55bCvD6ICmVdzPWWlsdMxm2rxF0+9MHIPdU0r4IlZhnozCzu8clmH5XvBRLocYFQvNY4j+UN9oJ6+JADDcHQuoHc3aTJl8OsnzWBoINtF4mlf+N0qnkN1sPpYdEI02e7MRI3/0gKW4y8d9IVxo4I7katGom9Vx05QqWJ7CpOl0PJGrC5VPtkNKuW1aoc6b+pWrWX3ALzAQBgaGS+bQTaSL1tN8C+yPfe8lnVjd8E1JvBYgkIlrWMQ04Zs3LUCmVlj9BItUAPoV6O/l4gg6BCCeUu08MCA9q9T8n1cQZq7K4WyretSTLsyq6xwTqcMe2NpvvzCqd9MRY576yiB0N1/w62do5iUcM0WCGfq6ZGejqhE1ZeYgS7CyRNmNN2LVmP4O08JXJfBQ2OX7k/608qvB5kuNU91gDF/S21oxejPg38BPdhY+1FRFPzsfIgCn269xKG/pOMFxqQJWx3H2iv5J7GglNZ03gpCxTof1NYVKpZtKcGGbzYnZ8J4hj3c+UU1gTQDWi6CrHjqn1KN8vKonZkOa+MjcW410XRn1gCwBPYG26FGpXLs6UXPhs2OqaM9iRjgHP0Dbsr8cJROCYMgK50fh8r0vS/ZWqKBRCD+yd5azO+qZfazbMXtsWuy0ylXW8ipYOwtrA+lzNFAEipuRfUc4eXHXhkJL5vcGlnQannQaouPpq1insGqYZyF6hRw7/Y9QUGzFRyk/9zXBiYxhgcMtv1//oD8BceFGtu141t6jLjs8d4eyI49mIBqpcu7B6ZMVnLvfnR/OuixMNCXrLo62RNywYJ7LgMR6WNy8s3xl5niS9sj7Wpkdl3z21IUyednBI9CNMub6onOBvjZQc+CcqPATRbj/LDxA3pV08AAjLFxnjp6xfWoJLNdJ61uv4Jg/3I6US1d83KWkxX78HSBi/j8EyI7KaTA9UesU8aohruDnPPFREyxibxswGCsxs+iPTi/mdE966yP7PSMEQ0CD7DuVHGfB2WPSU0PYh2Bb95GwZHdLa6dbQJ0hnJAZfZXzgJ5sbGdjxyDsCfD/nRcTs7r2fFly/zeWHnehrUf1nF/COF1p+w974WJTdFtat9VmRcRfSuyEvrPx/EnpostoWa1APdoW9UNFvtotFfB8PPh5pKjXcS6sDK86jnACguJQWDkHyUXT4H9BKDJ4C7ifzhih2vdDW+9Q7CqmPPQpy+A3BHPwWFEL+uDD0o4lnzwO23pv0vToutD8Wr8DKSWFgA+XxLHDfSC5ui5isn1jzsiyQmBM8BQlE5AlChpmXWXjZlj+rzd06FnYH7UD8gfDTgSZ9BpWwxxQ5ybXZM/n3swhQAFD8h47Pe8fMVkK9hJcv5NyYssv2xq7yS7yvz0JxWDp0quu+6+HVYfSkbKxmNqSStt9YRejhE3frxb4zf2EfF7PHQr8ksWrn91Rbq6yo31tci4ilZYztp3eq+qq5aH46M/jd3LLGiLiO5R+lTBK2EO8yM5e8DrPxAVtYcoWBMawednbXY2vsEgtodszLqUyjrpLcRPUN2TRLjqm6A1QMrDIaMwU2kWGV1xVTKDw7D4jP1O4HcAmSF/d3JLmj57ahUq7MGY6duFWNrEiWyk6abfUkjqb6dZA4pFczCov1I1pH6ILh/tkOiVLNKujuxiiOGolHkLNdj2WT5vUCLQSh+x6Bxf3ChNkbATyvgiERkCMzeolaYykmwL5ZrxKQISmFmP9m/R0A3pQQNQl7UKje3cKgWFwjLx1u5gONvx1Uosd6y6WSQVqKqUlXtWe5TK7lSHqo1H4St4+je4o5W01iOKbtldUk7VQ46wxWoilDjP8dNWiUb8m+fSYXqNufDn2PXTgCbup9EdZzzyvaJZ3rluI9neR+sh5t4VDic2ypky6A3toZsoTcQzrzTarVPwPEMMiZy1lxWzvfIXD+tlNSH8Z0qRyxcTsyqOAdhFBmpA35g4sJxVVBUWEtSR50XBFamWAoXIP53k0oL4FEVOQlmAds2SpcNAO7yUFFRNlW73J7OFzLI0IlEeZPPo/aZl2LVAiMAAWKpu0mxPMtAQ15W1tBdvVMUYSDLQbaZfunuzSFVlpWp0hsbzyzfqclhJtGX1+4GAGHLspCBQLR8+mbJT+2tda3wKpxT0ZoW2gtqk7BCwHhYEq8jyTZZJNZbUxRDYmXUnvXy9uiueE9zJyFrMlXmsWvEYix/JbK67YzdAYCAKNO9p/gcKwYl7QUVc+VQ80AypM8fgegtQnJaUed8Qk3C+L2lPtmOya5IzKINknR72vwJI4vchQAAxu254Nqtpsq39E7bK3X/RguASUbzzTheBpgZBoM1SviqQX2K2iMrnvcfpVFSIbP1mD7Aqn3hK9r80LT8UgaqU9OhRgPxcM13llqoWJCyQDOj6qAkAJQRjdkGwDQfGj89LFaIyv3mThscKxJMA9ekX1/eef/UhUMU3WIpeKICzjzNHuU0LpTMC7ooszjlWaeSxrw/zB9QfbkqjCidpDX+T05YGIsTDsWOWM7VAp4FUn1MVLSUeJk1yc1CUmkHRAiSmDUNjeZKMk66PuE+VcPST/ncvOD47IhTkQRGfNdwaxvlZHXRw/VtUgTV+JYJAKr+DJUywLYrezvqOUpik5bZt0TlaBfHSuY8qM/Z4HDbkpfLUcngxU2h9MocibzQJsPTqGRUI9HhLZcSKoP3oCyKJojaSD5GAUsbk03gUQTjWpD8xEqqMcoO3G+ybIvN9JqO/EOLp4ry64DYHG+AOLIsLgI39gLA00mTYYksjawTaIX59pB8aZa80IZ9VHtkNFRLu1DwQGa4Rq8PUznz6u9LwhQI4IbwcYfhAwaaU/sDwFDyOrtJZqt2Fg7l8dUSaJW2uCCuZvWUmT8jZFTZ9LSPtVgwvKGMqPu1Vf9K5ziJQe7O5N57+iZdRIDb+/udvsAdvhO8Idw/AL8CEsAbAMBAcC850te+ij0qy5OvpdBO9iNjtmzDOgVjN2e3G3GXU3naO53J7Mfv20Q6dGAlqvrGSnnPvOd0CYK4eqX08/brrz/Dn/8F4ctXQIJhALjP/94A3mAAuLM/vyGvFutE9c6mlr60CqtBXUOnmzpah/psazhpNe/n2tKUxPZRVE2nuhGMBnNcnKPN4ZeP39/gh6/w5Q2Gb3O9DwACuAMQwDD/JUnXlqUgYhvIvtS3LRESin+BDszx463Y+kga1CZWBfmoe0abgJHBSrwGgYepw5zxSKvAe8iIQBQ0paQwAtw+/vj7dwAEuAMAwABwm19/zMmcAUbDoBJP4oUzvajpkPVyz5QqnlIQ6vqmEtVByxuy8riaV/VWsEN/CMYKG5od1AjLdXDU83LrSYIV5o4YmfnRdADFSnOWvX376U8Ew/f5Xuj7PJsPgA+Ar3PtAYb7IgOjF7Sk+dNvlaKMXhScpAfD/lpureVURdvRNtjKVNVpt7Q8fQdJlSbZZimikucDkfcmVGonkHGz4b5Y3oA1br56OCf6CQBwFonkAt3+9OOPvwD8CvAV4PsU+wIB/AjwDgAA78V3qKeoYhk7FwPLqOfTFRgPidPDp0/ymEgBaK5T9hxOeymKdiNi1HTIBMwfLjdDbq3pTJZVnJvjHQjGIPj3t69/ALwD/DH5/fdvMADAd4AbwDvAlzkgnnX1EkxjSQ8Zmts2+ZT92oPv8+jfrwOi2ueVwy50iMmYsZnyNrMM1Dk8eyC3LJksmjeeebM0WwCYYoCfhm8E8DPQV0AE+A2GfwDcAb4AfAf4NtsB5AOPfbGSumMbRs6XOZjXVPV0I1YlHsY8J+QSg7a+NaUbhN4NU4LVIMmqgJOfw+ompkmaejokHv7nr//1O8Bv8NuvAB8AAO/fAWD+vpQ3Zf64DMTGjUw0itwC5/8kAaiOXuscWxqu9HSvDnU51qyv0XFDN1SrrXH/cgcnq4bpEQAMQAiIMADgKCe3f/3hT78D/AQ/0eTt3Gjm+3s2GIXUZOFLGPX207J8uVjawlpNEk+0+osMVOfvQPpbfiIj69Nf1VSz6E/7OOxUM5K1sZIytk+7ldbTR6+OJppMEpHMC8F9vvA23OF+A7h9G375PwACGgC/A3yZ4+BvAH8AfJ3dIWEYdQrswuDTbsjkErdQqV7cbbKmF+Eep3PUV5DTVUpIfUCaq2UOYpGpYzqM7WU58hTjseY58YEVONdmImSuDk4q/20mi0Y+v3254z8D/AT48zy9rwA/AgDAN4CP5U9pf5RcBOktOivpwWPCNRym9swcOaNzZyIFRWiUq8BM6gKjm81Dg3EKiwvx4SVdBERVIvlbSVLmoys5eI1StY9WnzYPDRZXaNYdxXEYAbwBfgACwA3ePu4A3wD+AvAd4AeAN4APgC8Av8/WAAFGWSAA9mGL7Bpz6JZoDJStisWO2a8ZKBhd1omMvlF58Zi0QiinxxidtPrt2Mvn9H3i0RzZn01j3cg+LLLlCmHprQMsiXUYH82mbAD4IKA3GN6n4124/UY/fQD8Av/7/f3f4AZ/B/oGCPDHz/D1O/z6T/DtBvcv8EaA75NZZEfC91zgAv7FtZCdkpxrr9dD1QYrPbX1deLwIjHfxyCuR9Kfpxh1NM3uECDgHd4+4GMAeiccEG4//+d//+U//v3Pv3zHb3/72+9/fPvhh/sbfvz6Tl//8eXXjx//9NP7HwBfBrrDgISICESjIMAcV2QTCHqrp4P4u/z7kQr9y4wepVqFmpmL5hcHw+JEy6SwkCB5SjGOJpjOU+t/YK5BRNjyLbTOHDfFJSPjpZQAldWn0GXetjsQIiAMBDQQwv2dAD8A8OPtDd7ubx/w/1lcuBBroJa/AAAAAElFTkSuQmCC\n",
|
| 541 |
"text/plain": [
|
| 542 |
+
"<PIL.Image.Image image mode=RGB size=256x256 at 0x7FA20677A400>"
|
| 543 |
]
|
| 544 |
},
|
| 545 |
+
"execution_count": 22,
|
| 546 |
"metadata": {},
|
| 547 |
"output_type": "execute_result"
|
| 548 |
}
|
| 549 |
],
|
| 550 |
"source": [
|
| 551 |
+
"custom_to_pil(np.asarray(get_images(jnp.expand_dims(greedy_output[0][0],0), model)[0]))"
|
| 552 |
]
|
| 553 |
}
|
| 554 |
],
|