How to use samiths/test-llama-1B with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("samiths/test-llama-1B", device_map="auto")
