Transformers How to use mole-code/org.springframework.ai-codegen-2B-mono-prefix with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("mole-code/org.springframework.ai-codegen-2B-mono-prefix", device_map="auto") 
