Model: Fix assorted bugs

and ignore warnings in pytest
This commit is contained in:
somebody
2023-03-09 20:59:27 -06:00
parent 3646aa9e83
commit 8c8bdfaf6a
6 changed files with 17 additions and 23 deletions

View File

@@ -30,12 +30,8 @@ class HFMTJInferenceModel(HFInferenceModel):
self,
model_name: str,
) -> None:
super().__init__()
super().__init__(model_name)
self.model_name = model_name
self.model = None
self.tokenizer = None
self.model_config = None
self.capabilties = ModelCapabilities(
embedding_manipulation=False,