HF 4.33.1

This commit is contained in:
Henk
2023-09-06 20:21:23 +02:00
parent cf4adfc7b4
commit dfb63b2340
5 changed files with 10 additions and 12 deletions

View File

@@ -230,6 +230,8 @@ class HFInferenceModel(InferenceModel):
def _post_load(self) -> None:
self.badwordsids = koboldai_settings.badwordsids_default
self.model_type = str(self.model_config.model_type)
self.model.use_cache = True # Workaround for models that accidentally uploaded with False
# These are model specific tokenizer overrides if a model has bad defaults
if self.model_type == "llama":
# Note: self.tokenizer is a GenericTokenizer, and self.tokenizer.tokenizer is the actual LlamaTokenizer