Commit Graph

13 Commits

Author SHA1 Message Date
Awni Hannun
06ddb8414d
Fix Qwen2 and SD (#441)
* fix qwen2

* version bump

* fix list shape
2024-02-14 13:43:12 -08:00
Nripesh Niketan
f1ef378a58
Feat: update pre-commit rev (#432) 2024-02-11 07:23:27 -08:00
Awni Hannun
ec14583c2a
work with tuple shape (#393) 2024-02-01 13:03:47 -08:00
AtomicVar
2ba5d3db14
Refactor activation function and loss calculation (#325) 2024-01-16 13:42:56 -08:00
Awni Hannun
a5d6d0436c
Support Hugging Face models (#215)
* support hf direct models
2024-01-03 15:13:26 -08:00
Angelos Katharopoulos
37fd2464dc
Add an image2image example in the stable diffusion (#198) 2023-12-28 18:31:45 -08:00
Awni Hannun
27c0a8c002
Add llms subdir + update README (#145)
* add llms subdir + update README

* nits

* use same pre-commit as mlx

* update readmes a bit

* format
2023-12-20 10:22:25 -08:00
Pawel Kowalski
fc1495abaa
Stable diffusion - check model weights shape and support int for "attention_head_dim" (#85)
* Allow integer as attention_head_dim
* Reshape downloaded weights to match model if there is a mismatch
2023-12-15 13:01:02 -08:00
Awni Hannun
a99e9d551e hf correction 2023-12-12 17:08:04 -08:00
Robert McCraith
4ed942d7f5 fix: typo in variable name 2023-12-07 21:30:04 +00:00
Awni Hannun
1900564f59 format 2023-11-30 11:52:47 -08:00
Awni Hannun
31bc57c4ff add copyright in source 2023-11-30 11:08:53 -08:00
Angelos Katharopoulos
b364cc56cd Add the Llama and Stable Diffusion examples 2023-11-29 10:38:20 -08:00