11 Commits

Author	SHA1	Message	Date
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