mirror of
https://git.datalinker.icu/ali-vilab/TeaCache
synced 2025-12-10 05:14:24 +08:00
4 lines
73 B
Python
4 lines
73 B
Python
import torch.nn as nn
|
|
|
|
approx_gelu = lambda: nn.GELU(approximate="tanh")
|