How to use shivanis14/output with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("shivanis14/output", device_map="auto")
