Commit Graph

12 Commits

Author SHA1 Message Date
Awni Hannun
fd351850e4 fp16, abstract tokenizer a bit, format 2023-12-18 13:15:02 -08:00
Juarez Bochi
55f204dd3a
Load config from HF to support any model 2023-12-18 08:42:06 -05:00
Juarez Bochi
b2a3782a96
Add argument to generate float16 npz 2023-12-18 08:21:20 -05:00
Awni Hannun
34843ddeb2 format 2023-12-17 21:30:28 -08:00
Awni Hannun
688a6e1e78 with cache 2023-12-17 17:35:53 -08:00
Juarez Bochi
31da1b0dab
LM head 2023-12-16 14:44:15 -05:00
Juarez Bochi
392b7a2f98
translate pytorch to mx 2023-12-15 16:51:01 -05:00
Juarez Bochi
330f024d1c
Move position biases to attention module 2023-12-15 11:30:17 -05:00
Juarez Bochi
d0497ddc0b
Load decoder weights 2023-12-15 10:50:04 -05:00
Juarez Bochi
009ed0179c
Load position bias embeddings 2023-12-15 10:16:11 -05:00
Juarez Bochi
c0001a94f2
Load all encoder weights 2023-12-14 15:38:41 -05:00
Juarez Bochi
bca5ca4f98
Add skeleton 2023-12-14 15:21:36 -05:00