Team Ai
Modelpublic

replit/replit-code-v1_5-3b

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
316likes457downloads
rootfc.py
fc.py7 linesDownload Raw Back to root
1from torch import nn2FC_CLASS_REGISTRY = {'torch': nn.Linear}3try:4    import transformer_engine.pytorch as te5    FC_CLASS_REGISTRY['te'] = te.Linear6except:7    pass