How to use stonkgs/protstonkgs with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForPreTraining tokenizer = AutoTokenizer.from_pretrained("stonkgs/protstonkgs") model = AutoModelForPreTraining.from_pretrained("stonkgs/protstonkgs", device_map="auto")
