Hugging Face
Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up
LaaZa
/
sealion-gptq-test
like
0
Text Generation
Transformers
mpt
custom_code
License:
cc-by-nc-sa-4.0
Model card
Files
Files and versions
Community
Train
Deploy
Use this model
95a93d0
sealion-gptq-test
/
fc.py
LaaZa
Upload 20 files
95a93d0
over 1 year ago
raw
Copy download link
history
blame
Safe
167 Bytes
from
torch
import
nn
FC_CLASS_REGISTRY = {
'torch'
: nn.Linear}
try
:
import
transformer_engine.pytorch
as
te
FC_CLASS_REGISTRY[
'te'
] = te.Linear
except
:
pass