17 Commits

Author	SHA1	Message	Date
Awni Hannun	7900a6c22c	fix	2025-02-24 09:10:24 -08:00
Awni Hannun	2edcc0355f	cleanup	2025-02-24 09:07:07 -08:00
Awni Hannun	9392bc70f7	cleanup	2025-02-24 08:51:12 -08:00
Shunta Saito	fb1559e1f3	Add __iter__ and __next__ methods to PlamoCache	2025-02-23 17:33:19 +09:00
Shunta Saito	e2d9d619c4	Add state property to PlamoCache	2025-02-23 16:06:38 +09:00
Shunta Saito	21c0abaf23	Fix to use repeat instead of tile	2025-02-23 14:54:23 +09:00
Shunta Saito	d7426c7750	Do not pass mask to prepare_inputs_for_generation	2025-02-23 14:47:49 +09:00
Shunta Saito	31225f4960	Remove unnecessary code and add a test for plamo2	2025-02-23 14:43:14 +09:00
Shunta Saito	d4e688edc3	Fix reference to layers	2025-02-22 19:55:42 +09:00
Shunta Saito	9f422b4729	Fix import	2025-02-15 07:04:53 +09:00
Shunta Saito	28f3f3adab	Give all inputs when it's the first time call of model	2025-02-15 07:04:53 +09:00
Shunta Saito	103c6616c4	Remove unused code part	2025-02-15 07:04:53 +09:00
Shunta Saito	66dd97ed3d	Fix channel first weights to channel last for right use of MLX's conv1d	2025-02-15 07:04:53 +09:00
Shunta Saito	81917d41d5	Allow a cache obj defined externally	2025-02-15 07:04:53 +09:00
Shunta Saito	00d13ebd40	Fix some part	2025-02-15 07:04:53 +09:00
Shunta Saito	fb5e225523	Apply formatter	2025-02-15 07:04:53 +09:00
Shunta Saito	07cf4336b3	Add plamo2.py	2025-02-15 07:04:53 +09:00