fix

Author: TheLastBen

fast-DreamBooth.ipynb

@@ -66,6 +66,9 @@
     "\n",
     "    os.environ['PYTHONWARNINGS'] = 'ignore'\n",
     "    !sed -i 's@HfFolder, cached_download@HfFolder@g' /usr/local/lib/python3.11/dist-packages/diffusers/utils/dynamic_modules_utils.py\n",
+    "    !sed -i 's@from pytorch_lightning.loggers.wandb import WandbLogger  # noqa: F401@@g' /usr/local/lib/python3.11/dist-packages/pytorch_lightning/loggers/__init__.py\n",
+    "    !sed -i 's@from .mailbox import ContextCancelledError@@g' /usr/local/lib/python3.11/dist-packages/wandb/sdk/lib/retry.py\n",
+    "    !sed -i 's@raise ContextCancelledError(\"retry timeout\")@print(\"retry timeout\")@g' /usr/local/lib/python3.11/dist-packages/wandb/sdk/lib/retry.py\n",
     "    !rm -r /usr/local/lib/python3.11/dist-packages/tensorflow*\n",
     "\n",
     "print('\u001b[1;32mDone, proceed')"

fast_stable_diffusion_AUTOMATIC1111.ipynb

@@ -148,7 +148,9 @@
     "  !rm -r /usr/local/lib/python3.11/dist-packages/tensorflow*\n",
     "  os.environ['PYTHONWARNINGS'] = 'ignore'\n",
     "  !sed -i 's@text = _formatwarnmsg(msg)@text =\\\"\\\"@g' /usr/lib/python3.11/warnings.py\n",
-    "  !sed -i 's@raise AttributeError(f\"module {module!r} has no attribute {name!r}\")@@g' /usr/local/lib/python3.11/dist-packages/jax/_src/deprecations.py\n",
+    "  !sed -i 's@from pytorch_lightning.loggers.wandb import WandbLogger  # noqa: F401@@g' /usr/local/lib/python3.11/dist-packages/pytorch_lightning/loggers/__init__.py\n",
+    "  !sed -i 's@from .mailbox import ContextCancelledError@@g' /usr/local/lib/python3.11/dist-packages/wandb/sdk/lib/retry.py\n",
+    "  !sed -i 's@raise ContextCancelledError(\"retry timeout\")@print(\"retry timeout\")@g' /usr/local/lib/python3.11/dist-packages/wandb/sdk/lib/retry.py\n",
     "\n",
     "clear_output()\n",
     "inf('\\u2714 Done','success', '50px')\n",