Project1Mvampire's picture
download
raw
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

Xet Storage Details

Size:
167 Bytes
·
Xet hash:
8b11b21ca6643f1ac840318d9d88719deed179145c6125bdfef1079adf4bb526

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.