diff --git a/colabs/diffusers/sdxl-compel.ipynb b/colabs/diffusers/sdxl-compel.ipynb
index 4920efd5..514f8173 100644
--- a/colabs/diffusers/sdxl-compel.ipynb
+++ b/colabs/diffusers/sdxl-compel.ipynb
@@ -64,7 +64,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "wandb.init(project=\"stable-diffusion-xl\", entity=\"geekyrakshit\", job_type=\"text-to-image-compel\", save_code=True)\n",
+    "wandb.init(project=\"stable-diffusion-xl\", job_type=\"text-to-image-compel\")\n",
     "\n",
     "config = wandb.config\n",
     "config.stable_diffusion_checkpoint = \"stabilityai/stable-diffusion-xl-base-1.0\"\n",
diff --git a/colabs/diffusers/sdxl-text-to-image.ipynb b/colabs/diffusers/sdxl-text-to-image.ipynb
index 3692ce12..444717ee 100644
--- a/colabs/diffusers/sdxl-text-to-image.ipynb
+++ b/colabs/diffusers/sdxl-text-to-image.ipynb
@@ -63,7 +63,7 @@
    "outputs": [],
    "source": [
     "# initialize a wandb run\n",
-    "wandb.init(project=\"stable-diffusion-xl\", entity=\"geekyrakshit\", job_type=\"text-to-image\", save_code=True)\n",
+    "wandb.init(project=\"stable-diffusion-xl\", job_type=\"text-to-image\")\n",
     "\n",
     "# define experiment configs\n",
     "config = wandb.config\n",