Fetching metadata from the HF Docker repository...
Added the model file
77c444a
verified
-
1.52 kB
initial commit
-
246 Bytes
initial commit
-
6.7 kB
Added app.py
finetuned_codet5_small.pkl
Detected Pickle imports (23)
- "transformers.models.t5.modeling_t5.T5Attention",
- "torch.nn.modules.dropout.Dropout",
- "transformers.models.t5.modeling_t5.T5LayerSelfAttention",
- "transformers.models.t5.modeling_t5.T5LayerFF",
- "transformers.models.t5.configuration_t5.T5Config",
- "torch.nn.modules.activation.ReLU",
- "torch._utils._rebuild_tensor_v2",
- "torch._utils._rebuild_parameter",
- "transformers.models.t5.modeling_t5.T5Stack",
- "collections.OrderedDict",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.t5.modeling_t5.T5LayerNorm",
- "transformers.generation.configuration_utils.GenerationConfig",
- "transformers.models.t5.modeling_t5.T5Block",
- "transformers.models.t5.modeling_t5.T5DenseActDense",
- "transformers.generation.configuration_utils.CompileConfig",
- "torch.nn.modules.linear.Linear",
- "transformers.models.t5.modeling_t5.T5ForConditionalGeneration",
- "transformers.models.t5.modeling_t5.T5LayerCrossAttention",
- "__builtin__.set",
- "torch.FloatStorage",
- "torch.float32",
- "torch.nn.modules.sparse.Embedding"
How to fix it?
242 MB
Added the model file
-
116 Bytes
Added requirements.txt