Update app.py
1356bea
verified
-
1.52 kB
initial commit
-
52.2 kB
added
-
252 Bytes
initial commit
-
10.4 kB
Update app.py
model_v1.0.pt
Detected Pickle imports (24)
- "transformers.models.vit.modeling_vit.ViTOutput",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.activations.GELUActivation",
- "collections.OrderedDict",
- "torch.nn.modules.dropout.Dropout",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "torch._utils._rebuild_parameter",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "torch.nn.modules.container.ModuleList",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "torch.nn.modules.conv.Conv2d",
- "torch._C._nn.gelu",
- "transformers.models.vit.modeling_vit.ViTForImageClassification",
- "__builtin__.set",
- "torch.FloatStorage",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "transformers.models.vit.modeling_vit.ViTModel",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
343 MB
init commit
model_v2.0.pt
Detected Pickle imports (24)
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "__builtin__.set",
- "transformers.models.vit.modeling_vit.ViTSelfAttention",
- "transformers.models.vit.modeling_vit.ViTForImageClassification",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "torch.nn.modules.container.ModuleList",
- "torch.FloatStorage",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.linear.Linear",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "torch.nn.modules.dropout.Dropout",
- "torch._C._nn.gelu",
- "torch.nn.modules.normalization.LayerNorm",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "collections.OrderedDict",
- "torch.nn.modules.conv.Conv2d",
- "transformers.models.vit.modeling_vit.ViTAttention",
- "transformers.activations.GELUActivation",
- "transformers.models.vit.modeling_vit.ViTModel"
How to fix it?
343 MB
added modelV2
-
103 Bytes
Update requirements.txt
-
585 kB
MODIFIED
-
183 kB
MODIFIED