Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Rasu23
/
testv31
like
0
Text Generation
Transformers
Safetensors
mpt
custom_code
text-generation-inference
Model card
Files
Files and versions
Community
Train
Deploy
Use this model
main
testv31
/
fc.py
Rasu23
Upload MPTForCausalLM
60ae810
verified
4 months ago
raw
history
blame
contribute
delete
No virus
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