liuhuayong/mlx-examples
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
fe5edee360
mlx-examples/llms/mlx_lm/models
History
Awni Hannun 8b05bb6d18
[mlx-lm] Use sdpa in llama / mistral model (#515) 
* use sdpa

* update a few more models

* version

* fix stablelm type
2024-03-07 17:41:23 -08:00
..
__init__.py Mlx llm package (#301) 2024-01-12 10:25:56 -08:00
base.py Mlx llm package (#301) 2024-01-12 10:25:56 -08:00
gemma.py [mlx-lm] Use sdpa in llama / mistral model (#515) 2024-03-07 17:41:23 -08:00
layers.py [mlx-lm] Add precompiled normalizations (#451) 2024-02-22 12:40:55 -08:00
llama.py [mlx-lm] Use sdpa in llama / mistral model (#515) 2024-03-07 17:41:23 -08:00
mixtral.py Enable unit testing in Circle and start some MLX LM tests (#545) 2024-03-07 09:31:57 -08:00
olmo.py Enable unit testing in Circle and start some MLX LM tests (#545) 2024-03-07 09:31:57 -08:00
phi.py Enable unit testing in Circle and start some MLX LM tests (#545) 2024-03-07 09:31:57 -08:00
phixtral.py [mlx-lm] Add precompiled normalizations (#451) 2024-02-22 12:40:55 -08:00
plamo.py Enable unit testing in Circle and start some MLX LM tests (#545) 2024-03-07 09:31:57 -08:00
qwen.py Enable unit testing in Circle and start some MLX LM tests (#545) 2024-03-07 09:31:57 -08:00
qwen2.py [mlx-lm] Use sdpa in llama / mistral model (#515) 2024-03-07 17:41:23 -08:00
stablelm.py [mlx-lm] Use sdpa in llama / mistral model (#515) 2024-03-07 17:41:23 -08:00
starcoder2.py [mlx-lm] Use sdpa in llama / mistral model (#515) 2024-03-07 17:41:23 -08:00