minor simplify
This commit is contained in:
@@ -5,9 +5,6 @@ import torch.nn.functional as F
|
||||
|
||||
class SiluAndMul(nn.Module):
|
||||
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
|
||||
@torch.compile
|
||||
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
||||
x, y = x.chunk(2, -1)
|
||||
|
||||
Reference in New Issue
Block a user