refactor: clean nn module access

modeling_stablelm_alpha.py

@@ -112,7 +112,7 @@ class MLP(nn.Module):
         ff_dim = int(8 * hidden_size / 3)
         intermediate_size = multiple_of * ((ff_dim + multiple_of - 1) // multiple_of)
 
-        self.gate_proj = torch.nn.Linear(hidden_size, 2 * intermediate_size, bias=False)
+        self.gate_proj = nn.Linear(hidden_size, 2 * intermediate_size, bias=False)
         self.out_proj = nn.Linear(intermediate_size, hidden_size, bias=False)
         self.act = nn.SiLU()
 
@@ -121,7 +121,7 @@ class MLP(nn.Module):
         return self.out_proj(ff * self.act(ff_gate))
 
 
-class RotaryEmbedding(torch.nn.Module):
+class RotaryEmbedding(nn.Module):
     def __init__(
         self,
         dim: int,