mlx-examples/llms/mlx_lm/models
2024-12-18 13:54:14 -08:00
..
__init__.py Mlx llm package (#301) 2024-01-12 10:25:56 -08:00
base.py length masking 2024-12-17 22:35:45 -08:00
cache.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
cohere2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
cohere.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
dbrx.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
deepseek_v2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
deepseek.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
exaone.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
gemma2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
gemma.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
gpt2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
gpt_bigcode.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
gpt_neox.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
hunyuan.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
internlm2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
llama.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
mamba.py Add support for falcon-mamba (#1074) 2024-11-04 12:23:30 -08:00
minicpm.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
mixtral.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
nemotron.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
olmo2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
olmo.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
openelm.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
phi3.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
phi3small.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
phi.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
phimoe.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
phixtral.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
plamo.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
qwen2_moe.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
qwen2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
qwen.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
recurrent_gemma.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
rope_utils.py Adds EXAONE architecture. (#1145) 2024-12-09 07:58:25 -08:00
stablelm.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
starcoder2.py add mask to mlx_lm model interface 2024-12-18 13:54:14 -08:00
su_rope.py Add Phi-3.5-MoE (#946) 2024-08-24 06:52:33 -07:00
switch_layers.py Handle longer prompt/generation (#931) 2024-08-16 15:28:39 -07:00