Initial model upload 708f203
Luca Zunino commited on
How to use lucazed/habitat_fine_tuned_vit_mae_base_not_frozen with Transformers:
# Load model directly
from transformers import AutoImageProcessor, AutoModelForPreTraining
processor = AutoImageProcessor.from_pretrained("lucazed/habitat_fine_tuned_vit_mae_base_not_frozen")
model = AutoModelForPreTraining.from_pretrained("lucazed/habitat_fine_tuned_vit_mae_base_not_frozen", device_map="auto")