Edit make average model to reduce memory usage