Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
efederici
/
ipt-350m
like
0
Text Generation
Transformers
PyTorch
oscar-corpus/OSCAR-2301
Italian
mpt
ipt
alibi
text-generation-inference
text generation
custom_code
arxiv:
2108.12409
arxiv:
2205.14135
arxiv:
1909.08053
License:
apache-2.0
Model card
Files
Files and versions
Community
Train
Deploy
Use this model
cb62549
ipt-350m
/
fc.py
efederici
Create fc.py
cb62549
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