great job! which model does it implement? gpt-2?