add config_path to ModelRecordChanges

This commit is contained in:
maryhipp 2024-03-06 14:09:00 -05:00 committed by psychedelicious
parent a00369a67a
commit cdc0d0c182

View File

@ -79,6 +79,7 @@ class ModelRecordChanges(BaseModelExcludeNull):
description="The prediction type of the model.", default=None
)
upcast_attention: Optional[bool] = Field(description="Whether to upcast attention.", default=None)
config_path: Optional[str] = Field(description="Path to config file for model", default=None)
class ModelRecordServiceBase(ABC):