Transformers How to use sohammandal01/gpt2-aligned-ppo-prefix-assignment with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("sohammandal01/gpt2-aligned-ppo-prefix-assignment", device_map="auto") 
