Commit Graph

11 Commits

Author SHA1 Message Date
Colin 085bd92fb9 Refine get_nodetree to support stride and tree tokens. 2025-08-18 11:17:35 +08:00
Colin 3e6ff2d580 Refine query_block_output. 2025-08-17 17:58:23 +08:00
Colin b2fe00c157 Add reserve_vocaba support, delete InitValDataset. 2025-08-14 14:51:43 +08:00
Colin 71ab0bb57d Add config of stride with_tree end, Rename dataset to meaning. 2025-08-10 13:17:04 +08:00
Colin 8882073978 Update test model. 2025-08-03 18:23:26 +08:00
Colin a7e0bd508c Update get_relation_distance 2025-06-26 21:42:38 +08:00
Colin 927c98e823 Add meaning dataset get_relation_table support and paint to qk image. 2025-06-25 20:13:48 +08:00
Colin 3eb711a97e Update wit. 2025-05-22 15:26:43 +08:00
Colin cda7f04e49 Fix model path. 2025-03-18 15:58:08 +08:00
Colin 90e94db2c1 Rename QwenModule to lightmodule. 2025-03-10 19:14:47 +08:00
Colin 002f132818 Add hook of attention for query qkv. 2025-03-03 15:38:00 +08:00