dvdmrs09

/

peft-gemma-2b

like 0

• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch._utils._rebuild_tensor_v2",
• "torch.ByteStorage",
• "deepspeed.utils.tensor_fragment.fragment_address"

How to fix it?

• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch._utils._rebuild_tensor_v2",
• "torch.ByteStorage",
• "deepspeed.utils.tensor_fragment.fragment_address"

How to fix it?

• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "torch.ByteStorage",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler",
• "torch.FloatStorage",
• "deepspeed.utils.tensor_fragment.fragment_address"

How to fix it?

• "deepspeed.runtime.zero.config.ZeroStageEnum",
• "deepspeed.runtime.fp16.loss_scaler.LossScaler",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch._utils._rebuild_tensor_v2",
• "torch.ByteStorage",
• "deepspeed.utils.tensor_fragment.fragment_address"

How to fix it?

• "torch._utils._rebuild_tensor_v2",
• "torch.BFloat16Storage",
• "torch.ByteStorage",
• "torch.Size",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "__builtin__.set"

How to fix it?