tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Fix typo (#1231) [skip ci]

Browse Source
This commit is contained in:
xhedit
2024-01-31 06:46:55 -05:00
committed by GitHub
parent 4cb7900a56
commit 8608d8003e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/axolotl/utils/models.py
View File

@@ -756,7 +756,7 @@ def load_llama_adapter(model, cfg):
) )
if cfg.lora_model_dir: if cfg.lora_model_dir:
LOG.debug("Loading pretained PEFT - llama_adapter") LOG.debug("Loading pretrained PEFT - llama_adapter")
model = PeftModel.from_pretrained( model = PeftModel.from_pretrained(
model, model,
cfg.lora_model_dir, cfg.lora_model_dir,
@@ -825,7 +825,7 @@ def load_lora(model, cfg, inference=False, config_only=False):
return None, lora_config return None, lora_config
if cfg.lora_model_dir: if cfg.lora_model_dir:
LOG.debug("Loading pretained PEFT - LoRA") LOG.debug("Loading pretrained PEFT - LoRA")
model_kwargs: Any = {} model_kwargs: Any = {}
if cfg.lora_on_cpu: if cfg.lora_on_cpu:
model_kwargs["max_memory"] = {"cpu": "256GiB"} model_kwargs["max_memory"] = {"cpu": "256GiB"}