AMG model upgrade