Fetching metadata from the HF Docker repository...
Update app.py
fda9ec7 verified - employees Upload 9 files
- 1.52 kB initial commit
- 313 Bytes initial commit
- 4.99 kB Update app.py
- 92 Bytes Update requirements.txt
vit_pytorch_GPU_1.pt Detected Pickle imports (26)
- "torch.nn.modules.container.ModuleList",
- "transformers.models.vit.modeling_vit.ViTEncoder",
- "__builtin__.set",
- "torch.nn.modules.activation.Tanh",
- "transformers.models.vit.modeling_vit.ViTSelfOutput",
- "torch.nn.modules.conv.Conv2d",
- "transformers.models.vit.modeling_vit.ViTSdpaSelfAttention",
- "torch.FloatStorage",
- "transformers.models.vit.configuration_vit.ViTConfig",
- "torch._utils._rebuild_tensor_v2",
- "transformers.models.vit.modeling_vit.ViTEmbeddings",
- "torch.nn.modules.linear.Linear",
- "collections.OrderedDict",
- "torch._C._nn.gelu",
- "transformers.models.vit.modeling_vit.ViTPatchEmbeddings",
- "transformers.models.vit.modeling_vit.ViTIntermediate",
- "transformers.models.vit.modeling_vit.ViTModel",
- "transformers.models.vit.modeling_vit.ViTLayer",
- "transformers.models.vit.modeling_vit.ViTSdpaAttention",
- "__main__.FaceEmbeddingModel",
- "transformers.activations.GELUActivation",
- "transformers.models.vit.modeling_vit.ViTOutput",
- "transformers.models.vit.modeling_vit.ViTPooler",
- "torch.nn.modules.normalization.LayerNorm",
- "torch.nn.modules.dropout.Dropout",
- "torch._utils._rebuild_parameter"
How to fix it?
348 MB Upload vit_pytorch_GPU_1.pt