Fetching metadata from the HF Docker repository...
Upload model-cpu.pkl
3e8eabc
-
1.48 kB
initial commit
-
230 Bytes
initial commit
-
2.75 kB
Create 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