Adds kwargs parsing during model loading.