6 lines
147 B
Python
6 lines
147 B
Python
|
# https://github.com/ridgerchu/matmulfreellm
|
||
|
|
|
||
|
|
import torch
|
||
|
|
import torch.nn.functional as F
|
||
|
|
|
||
|
|
from mmfreelm.models import HGRNBitConfig, HGRNBitModel
|