Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
gretelai
/
mpt-7b
like
5
Text Generation
Transformers
PyTorch
5 datasets
mpt
Composer
MosaicML
llm-foundry
StreamingDatasets
custom_code
text-generation-inference
6 papers
License:
apache-2.0
Model card
Files
Files and versions
Community
7
Train
Deploy
Use this model
refs/pr/7
mpt-7b
/
fc.py
sami-t
Update to Latest Mosaic Version (
#2
)
c5e05a7
verified
7 months ago
raw
Copy download link
history
blame
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