Fetching metadata from the HF Docker repository...
Update app.py
43789bc - 1.48 kB initial commit
- 230 Bytes initial commit
- 2.73 kB Update app.py
model-cpu.pkl Detected Pickle imports (19)
- "transformers.models.opt.modeling_opt.OPTAttention",
- "torch.nn.modules.sparse.Embedding",
- "transformers.models.opt.modeling_opt.OPTLearnedPositionalEmbedding",
- "transformers.models.opt.modeling_opt.OPTDecoder",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.opt.modeling_opt.OPTDecoderLayer",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict",
- "transformers.generation.configuration_utils.GenerationConfig",
- "torch.nn.modules.activation.ReLU",
- "transformers.models.opt.modeling_opt.OPTForCausalLM",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "torch.float16",
- "transformers.models.opt.configuration_opt.OPTConfig",
- "transformers.models.opt.modeling_opt.OPTModel",
- "__builtin__.set"
How to fix it?
501 MB Upload model-cpu.pkl - 18 Bytes Create requirements.txt