style: lint

tools/inference/inference_pipeline.ipynb

@@ -240,7 +240,7 @@
     "import random\n",
     "\n",
     "# create a random key\n",
-    "seed = random.randint(0, 2 ** 32 - 1)\n",
+    "seed = random.randint(0, 2**32 - 1)\n",
     "key = jax.random.PRNGKey(seed)"
    ]
   },

tools/train/train.py

@@ -58,7 +58,7 @@ from dalle_mini.model import (
 )
 
 cc.initialize_cache(
-    "/home/boris/dalle-mini/jax_cache", max_cache_size_bytes=5 * 2 ** 30
+    "/home/boris/dalle-mini/jax_cache", max_cache_size_bytes=5 * 2**30
 )
 
 logger = logging.getLogger(__name__)