remove unnecessary import

ControlNet · Feb 9, 2024 · 2349149 · 2349149
1 parent 59b78ef
commit 2349149
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/model/classifier.py b/model/classifier.py
@@ -1,6 +1,5 @@
 from typing import Optional, Union, Sequence, Dict, Literal, Any
 
-from memory_profiler import profile
 from pytorch_lightning import LightningModule
 from torch import Tensor
 from torch.nn import CrossEntropyLoss, Linear, Identity, BCEWithLogitsLoss