Add basic hf backend

This commit is contained in:
somebody
2023-07-08 17:12:16 -05:00
parent f9c38acea8
commit 20b4b4bcef
2 changed files with 106 additions and 0 deletions

View File

@@ -17,6 +17,7 @@ class HFInferenceModel(InferenceModel):
self.model_config = None
#self.model_name = model_name
self.hf_torch = False
self.model = None
self.tokenizer = None
self.badwordsids = koboldai_settings.badwordsids_default