Transformers How to use nmanali/lora_model_ep1_2stage with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("nmanali/lora_model_ep1_2stage", device_map="auto") 
