Allow to specify a model configuration when publishing a new model version

Training workers need to store the relevant model configuration used during training so that it will be available at inference time.

Please expose a new configuration: dict argument on publish_model_version and use the configuration when making the CreateModelVersion call later on.