apply config correctly, adjust defaults

This commit is contained in:
mertalev 2025-06-16 16:24:26 -04:00
parent 22690fa096
commit 585d093baf
No known key found for this signature in database
GPG key ID: DF6ABC77AAD98C95
10 changed files with 43 additions and 35 deletions

View file

@ -57,7 +57,7 @@ class InferenceModel(ABC):
self.load()
if model_kwargs:
self.configure(**model_kwargs)
return self._predict(*inputs, **model_kwargs)
return self._predict(*inputs)
@abstractmethod
def _predict(self, *inputs: Any, **model_kwargs: Any) -> Any: ...