rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Llama fixes for Mistral

Browse Source
This commit is contained in:
Henk
2023-09-28 01:08:53 +02:00
parent 2f3b24eba7
commit 9e51a50bcd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modeling/inference_models/hf.py
View File

@@ -232,7 +232,7 @@ class HFInferenceModel(InferenceModel):
self.model_type = str(self.model_config.model_type) self.model_type = str(self.model_config.model_type)
# These are model specific tokenizer overrides if a model has bad defaults # These are model specific tokenizer overrides if a model has bad defaults
if self.model_type == "llama": if self.model_type == "llama" or self.model_type == "mistral":
# Note: self.tokenizer is a GenericTokenizer, and self.tokenizer.tokenizer is the actual LlamaTokenizer # Note: self.tokenizer is a GenericTokenizer, and self.tokenizer.tokenizer is the actual LlamaTokenizer
self.tokenizer.add_bos_token = False self.tokenizer.add_bos_token = False
self.tokenizer.legacy = False self.tokenizer.legacy = False