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 |