def create_model_with_temperature_scaling(config: ModelConfigBase) -> Any: """ Create a model with temperature scaling by wrapping the result of config.create_model with ModelWithTemperature, if temperature scaling config has been provided, otherwise return the result of config.create_model """ # wrap the model around a temperature scaling model if required model = config.create_model() return model
def create_model_with_temperature_scaling(config: ModelConfigBase) -> Any: """ Create a model with temperature scaling by wrapping the result of config.create_model with ModelWithTemperature, if temperature scaling config has been provided, otherwise return the result of config.create_model """ # wrap the model around a temperature scaling model if required model = config.create_model() if isinstance(config, SequenceModelBase) and config.temperature_scaling_config: model = ModelWithTemperature(model, config.temperature_scaling_config) return model