Remove setMemoryBudget, it has no effect anyway.