Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: ananthsub <[email protected]>
  • Loading branch information
ananthsub committed Nov 20, 2024
1 parent d092551 commit e260265
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion nemo/collections/diffusion/encoders/conditioner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit e260265

Please sign in to comment.