Update code/inference.py
8ac4df3
-
code
Update code/inference.py
-
1.48 kB
initial commit
-
4.33 kB
Upload 7 files
-
1.35 kB
Create default_template.txt
-
2.18 kB
Update handler.py
-
456 kB
Upload 7 files
-
18 Bytes
Create requirements.txt
-
470 Bytes
Upload 7 files
-
2.14 MB
Upload 7 files
-
709 Bytes
Upload 7 files
torch_model.pt
Detected Pickle imports (21)
- "transformers.models.gptj.modeling_gptj.GPTJBlock",
- "transformers.models.gptj.modeling_gptj.GPTJAttention",
- "transformers.generation.configuration_utils.GenerationConfig",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.gptj.modeling_gptj.GPTJModel",
- "torch.FloatStorage",
- "torch.nn.modules.sparse.Embedding",
- "transformers.activations.NewGELUActivation",
- "collections.OrderedDict",
- "torch.BoolStorage",
- "transformers.models.gptj.configuration_gptj.GPTJConfig",
- "torch.nn.modules.dropout.Dropout",
- "torch.nn.modules.normalization.LayerNorm",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.gptj.modeling_gptj.GPTJForCausalLM",
- "torch._utils._rebuild_parameter",
- "transformers.models.gptj.modeling_gptj.GPTJMLP",
- "torch.float16",
- "__builtin__.set",
- "torch.HalfStorage",
- "torch.nn.modules.linear.Linear"
How to fix it?
12.2 GB
Upload 7 files
-
798 kB
Upload 7 files
