ds_base.py 337 B

1234567891011
  1. '''Copyright The Microsoft DeepSpeed Team'''
  2. import torch.nn as nn
  3. class DeepSpeedTransformerBase(nn.module):
  4. def __init__(self):
  5. pass
  6. # this would be the new clean base class that will replace DeepSpeedTransformerInference.
  7. # we currently don't know how this will look like but keeping it here as a placeholder.