sdmlai
/

plantdoc-predictor

Model card Files Files and versions

plantdoc-predictor / models /vit_tiny_v1

22.7 MB

Ctrl+K

Ctrl+K

1 contributor

History: 1 commit

This model has 4 files scanned as suspicious.

sdmlai's picture

Added vit variants, deit models

bf61a1e 6 months ago

vit_tiny_v1.pth
Detected Pickle imports (20)
- "collections.OrderedDict",
- "timm.models.vision_transformer.Block",
- "torch._utils._rebuild_parameter",
- "torch.nn.modules.conv.Conv2d",
- "torch.nn.modules.linear.Linear",
- "timm.layers.mlp.Mlp",
- "timm.layers.attention.Attention",
- "timm.layers.norm.LayerNorm",
- "torch.nn.modules.linear.Identity",
- "torch.FloatStorage",
- "torch.nn.modules.dropout.Dropout",
- "torch.device",
- "timm.layers.patch_embed.PatchEmbed",
- "torch.nn.parallel.data_parallel.DataParallel",
- "timm.models.vision_transformer.VisionTransformer",
- "timm.layers.format.Format",
- "__builtin__.set",
- "torch.nn.modules.activation.GELU",
- "torch.nn.modules.container.Sequential",
- "torch._utils._rebuild_tensor_v2"
How to fix it?
22.7 MB
xet

Added vit variants, deit models 6 months ago