From e2602651cf128c315a77260d3fba0780c175c093 Mon Sep 17 00:00:00 2001 From: ananthsub Date: Wed, 20 Nov 2024 19:02:25 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: ananthsub --- nemo/collections/diffusion/encoders/conditioner.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/nemo/collections/diffusion/encoders/conditioner.py b/nemo/collections/diffusion/encoders/conditioner.py index 643c5bfd5108..d8928893c659 100644 --- a/nemo/collections/diffusion/encoders/conditioner.py +++ b/nemo/collections/diffusion/encoders/conditioner.py @@ -20,7 +20,12 @@ class AbstractEmbModel(nn.Module): - def __init__(self, enable_lora_finetune: bool = False, target_block: Optional[List[str]] = None, target_module: Optional[List[str]] = None) -> None: + def __init__( + self, + enable_lora_finetune: bool = False, + target_block: Optional[List[str]] = None, + target_module: Optional[List[str]] = None, + ) -> None: super().__init__() self._is_trainable = None self._ucg_rate = None