R0402

matcha/models/components/decoder.py

@@ -2,7 +2,7 @@ import math
 from typing import Optional
 
 import torch
-import torch.nn as nn
+from torch import nn
 import torch.nn.functional as F
 from conformer import ConformerBlock
 from diffusers.models.activations import get_activation

matcha/models/components/text_encoder.py

@@ -3,10 +3,10 @@
 import math
 
 import torch
-import torch.nn as nn
+from torch import nn
 from einops import rearrange
 
-import matcha.utils as utils
+from matcha import utils
 from matcha.utils.model import sequence_mask
 
 log = utils.get_pylogger(__name__)

matcha/models/components/transformer.py

@@ -1,7 +1,7 @@
 from typing import Any, Dict, Optional
 
 import torch
-import torch.nn as nn
+from torch import nn
 from diffusers.models.attention import (
     GEGLU,
     GELU,