pepecalero

/

TorchscriptSplitModels

Like 0

• "__torch__.torch.nn.modules.container.___torch_mangle_8671.Sequential",
• "torch.FloatStorage",
• "__torch__.transformers.models.gpt2.modeling_gpt2.___torch_mangle_8661.GPT2SdpaAttention",
• "__torch__.torch.nn.modules.container.___torch_mangle_8669.ModuleList",
• "torch.BoolStorage",
• "__torch__.torch.nn.modules.container.___torch_mangle_8654.ModuleList",
• "__torch__.torch.nn.modules.dropout.___torch_mangle_8660.Dropout",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8658.Conv1D",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8657.Conv1D",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_8662.LayerNorm",
• "__torch__.transformers.models.gpt2.modeling_gpt2.___torch_mangle_8668.GPT2Block",
• "__torch__.transformers.models.gpt2.modeling_gpt2.___torch_mangle_8667.GPT2MLP",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8642.Conv1D",
• "__torch__.transformers.models.gpt2.modeling_gpt2.___torch_mangle_8653.GPT2Block",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8648.Conv1D",
• "__torch__.splitters.___torch_mangle_8655.MiddlePart",
• "__torch__.torch.nn.modules.dropout.___torch_mangle_8644.Dropout",
• "__torch__.splitters.___torch_mangle_8672.Part",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_8656.LayerNorm",
• "__torch__.transformers.activations.___torch_mangle_8665.NewGELUActivation",
• "__torch__.transformers.activations.___torch_mangle_8650.NewGELUActivation",
• "__torch__.torch.nn.modules.dropout.___torch_mangle_8651.Dropout",
• "collections.OrderedDict",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8663.Conv1D",
• "__torch__.transformers.models.gpt2.modeling_gpt2.___torch_mangle_8652.GPT2MLP",
• "__torch__.torch.nn.modules.dropout.___torch_mangle_8666.Dropout",
• "__torch__.torch.nn.modules.dropout.___torch_mangle_8645.Dropout",
• "__torch__.transformers.models.gpt2.modeling_gpt2.___torch_mangle_8646.GPT2SdpaAttention",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8643.Conv1D",
• "__torch__.splitters.___torch_mangle_8670.MiddlePart",
• "torch._utils._rebuild_tensor_v2",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_8641.LayerNorm",
• "__torch__.torch.nn.modules.dropout.___torch_mangle_8659.Dropout",
• "__torch__.torch.nn.modules.normalization.___torch_mangle_8647.LayerNorm",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8649.Conv1D",
• "__torch__.transformers.pytorch_utils.___torch_mangle_8664.Conv1D",
• "torch.DoubleStorage",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2",
• "torch.FloatStorage",
• "torch.jit._pickle.restore_type_tag",
• "collections.OrderedDict",
• "torch._utils._rebuild_tensor_v2"

How to fix it?