DeepSpeed

dba0d968 - Update place where we missed instance of new_param not being in json_schema_extra