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