Duplicated from LLM-LAT/robust-llama3-8b-instruct
How to use FTK11558/robust-llama3-8b-instruct-APS with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="FTK11558/robust-llama3-8b-instruct-APS", trust_remote_code=True, device_map="auto")
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("FTK11558/robust-llama3-8b-instruct-APS", trust_remote_code=True) model = AutoModel.from_pretrained("FTK11558/robust-llama3-8b-instruct-APS", trust_remote_code=True, device_map="auto")
