Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Hardware
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

ajaye2
/
vit_mae_img_large

Image Classification
Transformers
PyTorch
vit_mae_custom
custom_code
Model card Files Files and versions
xet
Community
1

Instructions to use ajaye2/vit_mae_img_large with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • Transformers

    How to use ajaye2/vit_mae_img_large with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("image-classification", model="ajaye2/vit_mae_img_large", trust_remote_code=True)
    pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")
    # Load model directly
    from transformers import AutoModelForImageClassification
    model = AutoModelForImageClassification.from_pretrained("ajaye2/vit_mae_img_large", trust_remote_code=True, dtype="auto")
  • Notebooks
  • Google Colab
  • Kaggle
vit_mae_img_large
1.21 GB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 2 commits
ajaye2's picture
ajaye2
Upload model
1f6a0a1 almost 3 years ago
  • .gitattributes
    1.52 kB
    initial commit almost 3 years ago
  • ViTMAETimm.py
    1.39 kB
    Upload model almost 3 years ago
  • config.json
    394 Bytes
    Upload model almost 3 years ago
  • pytorch_model.bin

    Detected Pickle imports (3)

    • "collections.OrderedDict",
    • "torch._utils._rebuild_tensor_v2",
    • "torch.FloatStorage"

    What is a pickle import?

    1.21 GB
    xet
    Upload model almost 3 years ago