Transformers How to use InfernoDeep/CustomBeit with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoImageProcessor, BeitForSemanticSegmentation
processor = AutoImageProcessor.from_pretrained("InfernoDeep/CustomBeit")
model = BeitForSemanticSegmentation.from_pretrained("InfernoDeep/CustomBeit", device_map="auto") 
# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, BeitForSemanticSegmentation processor = AutoImageProcessor.from_pretrained("InfernoDeep/CustomBeit") model = BeitForSemanticSegmentation.from_pretrained("InfernoDeep/CustomBeit", device_map="auto")