Disable mmap when applying lora weights. Closes #107

This commit is contained in:
Andrei Betlen 2023-04-23 14:53:17 -04:00
parent e99caedbbd
commit 7230599593

View file

@ -79,7 +79,7 @@ class Llama:
self.params.f16_kv = f16_kv
self.params.logits_all = logits_all
self.params.vocab_only = vocab_only
self.params.use_mmap = use_mmap
self.params.use_mmap = use_mmap if lora_path is None else False
self.params.use_mlock = use_mlock
self.params.embedding = embedding