diff --git a/colabs/pytorch-lightning/Track_PyTorch_Lightning_with_Fabric_and_Wandb.ipynb b/colabs/pytorch-lightning/Track_PyTorch_Lightning_with_Fabric_and_Wandb.ipynb index 43f8170e..07dd5523 100644 --- a/colabs/pytorch-lightning/Track_PyTorch_Lightning_with_Fabric_and_Wandb.ipynb +++ b/colabs/pytorch-lightning/Track_PyTorch_Lightning_with_Fabric_and_Wandb.ipynb @@ -72,8 +72,6 @@ "outputs": [], "source": [ "%%capture\n", - "# Remove when PR Merged\n", - "!pip install git+https://github.com/ash0ts/lightning.git@fabric-wandb\n", "!pip install wandb" ] }, @@ -115,7 +113,8 @@ "source": [ "import lightning as L\n", "import torch; import torchvision as tv\n", - "from lightning.fabric.loggers import WandbLogger" + "from wandb.integration.lightning.fabric import WandbLogger\n", + "\n" ] }, {