How to use Salesforce/codet5p-2b with Transformers:
# Load model directly from transformers import AutoModelForSeq2SeqLM model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5p-2b", trust_remote_code=True, dtype="auto")
Adds _no_split_modules attribute on the EncoderDecoder class for accelerate support
· Sign up or log in to comment