mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-01 12:49:50 +08:00
DBRX (#628)
* dbrx * format * format * comments * change scores slightly * remove inadvertant import
This commit is contained in:
@@ -60,6 +60,8 @@ def linear_to_lora_layers(
|
||||
keys = set(["att_proj"])
|
||||
elif model.model_type == "phi-msft":
|
||||
keys = set(["mixer.Wqkv", "moe.gate"])
|
||||
elif model.model_type == "dbrx":
|
||||
keys = set(["norm_attn_norm.attn.Wqkv", "ffn.router.layer"])
|
||||
else:
|
||||
raise ValueError(f"Lora does not support {model.model_type}")
|
||||
|
||||
|
Reference in New Issue
Block a user