This Model is currently anonymized during the paper review process.
The AP-MAE transformer model design and configuration is available in the reproduction package attached to the submission
This version of AP-MAE is trained on attention heads generated by StarCoder2-3B during inference. The inference task used for generating attention outputs is FiM token prediction for a random 3-10 length masked section of Java code, with exactly 256 tokens of surrounding context.
Usage:
from ap_mae import APMAE
model = APMAE.from_pretrained(
"AISE-TUDelft/AP-MAE-SC2-3B"
)
