mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-12-15 01:42:31 +08:00
feat(mlx-lm): add lora hypeparameters in lora layer (#366)
* feat(mlx-lm): add lora hypeparameters in lora layer * chore: address comments
This commit is contained in:
@@ -278,6 +278,7 @@ def load(
|
||||
model = load_model(model_path)
|
||||
if adapter_file is not None:
|
||||
model = apply_lora_layers(model, adapter_file)
|
||||
model.eval()
|
||||
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_path, **tokenizer_config)
|
||||
return model, tokenizer
|
||||
|
||||
Reference in New Issue
Block a user