1 files changed, 0 insertions, 4 deletions

diff --git a/training/optimization.py b/training/optimization.py
index 14c2bd5..dd84f9c 100644
--- a/training/optimization.py
+++ b/training/optimization.py
@@ -5,10 +5,6 @@ from functools import partial
 import torch
 from torch.optim.lr_scheduler import LambdaLR
 
-from diffusers.utils import logging
-
-logger = logging.get_logger(__name__)
-
 
 class OneCyclePhase(NamedTuple):
     step_min: int