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