Special-case GBM model builder
This is a temporary fix.