Added in model backend to the command line arguments

This commit is contained in:
ebolam
2023-05-18 16:01:17 -04:00
parent f027d8b6e5
commit 182ecff202
4 changed files with 54 additions and 27 deletions

View File

@@ -178,6 +178,8 @@ class InferenceModel:
return {}
def set_input_parameters(self, parameters):
for parameter in parameters:
setattr(self, parameter, parameters[parameter])
return
def load(self, save_model: bool = False, initial_load: bool = False) -> None: