Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
mosaicml
/
mpt-7b-chat
like
509
Text Generation
Transformers
PyTorch
5 datasets
mpt
Composer
MosaicML
llm-foundry
custom_code
text-generation-inference
arxiv:
2205.14135
arxiv:
2108.12409
arxiv:
2010.04245
License:
cc-by-nc-sa-4.0
Model card
Files
Files and versions
Community
47
Train
Deploy
Use this model
refs/pr/38
mpt-7b-chat
/
fc.py
daking
LLM-foundry update October 17, 2023 23:04:30 (
#36
)
35112fd
10 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