automatic/modules/omnigen2/models/transformers/components.py

5 lines
111 B
Python

import torch.nn.functional as F
def swiglu(x, y):
return F.silu(x.float(), inplace=False).to(x.dtype) * y