llama-arch.cpp 148 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751
  1. #include "llama-arch.h"
  2. #include "llama-impl.h"
  3. #include <map>
  4. static const std::map<llm_arch, const char *> LLM_ARCH_NAMES = {
  5. { LLM_ARCH_CLIP, "clip" }, // dummy, only used by llama-quantize
  6. { LLM_ARCH_LLAMA, "llama" },
  7. { LLM_ARCH_LLAMA4, "llama4" },
  8. { LLM_ARCH_DECI, "deci" },
  9. { LLM_ARCH_FALCON, "falcon" },
  10. { LLM_ARCH_GROK, "grok" },
  11. { LLM_ARCH_GPT2, "gpt2" },
  12. { LLM_ARCH_GPTJ, "gptj" },
  13. { LLM_ARCH_GPTNEOX, "gptneox" },
  14. { LLM_ARCH_MPT, "mpt" },
  15. { LLM_ARCH_BAICHUAN, "baichuan" },
  16. { LLM_ARCH_STARCODER, "starcoder" },
  17. { LLM_ARCH_REFACT, "refact" },
  18. { LLM_ARCH_BERT, "bert" },
  19. { LLM_ARCH_NOMIC_BERT, "nomic-bert" },
  20. { LLM_ARCH_NOMIC_BERT_MOE, "nomic-bert-moe" },
  21. { LLM_ARCH_NEO_BERT, "neo-bert" },
  22. { LLM_ARCH_JINA_BERT_V2, "jina-bert-v2" },
  23. { LLM_ARCH_JINA_BERT_V3, "jina-bert-v3" },
  24. { LLM_ARCH_BLOOM, "bloom" },
  25. { LLM_ARCH_STABLELM, "stablelm" },
  26. { LLM_ARCH_QWEN, "qwen" },
  27. { LLM_ARCH_QWEN2, "qwen2" },
  28. { LLM_ARCH_QWEN2MOE, "qwen2moe" },
  29. { LLM_ARCH_QWEN2VL, "qwen2vl" },
  30. { LLM_ARCH_QWEN3, "qwen3" },
  31. { LLM_ARCH_QWEN3MOE, "qwen3moe" },
  32. { LLM_ARCH_QWEN3VL, "qwen3vl" },
  33. { LLM_ARCH_QWEN3VLMOE, "qwen3vlmoe" },
  34. { LLM_ARCH_PHI2, "phi2" },
  35. { LLM_ARCH_PHI3, "phi3" },
  36. { LLM_ARCH_PHIMOE, "phimoe" },
  37. { LLM_ARCH_PLAMO, "plamo" },
  38. { LLM_ARCH_PLAMO2, "plamo2" },
  39. { LLM_ARCH_CODESHELL, "codeshell" },
  40. { LLM_ARCH_ORION, "orion" },
  41. { LLM_ARCH_INTERNLM2, "internlm2" },
  42. { LLM_ARCH_MINICPM, "minicpm" },
  43. { LLM_ARCH_MINICPM3, "minicpm3" },
  44. { LLM_ARCH_GEMMA, "gemma" },
  45. { LLM_ARCH_GEMMA2, "gemma2" },
  46. { LLM_ARCH_GEMMA3, "gemma3" },
  47. { LLM_ARCH_GEMMA3N, "gemma3n" },
  48. { LLM_ARCH_GEMMA_EMBEDDING, "gemma-embedding" },
  49. { LLM_ARCH_STARCODER2, "starcoder2" },
  50. { LLM_ARCH_MAMBA, "mamba" },
  51. { LLM_ARCH_MAMBA2, "mamba2" },
  52. { LLM_ARCH_JAMBA, "jamba" },
  53. { LLM_ARCH_FALCON_H1, "falcon-h1" },
  54. { LLM_ARCH_XVERSE, "xverse" },
  55. { LLM_ARCH_COMMAND_R, "command-r" },
  56. { LLM_ARCH_COHERE2, "cohere2" },
  57. { LLM_ARCH_DBRX, "dbrx" },
  58. { LLM_ARCH_OLMO, "olmo" },
  59. { LLM_ARCH_OLMO2, "olmo2" },
  60. { LLM_ARCH_OLMOE, "olmoe" },
  61. { LLM_ARCH_OPENELM, "openelm" },
  62. { LLM_ARCH_ARCTIC, "arctic" },
  63. { LLM_ARCH_DEEPSEEK, "deepseek" },
  64. { LLM_ARCH_DEEPSEEK2, "deepseek2" },
  65. { LLM_ARCH_CHATGLM, "chatglm" },
  66. { LLM_ARCH_GLM4, "glm4" },
  67. { LLM_ARCH_GLM4_MOE, "glm4moe" },
  68. { LLM_ARCH_BITNET, "bitnet" },
  69. { LLM_ARCH_T5, "t5" },
  70. { LLM_ARCH_T5ENCODER, "t5encoder" },
  71. { LLM_ARCH_JAIS, "jais" },
  72. { LLM_ARCH_NEMOTRON, "nemotron" },
  73. { LLM_ARCH_NEMOTRON_H, "nemotron_h" },
  74. { LLM_ARCH_EXAONE, "exaone" },
  75. { LLM_ARCH_EXAONE4, "exaone4" },
  76. { LLM_ARCH_RWKV6, "rwkv6" },
  77. { LLM_ARCH_RWKV6QWEN2, "rwkv6qwen2" },
  78. { LLM_ARCH_RWKV7, "rwkv7" },
  79. { LLM_ARCH_ARWKV7, "arwkv7" },
  80. { LLM_ARCH_GRANITE, "granite" },
  81. { LLM_ARCH_GRANITE_MOE, "granitemoe" },
  82. { LLM_ARCH_GRANITE_HYBRID, "granitehybrid" },
  83. { LLM_ARCH_CHAMELEON, "chameleon" },
  84. { LLM_ARCH_WAVTOKENIZER_DEC, "wavtokenizer-dec" },
  85. { LLM_ARCH_PLM, "plm" },
  86. { LLM_ARCH_BAILINGMOE, "bailingmoe" },
  87. { LLM_ARCH_BAILINGMOE2, "bailingmoe2" },
  88. { LLM_ARCH_DOTS1, "dots1" },
  89. { LLM_ARCH_ARCEE, "arcee" },
  90. { LLM_ARCH_AFMOE, "afmoe" },
  91. { LLM_ARCH_ERNIE4_5, "ernie4_5" },
  92. { LLM_ARCH_ERNIE4_5_MOE, "ernie4_5-moe" },
  93. { LLM_ARCH_HUNYUAN_MOE, "hunyuan-moe" },
  94. { LLM_ARCH_HUNYUAN_DENSE, "hunyuan-dense" },
  95. { LLM_ARCH_SMOLLM3, "smollm3" },
  96. { LLM_ARCH_OPENAI_MOE, "gpt-oss" },
  97. { LLM_ARCH_LFM2, "lfm2" },
  98. { LLM_ARCH_LFM2MOE, "lfm2moe" },
  99. { LLM_ARCH_DREAM, "dream" },
  100. { LLM_ARCH_SMALLTHINKER, "smallthinker" },
  101. { LLM_ARCH_LLADA, "llada" },
  102. { LLM_ARCH_LLADA_MOE, "llada-moe" },
  103. { LLM_ARCH_SEED_OSS, "seed_oss" },
  104. { LLM_ARCH_GROVEMOE, "grovemoe" },
  105. { LLM_ARCH_APERTUS, "apertus" },
  106. { LLM_ARCH_MINIMAX_M2, "minimax-m2" },
  107. { LLM_ARCH_COGVLM, "cogvlm" },
  108. { LLM_ARCH_RND1, "rnd1" },
  109. { LLM_ARCH_PANGU_EMBED, "pangu-embedded" },
  110. { LLM_ARCH_UNKNOWN, "(unknown)" },
  111. };
  112. static const std::map<llm_kv, const char *> LLM_KV_NAMES = {
  113. { LLM_KV_GENERAL_TYPE, "general.type" },
  114. { LLM_KV_GENERAL_ARCHITECTURE, "general.architecture" },
  115. { LLM_KV_GENERAL_QUANTIZATION_VERSION, "general.quantization_version" },
  116. { LLM_KV_GENERAL_ALIGNMENT, "general.alignment" },
  117. { LLM_KV_GENERAL_FILE_TYPE, "general.file_type" },
  118. { LLM_KV_GENERAL_NAME, "general.name" },
  119. { LLM_KV_GENERAL_AUTHOR, "general.author" },
  120. { LLM_KV_GENERAL_VERSION, "general.version" },
  121. { LLM_KV_GENERAL_URL, "general.url" },
  122. { LLM_KV_GENERAL_DESCRIPTION, "general.description" },
  123. { LLM_KV_GENERAL_LICENSE, "general.license" },
  124. { LLM_KV_GENERAL_SOURCE_URL, "general.source.url" },
  125. { LLM_KV_GENERAL_SOURCE_HF_REPO, "general.source.huggingface.repository" },
  126. { LLM_KV_VOCAB_SIZE, "%s.vocab_size" },
  127. { LLM_KV_CONTEXT_LENGTH, "%s.context_length" },
  128. { LLM_KV_EMBEDDING_LENGTH, "%s.embedding_length" },
  129. { LLM_KV_FEATURES_LENGTH, "%s.features_length" },
  130. { LLM_KV_BLOCK_COUNT, "%s.block_count" },
  131. { LLM_KV_LEADING_DENSE_BLOCK_COUNT, "%s.leading_dense_block_count" },
  132. { LLM_KV_FEED_FORWARD_LENGTH, "%s.feed_forward_length" },
  133. { LLM_KV_EXPERT_FEED_FORWARD_LENGTH, "%s.expert_feed_forward_length" },
  134. { LLM_KV_EXPERT_SHARED_FEED_FORWARD_LENGTH, "%s.expert_shared_feed_forward_length" },
  135. { LLM_KV_EXPERT_CHUNK_FEED_FORWARD_LENGTH, "%s.expert_chunk_feed_forward_length" },
  136. { LLM_KV_USE_PARALLEL_RESIDUAL, "%s.use_parallel_residual" },
  137. { LLM_KV_TENSOR_DATA_LAYOUT, "%s.tensor_data_layout" },
  138. { LLM_KV_EXPERT_COUNT, "%s.expert_count" },
  139. { LLM_KV_EXPERT_USED_COUNT, "%s.expert_used_count" },
  140. { LLM_KV_EXPERT_SHARED_COUNT, "%s.expert_shared_count" },
  141. { LLM_KV_EXPERT_GROUP_COUNT, "%s.expert_group_count" },
  142. { LLM_KV_EXPERT_GROUP_USED_COUNT, "%s.expert_group_used_count" },
  143. { LLM_KV_EXPERT_WEIGHTS_SCALE, "%s.expert_weights_scale" },
  144. { LLM_KV_EXPERT_WEIGHTS_NORM, "%s.expert_weights_norm" },
  145. { LLM_KV_EXPERT_GATING_FUNC, "%s.expert_gating_func" },
  146. { LLM_KV_EXPERT_GROUP_SCALE, "%s.expert_group_scale" },
  147. { LLM_KV_EXPERTS_PER_GROUP, "%s.experts_per_group" },
  148. { LLM_KV_MOE_EVERY_N_LAYERS, "%s.moe_every_n_layers" },
  149. { LLM_KV_NEXTN_PREDICT_LAYERS, "%s.nextn_predict_layers" },
  150. { LLM_KV_NUM_DEEPSTACK_LAYERS, "%s.n_deepstack_layers" },
  151. { LLM_KV_POOLING_TYPE, "%s.pooling_type" },
  152. { LLM_KV_LOGIT_SCALE, "%s.logit_scale" },
  153. { LLM_KV_DECODER_START_TOKEN_ID, "%s.decoder_start_token_id" },
  154. { LLM_KV_DECODER_BLOCK_COUNT, "%s.decoder_block_count" },
  155. { LLM_KV_ATTN_LOGIT_SOFTCAPPING, "%s.attn_logit_softcapping" },
  156. { LLM_KV_ROUTER_LOGIT_SOFTCAPPING, "%s.router_logit_softcapping" },
  157. { LLM_KV_FINAL_LOGIT_SOFTCAPPING, "%s.final_logit_softcapping" },
  158. { LLM_KV_SWIN_NORM, "%s.swin_norm" },
  159. { LLM_KV_RESCALE_EVERY_N_LAYERS, "%s.rescale_every_n_layers" },
  160. { LLM_KV_TIME_MIX_EXTRA_DIM, "%s.time_mix_extra_dim" },
  161. { LLM_KV_TIME_DECAY_EXTRA_DIM, "%s.time_decay_extra_dim" },
  162. { LLM_KV_RESIDUAL_SCALE, "%s.residual_scale" },
  163. { LLM_KV_EMBEDDING_SCALE, "%s.embedding_scale" },
  164. { LLM_KV_TOKEN_SHIFT_COUNT, "%s.token_shift_count" },
  165. { LLM_KV_INTERLEAVE_MOE_LAYER_STEP, "%s.interleave_moe_layer_step" },
  166. { LLM_KV_ATTENTION_HEAD_COUNT, "%s.attention.head_count" },
  167. { LLM_KV_ATTENTION_HEAD_COUNT_KV, "%s.attention.head_count_kv" },
  168. { LLM_KV_ATTENTION_MAX_ALIBI_BIAS, "%s.attention.max_alibi_bias" },
  169. { LLM_KV_ATTENTION_CLAMP_KQV, "%s.attention.clamp_kqv" },
  170. { LLM_KV_ATTENTION_KEY_LENGTH, "%s.attention.key_length" },
  171. { LLM_KV_ATTENTION_VALUE_LENGTH, "%s.attention.value_length" },
  172. { LLM_KV_ATTENTION_LAYERNORM_EPS, "%s.attention.layer_norm_epsilon" },
  173. { LLM_KV_ATTENTION_LAYERNORM_RMS_EPS, "%s.attention.layer_norm_rms_epsilon" },
  174. { LLM_KV_ATTENTION_GROUPNORM_EPS, "%s.attention.group_norm_epsilon" },
  175. { LLM_KV_ATTENTION_GROUPNORM_GROUPS, "%s.attention.group_norm_groups" },
  176. { LLM_KV_ATTENTION_CAUSAL, "%s.attention.causal" },
  177. { LLM_KV_ATTENTION_Q_LORA_RANK, "%s.attention.q_lora_rank" },
  178. { LLM_KV_ATTENTION_KV_LORA_RANK, "%s.attention.kv_lora_rank" },
  179. { LLM_KV_ATTENTION_DECAY_LORA_RANK, "%s.attention.decay_lora_rank" },
  180. { LLM_KV_ATTENTION_ICLR_LORA_RANK, "%s.attention.iclr_lora_rank" },
  181. { LLM_KV_ATTENTION_VALUE_RESIDUAL_MIX_LORA_RANK, "%s.attention.value_residual_mix_lora_rank" },
  182. { LLM_KV_ATTENTION_GATE_LORA_RANK, "%s.attention.gate_lora_rank" },
  183. { LLM_KV_ATTENTION_RELATIVE_BUCKETS_COUNT, "%s.attention.relative_buckets_count" },
  184. { LLM_KV_ATTENTION_SLIDING_WINDOW, "%s.attention.sliding_window" },
  185. { LLM_KV_ATTENTION_SCALE, "%s.attention.scale" },
  186. { LLM_KV_ATTENTION_OUTPUT_SCALE, "%s.attention.output_scale" },
  187. { LLM_KV_ATTENTION_TEMPERATURE_LENGTH, "%s.attention.temperature_length" },
  188. { LLM_KV_ATTENTION_KEY_LENGTH_MLA, "%s.attention.key_length_mla" },
  189. { LLM_KV_ATTENTION_VALUE_LENGTH_MLA, "%s.attention.value_length_mla" },
  190. { LLM_KV_ROPE_DIMENSION_COUNT, "%s.rope.dimension_count" },
  191. { LLM_KV_ROPE_DIMENSION_SECTIONS, "%s.rope.dimension_sections" },
  192. { LLM_KV_ROPE_FREQ_BASE, "%s.rope.freq_base" },
  193. { LLM_KV_ROPE_SCALE_LINEAR, "%s.rope.scale_linear" },
  194. { LLM_KV_ROPE_SCALING_TYPE, "%s.rope.scaling.type" },
  195. { LLM_KV_ROPE_SCALING_FACTOR, "%s.rope.scaling.factor" },
  196. { LLM_KV_ROPE_SCALING_ATTN_FACTOR, "%s.rope.scaling.attn_factor" },
  197. { LLM_KV_ROPE_SCALING_ORIG_CTX_LEN, "%s.rope.scaling.original_context_length" },
  198. { LLM_KV_ROPE_SCALING_FINETUNED, "%s.rope.scaling.finetuned" },
  199. { LLM_KV_ROPE_SCALING_YARN_LOG_MUL, "%s.rope.scaling.yarn_log_multiplier" },
  200. { LLM_KV_ROPE_SCALING_YARN_EXT_FACTOR, "%s.rope.scaling.yarn_ext_factor" },
  201. { LLM_KV_ROPE_SCALING_YARN_ATTN_FACTOR, "%s.rope.scaling.yarn_attn_factor" },
  202. { LLM_KV_ROPE_SCALING_YARN_BETA_FAST, "%s.rope.scaling.yarn_beta_fast" },
  203. { LLM_KV_ROPE_SCALING_YARN_BETA_SLOW, "%s.rope.scaling.yarn_beta_slow" },
  204. { LLM_KV_SPLIT_NO, "split.no" },
  205. { LLM_KV_SPLIT_COUNT, "split.count" },
  206. { LLM_KV_SPLIT_TENSORS_COUNT, "split.tensors.count" },
  207. { LLM_KV_SSM_CONV_KERNEL, "%s.ssm.conv_kernel" },
  208. { LLM_KV_SSM_INNER_SIZE, "%s.ssm.inner_size" },
  209. { LLM_KV_SSM_STATE_SIZE, "%s.ssm.state_size" },
  210. { LLM_KV_SSM_TIME_STEP_RANK, "%s.ssm.time_step_rank" },
  211. { LLM_KV_SSM_GROUP_COUNT, "%s.ssm.group_count" },
  212. { LLM_KV_SSM_DT_B_C_RMS, "%s.ssm.dt_b_c_rms" },
  213. { LLM_KV_WKV_HEAD_SIZE, "%s.wkv.head_size" },
  214. { LLM_KV_POSNET_EMBEDDING_LENGTH, "%s.posnet.embedding_length" },
  215. { LLM_KV_POSNET_BLOCK_COUNT, "%s.posnet.block_count" },
  216. { LLM_KV_CONVNEXT_EMBEDDING_LENGTH, "%s.convnext.embedding_length" },
  217. { LLM_KV_CONVNEXT_BLOCK_COUNT, "%s.convnext.block_count" },
  218. { LLM_KV_CLASSIFIER_OUTPUT_LABELS, "%s.classifier.output_labels" },
  219. { LLM_KV_SHORTCONV_L_CACHE, "%s.shortconv.l_cache" },
  220. // sentence-transformers dense modules feature dims
  221. { LLM_KV_DENSE_2_FEAT_IN, "%s.dense_2_feat_in" },
  222. { LLM_KV_DENSE_2_FEAT_OUT, "%s.dense_2_feat_out" },
  223. { LLM_KV_DENSE_3_FEAT_IN, "%s.dense_3_feat_in" },
  224. { LLM_KV_DENSE_3_FEAT_OUT, "%s.dense_3_feat_out" },
  225. { LLM_KV_TOKENIZER_MODEL, "tokenizer.ggml.model" },
  226. { LLM_KV_TOKENIZER_PRE, "tokenizer.ggml.pre" },
  227. { LLM_KV_TOKENIZER_LIST, "tokenizer.ggml.tokens" },
  228. { LLM_KV_TOKENIZER_TOKEN_TYPE, "tokenizer.ggml.token_type" },
  229. { LLM_KV_TOKENIZER_TOKEN_TYPE_COUNT, "tokenizer.ggml.token_type_count" },
  230. { LLM_KV_TOKENIZER_SCORES, "tokenizer.ggml.scores" },
  231. { LLM_KV_TOKENIZER_MERGES, "tokenizer.ggml.merges" },
  232. { LLM_KV_TOKENIZER_BOS_ID, "tokenizer.ggml.bos_token_id" },
  233. { LLM_KV_TOKENIZER_EOS_ID, "tokenizer.ggml.eos_token_id" },
  234. { LLM_KV_TOKENIZER_EOT_ID, "tokenizer.ggml.eot_token_id" },
  235. { LLM_KV_TOKENIZER_EOM_ID, "tokenizer.ggml.eom_token_id" },
  236. { LLM_KV_TOKENIZER_UNK_ID, "tokenizer.ggml.unknown_token_id" },
  237. { LLM_KV_TOKENIZER_SEP_ID, "tokenizer.ggml.seperator_token_id" },
  238. { LLM_KV_TOKENIZER_PAD_ID, "tokenizer.ggml.padding_token_id" },
  239. { LLM_KV_TOKENIZER_CLS_ID, "tokenizer.ggml.cls_token_id" },
  240. { LLM_KV_TOKENIZER_MASK_ID, "tokenizer.ggml.mask_token_id" },
  241. { LLM_KV_TOKENIZER_ADD_BOS, "tokenizer.ggml.add_bos_token" },
  242. { LLM_KV_TOKENIZER_ADD_EOS, "tokenizer.ggml.add_eos_token" },
  243. { LLM_KV_TOKENIZER_ADD_SEP, "tokenizer.ggml.add_sep_token" },
  244. { LLM_KV_TOKENIZER_ADD_PREFIX, "tokenizer.ggml.add_space_prefix" },
  245. { LLM_KV_TOKENIZER_REMOVE_EXTRA_WS, "tokenizer.ggml.remove_extra_whitespaces" },
  246. { LLM_KV_TOKENIZER_PRECOMPILED_CHARSMAP, "tokenizer.ggml.precompiled_charsmap" },
  247. { LLM_KV_TOKENIZER_HF_JSON, "tokenizer.huggingface.json" },
  248. { LLM_KV_TOKENIZER_RWKV, "tokenizer.rwkv.world" },
  249. { LLM_KV_TOKENIZER_CHAT_TEMPLATE, "tokenizer.chat_template" },
  250. { LLM_KV_TOKENIZER_FIM_PRE_ID, "tokenizer.ggml.fim_pre_token_id" },
  251. { LLM_KV_TOKENIZER_FIM_SUF_ID, "tokenizer.ggml.fim_suf_token_id" },
  252. { LLM_KV_TOKENIZER_FIM_MID_ID, "tokenizer.ggml.fim_mid_token_id" },
  253. { LLM_KV_TOKENIZER_FIM_PAD_ID, "tokenizer.ggml.fim_pad_token_id" },
  254. { LLM_KV_TOKENIZER_FIM_REP_ID, "tokenizer.ggml.fim_rep_token_id" },
  255. { LLM_KV_TOKENIZER_FIM_SEP_ID, "tokenizer.ggml.fim_sep_token_id" },
  256. { LLM_KV_ADAPTER_TYPE, "adapter.type" },
  257. { LLM_KV_ADAPTER_LORA_ALPHA, "adapter.lora.alpha" },
  258. { LLM_KV_ADAPTER_LORA_TASK_NAME, "adapter.lora.task_name" },
  259. { LLM_KV_ADAPTER_LORA_PROMPT_PREFIX, "adapter.lora.prompt_prefix" },
  260. { LLM_KV_ADAPTER_ALORA_INVOCATION_TOKENS, "adapter.alora.invocation_tokens" },
  261. { LLM_KV_XIELU_ALPHA_N, "xielu.alpha_n" },
  262. { LLM_KV_XIELU_ALPHA_P, "xielu.alpha_p" },
  263. { LLM_KV_XIELU_BETA, "xielu.beta" },
  264. { LLM_KV_XIELU_EPS, "xielu.eps" },
  265. // deprecated
  266. { LLM_KV_TOKENIZER_PREFIX_ID, "tokenizer.ggml.prefix_token_id" },
  267. { LLM_KV_TOKENIZER_SUFFIX_ID, "tokenizer.ggml.suffix_token_id" },
  268. { LLM_KV_TOKENIZER_MIDDLE_ID, "tokenizer.ggml.middle_token_id" },
  269. };
  270. static const std::map<llm_arch, std::map<llm_tensor, const char *>> LLM_TENSOR_NAMES = {
  271. {
  272. LLM_ARCH_CLIP,
  273. {},
  274. },
  275. {
  276. LLM_ARCH_LLAMA,
  277. {
  278. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  279. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  280. { LLM_TENSOR_OUTPUT, "output" },
  281. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  282. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  283. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  284. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  285. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  286. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  287. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  288. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  289. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  290. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  291. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  292. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  293. { LLM_TENSOR_FFN_GATE_EXP, "blk.%d.ffn_gate.%d" },
  294. { LLM_TENSOR_FFN_DOWN_EXP, "blk.%d.ffn_down.%d" },
  295. { LLM_TENSOR_FFN_UP_EXP, "blk.%d.ffn_up.%d" },
  296. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  297. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  298. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  299. },
  300. },
  301. {
  302. LLM_ARCH_ARCEE,
  303. {
  304. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  305. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  306. { LLM_TENSOR_OUTPUT, "output" },
  307. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  308. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  309. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  310. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  311. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  312. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  313. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  314. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  315. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  316. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  317. },
  318. },
  319. {
  320. LLM_ARCH_AFMOE,
  321. {
  322. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  323. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  324. { LLM_TENSOR_OUTPUT, "output" },
  325. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  326. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  327. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  328. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  329. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  330. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  331. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  332. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  333. { LLM_TENSOR_ATTN_GATE, "blk.%d.attn_gate" },
  334. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  335. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  336. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  337. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  338. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  339. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  340. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  341. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  342. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  343. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  344. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  345. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  346. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  347. },
  348. },
  349. {
  350. LLM_ARCH_LLAMA4,
  351. {
  352. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  353. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  354. { LLM_TENSOR_OUTPUT, "output" },
  355. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  356. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  357. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  358. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  359. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  360. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  361. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  362. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  363. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  364. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  365. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  366. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  367. { LLM_TENSOR_FFN_GATE_EXP, "blk.%d.ffn_gate.%d" },
  368. { LLM_TENSOR_FFN_DOWN_EXP, "blk.%d.ffn_down.%d" },
  369. { LLM_TENSOR_FFN_UP_EXP, "blk.%d.ffn_up.%d" },
  370. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  371. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  372. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  373. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  374. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  375. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  376. },
  377. },
  378. {
  379. LLM_ARCH_DECI,
  380. {
  381. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  382. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  383. { LLM_TENSOR_OUTPUT, "output" },
  384. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  385. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  386. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  387. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  388. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  389. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  390. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  391. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  392. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  393. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  394. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  395. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  396. { LLM_TENSOR_FFN_GATE_EXP, "blk.%d.ffn_gate.%d" },
  397. { LLM_TENSOR_FFN_DOWN_EXP, "blk.%d.ffn_down.%d" },
  398. { LLM_TENSOR_FFN_UP_EXP, "blk.%d.ffn_up.%d" },
  399. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  400. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  401. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  402. },
  403. },
  404. {
  405. LLM_ARCH_BAICHUAN,
  406. {
  407. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  408. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  409. { LLM_TENSOR_OUTPUT, "output" },
  410. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  411. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  412. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  413. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  414. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  415. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  416. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  417. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  418. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  419. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  420. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  421. },
  422. },
  423. {
  424. LLM_ARCH_FALCON,
  425. {
  426. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  427. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  428. { LLM_TENSOR_OUTPUT, "output" },
  429. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  430. { LLM_TENSOR_ATTN_NORM_2, "blk.%d.attn_norm_2" },
  431. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  432. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  433. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  434. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  435. },
  436. },
  437. {
  438. LLM_ARCH_GROK,
  439. {
  440. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  441. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  442. { LLM_TENSOR_OUTPUT, "output" },
  443. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  444. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  445. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  446. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  447. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  448. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  449. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  450. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  451. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  452. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  453. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  454. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  455. { LLM_TENSOR_FFN_GATE_EXP, "blk.%d.ffn_gate.%d" },
  456. { LLM_TENSOR_FFN_DOWN_EXP, "blk.%d.ffn_down.%d" },
  457. { LLM_TENSOR_FFN_UP_EXP, "blk.%d.ffn_up.%d" },
  458. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  459. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  460. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  461. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  462. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  463. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  464. },
  465. },
  466. {
  467. LLM_ARCH_GPT2,
  468. {
  469. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  470. { LLM_TENSOR_POS_EMBD, "position_embd" },
  471. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  472. { LLM_TENSOR_OUTPUT, "output" },
  473. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  474. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  475. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  476. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  477. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  478. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  479. },
  480. },
  481. {
  482. LLM_ARCH_GPTJ,
  483. {
  484. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  485. },
  486. },
  487. {
  488. LLM_ARCH_GPTNEOX,
  489. {
  490. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  491. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  492. { LLM_TENSOR_OUTPUT, "output" },
  493. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  494. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  495. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  496. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  497. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  498. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  499. },
  500. },
  501. {
  502. LLM_ARCH_MPT,
  503. {
  504. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  505. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  506. { LLM_TENSOR_OUTPUT, "output"},
  507. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  508. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  509. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  510. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  511. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  512. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  513. { LLM_TENSOR_FFN_ACT, "blk.%d.ffn.act" },
  514. { LLM_TENSOR_POS_EMBD, "position_embd" },
  515. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm"},
  516. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm"},
  517. },
  518. },
  519. {
  520. LLM_ARCH_STARCODER,
  521. {
  522. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  523. { LLM_TENSOR_POS_EMBD, "position_embd" },
  524. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  525. { LLM_TENSOR_OUTPUT, "output" },
  526. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  527. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  528. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  529. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  530. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  531. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  532. },
  533. },
  534. {
  535. LLM_ARCH_REFACT,
  536. {
  537. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  538. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  539. { LLM_TENSOR_OUTPUT, "output" },
  540. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  541. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  542. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  543. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  544. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  545. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  546. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  547. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  548. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  549. },
  550. },
  551. {
  552. LLM_ARCH_BERT,
  553. {
  554. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  555. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  556. { LLM_TENSOR_TOKEN_TYPES, "token_types" },
  557. { LLM_TENSOR_POS_EMBD, "position_embd" },
  558. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  559. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  560. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  561. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  562. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  563. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  564. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  565. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  566. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  567. { LLM_TENSOR_CLS, "cls" },
  568. { LLM_TENSOR_CLS_OUT, "cls.output" },
  569. },
  570. },
  571. {
  572. LLM_ARCH_NOMIC_BERT,
  573. {
  574. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  575. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  576. { LLM_TENSOR_TOKEN_TYPES, "token_types" },
  577. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  578. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  579. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  580. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  581. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  582. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  583. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  584. },
  585. },
  586. {
  587. LLM_ARCH_NOMIC_BERT_MOE,
  588. {
  589. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  590. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  591. { LLM_TENSOR_TOKEN_TYPES, "token_types" },
  592. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  593. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  594. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  595. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  596. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  597. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  598. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  599. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  600. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  601. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  602. },
  603. },
  604. {
  605. LLM_ARCH_NEO_BERT,
  606. {
  607. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  608. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  609. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  610. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  611. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  612. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  613. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  614. { LLM_TENSOR_ENC_OUTPUT_NORM, "enc.output_norm" },
  615. { LLM_TENSOR_CLS, "cls" },
  616. { LLM_TENSOR_CLS_OUT, "cls.output" },
  617. },
  618. },
  619. {
  620. LLM_ARCH_JINA_BERT_V2,
  621. {
  622. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  623. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  624. { LLM_TENSOR_TOKEN_TYPES, "token_types" },
  625. { LLM_TENSOR_ATTN_NORM_2, "blk.%d.attn_norm_2" },
  626. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  627. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  628. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  629. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  630. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  631. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  632. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  633. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  634. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  635. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  636. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  637. { LLM_TENSOR_CLS, "cls" },
  638. },
  639. },
  640. {
  641. LLM_ARCH_JINA_BERT_V3,
  642. {
  643. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  644. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  645. { LLM_TENSOR_TOKEN_TYPES, "token_types" },
  646. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  647. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  648. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  649. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  650. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  651. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  652. },
  653. },
  654. {
  655. LLM_ARCH_BLOOM,
  656. {
  657. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  658. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  659. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  660. { LLM_TENSOR_OUTPUT, "output" },
  661. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  662. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  663. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  664. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  665. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  666. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  667. },
  668. },
  669. {
  670. LLM_ARCH_STABLELM,
  671. {
  672. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  673. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  674. { LLM_TENSOR_OUTPUT, "output" },
  675. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  676. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  677. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  678. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  679. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  680. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  681. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  682. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  683. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  684. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  685. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  686. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  687. },
  688. },
  689. {
  690. LLM_ARCH_QWEN,
  691. {
  692. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  693. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  694. { LLM_TENSOR_OUTPUT, "output" },
  695. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  696. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  697. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  698. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  699. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  700. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  701. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  702. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  703. },
  704. },
  705. {
  706. LLM_ARCH_QWEN2,
  707. {
  708. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  709. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  710. { LLM_TENSOR_OUTPUT, "output" },
  711. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  712. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  713. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  714. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  715. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  716. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  717. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  718. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  719. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  720. },
  721. },
  722. {
  723. LLM_ARCH_QWEN2VL,
  724. {
  725. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  726. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  727. { LLM_TENSOR_OUTPUT, "output" },
  728. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  729. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  730. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  731. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  732. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  733. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  734. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  735. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  736. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  737. },
  738. },
  739. {
  740. LLM_ARCH_QWEN2MOE,
  741. {
  742. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  743. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  744. { LLM_TENSOR_OUTPUT, "output" },
  745. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  746. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  747. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  748. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  749. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  750. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  751. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  752. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  753. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  754. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  755. { LLM_TENSOR_FFN_GATE_INP_SHEXP, "blk.%d.ffn_gate_inp_shexp" },
  756. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  757. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  758. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  759. },
  760. },
  761. {
  762. LLM_ARCH_QWEN3,
  763. {
  764. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  765. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  766. { LLM_TENSOR_OUTPUT, "output" },
  767. { LLM_TENSOR_CLS_OUT, "cls.output" },
  768. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  769. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  770. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  771. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  772. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  773. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  774. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  775. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  776. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  777. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  778. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  779. },
  780. },
  781. {
  782. LLM_ARCH_QWEN3MOE,
  783. {
  784. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  785. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  786. { LLM_TENSOR_OUTPUT, "output" },
  787. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  788. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  789. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  790. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  791. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  792. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  793. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  794. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  795. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  796. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  797. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  798. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  799. },
  800. },
  801. {
  802. LLM_ARCH_QWEN3VL,
  803. {
  804. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  805. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  806. { LLM_TENSOR_OUTPUT, "output" },
  807. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  808. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  809. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  810. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  811. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  812. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  813. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  814. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  815. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  816. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  817. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  818. },
  819. },
  820. {
  821. LLM_ARCH_QWEN3VLMOE,
  822. {
  823. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  824. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  825. { LLM_TENSOR_OUTPUT, "output" },
  826. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  827. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  828. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  829. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  830. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  831. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  832. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  833. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  834. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  835. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  836. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  837. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  838. },
  839. },
  840. {
  841. LLM_ARCH_PHI2,
  842. {
  843. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  844. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  845. { LLM_TENSOR_OUTPUT, "output" },
  846. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  847. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  848. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  849. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  850. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  851. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  852. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  853. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  854. },
  855. },
  856. {
  857. LLM_ARCH_PHI3,
  858. {
  859. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  860. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  861. { LLM_TENSOR_OUTPUT, "output" },
  862. { LLM_TENSOR_ROPE_FACTORS_LONG, "rope_factors_long" },
  863. { LLM_TENSOR_ROPE_FACTORS_SHORT, "rope_factors_short" },
  864. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  865. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  866. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  867. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  868. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  869. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  870. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  871. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  872. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  873. },
  874. },
  875. {
  876. LLM_ARCH_PHIMOE,
  877. {
  878. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  879. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  880. { LLM_TENSOR_OUTPUT, "output" },
  881. { LLM_TENSOR_ROPE_FACTORS_LONG, "rope_factors_long" },
  882. { LLM_TENSOR_ROPE_FACTORS_SHORT, "rope_factors_short" },
  883. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  884. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  885. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  886. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  887. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  888. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  889. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  890. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  891. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  892. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  893. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  894. },
  895. },
  896. {
  897. LLM_ARCH_PLAMO,
  898. {
  899. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  900. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  901. { LLM_TENSOR_OUTPUT, "output" },
  902. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  903. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  904. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  905. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  906. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  907. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  908. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  909. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  910. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  911. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  912. },
  913. },
  914. {
  915. LLM_ARCH_PLAMO2,
  916. {
  917. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  918. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  919. { LLM_TENSOR_OUTPUT, "output" },
  920. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  921. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  922. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  923. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  924. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  925. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  926. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  927. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  928. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  929. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  930. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  931. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  932. { LLM_TENSOR_SSM_X, "blk.%d.ssm_x" },
  933. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  934. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  935. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  936. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  937. { LLM_TENSOR_SSM_DT_NORM, "blk.%d.ssm_dt_norm" },
  938. { LLM_TENSOR_SSM_B_NORM, "blk.%d.ssm_b_norm" },
  939. { LLM_TENSOR_SSM_C_NORM, "blk.%d.ssm_c_norm" },
  940. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  941. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  942. },
  943. },
  944. {
  945. LLM_ARCH_CODESHELL,
  946. {
  947. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  948. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  949. { LLM_TENSOR_OUTPUT, "output" },
  950. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  951. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  952. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  953. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  954. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  955. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  956. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  957. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  958. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  959. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  960. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  961. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  962. },
  963. },
  964. {
  965. LLM_ARCH_ORION,
  966. {
  967. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  968. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  969. { LLM_TENSOR_OUTPUT, "output" },
  970. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  971. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  972. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  973. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  974. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  975. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  976. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  977. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  978. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  979. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  980. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  981. },
  982. },
  983. {
  984. LLM_ARCH_INTERNLM2,
  985. {
  986. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  987. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  988. { LLM_TENSOR_OUTPUT, "output" },
  989. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  990. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  991. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  992. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  993. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  994. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  995. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  996. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  997. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  998. },
  999. },
  1000. {
  1001. LLM_ARCH_MINICPM,
  1002. {
  1003. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1004. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1005. { LLM_TENSOR_OUTPUT, "output" },
  1006. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1007. { LLM_TENSOR_ROPE_FACTORS_LONG, "rope_factors_long" },
  1008. { LLM_TENSOR_ROPE_FACTORS_SHORT, "rope_factors_short" },
  1009. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1010. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1011. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1012. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1013. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1014. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  1015. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1016. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1017. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1018. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1019. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1020. { LLM_TENSOR_FFN_GATE_EXP, "blk.%d.ffn_gate.%d" },
  1021. { LLM_TENSOR_FFN_DOWN_EXP, "blk.%d.ffn_down.%d" },
  1022. { LLM_TENSOR_FFN_UP_EXP, "blk.%d.ffn_up.%d" },
  1023. },
  1024. },
  1025. {
  1026. LLM_ARCH_MINICPM3,
  1027. {
  1028. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1029. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1030. { LLM_TENSOR_OUTPUT, "output" },
  1031. { LLM_TENSOR_ROPE_FACTORS_LONG, "rope_factors_long" },
  1032. { LLM_TENSOR_ROPE_FACTORS_SHORT, "rope_factors_short" },
  1033. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1034. { LLM_TENSOR_ATTN_Q_A_NORM, "blk.%d.attn_q_a_norm" },
  1035. { LLM_TENSOR_ATTN_KV_A_NORM, "blk.%d.attn_kv_a_norm" },
  1036. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1037. { LLM_TENSOR_ATTN_Q_A, "blk.%d.attn_q_a" },
  1038. { LLM_TENSOR_ATTN_Q_B, "blk.%d.attn_q_b" },
  1039. { LLM_TENSOR_ATTN_KV_A_MQA, "blk.%d.attn_kv_a_mqa" },
  1040. { LLM_TENSOR_ATTN_KV_B, "blk.%d.attn_kv_b" },
  1041. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1042. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1043. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1044. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1045. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1046. },
  1047. },
  1048. {
  1049. LLM_ARCH_GEMMA,
  1050. {
  1051. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1052. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1053. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1054. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1055. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1056. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1057. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1058. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1059. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1060. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1061. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1062. },
  1063. },
  1064. {
  1065. LLM_ARCH_GEMMA2,
  1066. {
  1067. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1068. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1069. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1070. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1071. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1072. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1073. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1074. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1075. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1076. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1077. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1078. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1079. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1080. },
  1081. },
  1082. {
  1083. LLM_ARCH_GEMMA3,
  1084. {
  1085. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1086. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1087. { LLM_TENSOR_OUTPUT, "output" },
  1088. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1089. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1090. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1091. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1092. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1093. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1094. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1095. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1096. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1097. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1098. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1099. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1100. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1101. },
  1102. },
  1103. {
  1104. LLM_ARCH_GEMMA3N,
  1105. {
  1106. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1107. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1108. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1109. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1110. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1111. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1112. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1113. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1114. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1115. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1116. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1117. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1118. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1119. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1120. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1121. { LLM_TENSOR_PER_LAYER_TOKEN_EMBD, "per_layer_token_embd" },
  1122. { LLM_TENSOR_PER_LAYER_MODEL_PROJ, "per_layer_model_proj" },
  1123. { LLM_TENSOR_PER_LAYER_PROJ_NORM, "per_layer_proj_norm" },
  1124. { LLM_TENSOR_ALTUP_UNEMBD_PROJ, "altup_unembd_proj" },
  1125. { LLM_TENSOR_ALTUP_PROJ, "altup_proj" },
  1126. { LLM_TENSOR_PER_LAYER_INP_GATE, "blk.%d.inp_gate" },
  1127. { LLM_TENSOR_PER_LAYER_PROJ, "blk.%d.proj" },
  1128. { LLM_TENSOR_PER_LAYER_POST_NORM, "blk.%d.post_norm" },
  1129. { LLM_TENSOR_ALTUP_CORRECT_COEF, "blk.%d.altup_correct_coef" },
  1130. { LLM_TENSOR_ALTUP_CORRECT_SCALE, "blk.%d.altup_correct_scale" },
  1131. { LLM_TENSOR_ALTUP_PREDICT_COEF, "blk.%d.altup_predict_coef" },
  1132. { LLM_TENSOR_ALTUP_ROUTER, "blk.%d.altup_router" },
  1133. { LLM_TENSOR_ALTUP_ROUTER_NORM, "blk.%d.altup_router_norm" },
  1134. { LLM_TENSOR_LAUREL_L, "blk.%d.laurel_l" },
  1135. { LLM_TENSOR_LAUREL_R, "blk.%d.laurel_r" },
  1136. { LLM_TENSOR_LAUREL_POST_NORM, "blk.%d.laurel_post_norm" },
  1137. },
  1138. },
  1139. {
  1140. LLM_ARCH_GEMMA_EMBEDDING,
  1141. {
  1142. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1143. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1144. { LLM_TENSOR_OUTPUT, "output" },
  1145. { LLM_TENSOR_DENSE_2_OUT, "dense_2" },
  1146. { LLM_TENSOR_DENSE_3_OUT, "dense_3" },
  1147. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1148. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1149. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1150. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1151. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1152. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1153. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1154. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1155. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1156. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1157. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1158. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1159. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1160. },
  1161. },
  1162. {
  1163. LLM_ARCH_STARCODER2,
  1164. {
  1165. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1166. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1167. { LLM_TENSOR_OUTPUT, "output" },
  1168. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1169. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1170. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1171. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1172. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1173. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1174. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  1175. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1176. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1177. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1178. },
  1179. },
  1180. {
  1181. LLM_ARCH_MAMBA,
  1182. {
  1183. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1184. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1185. { LLM_TENSOR_OUTPUT, "output" },
  1186. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1187. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  1188. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  1189. { LLM_TENSOR_SSM_X, "blk.%d.ssm_x" },
  1190. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  1191. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  1192. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  1193. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  1194. },
  1195. },
  1196. {
  1197. LLM_ARCH_MAMBA2,
  1198. {
  1199. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1200. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1201. { LLM_TENSOR_OUTPUT, "output" },
  1202. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1203. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  1204. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  1205. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  1206. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  1207. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  1208. { LLM_TENSOR_SSM_NORM, "blk.%d.ssm_norm" },
  1209. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  1210. },
  1211. },
  1212. {
  1213. LLM_ARCH_JAMBA,
  1214. {
  1215. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1216. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1217. { LLM_TENSOR_OUTPUT, "output" },
  1218. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1219. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  1220. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  1221. { LLM_TENSOR_SSM_X, "blk.%d.ssm_x" },
  1222. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  1223. { LLM_TENSOR_SSM_DT_NORM, "blk.%d.ssm_dt_norm" },
  1224. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  1225. { LLM_TENSOR_SSM_B_NORM, "blk.%d.ssm_b_norm" },
  1226. { LLM_TENSOR_SSM_C_NORM, "blk.%d.ssm_c_norm" },
  1227. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  1228. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  1229. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1230. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1231. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1232. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1233. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1234. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1235. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1236. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1237. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1238. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1239. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1240. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1241. },
  1242. },
  1243. {
  1244. LLM_ARCH_FALCON_H1,
  1245. {
  1246. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1247. { LLM_TENSOR_OUTPUT, "output" },
  1248. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1249. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1250. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1251. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1252. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1253. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1254. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  1255. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  1256. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  1257. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  1258. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  1259. { LLM_TENSOR_SSM_NORM, "blk.%d.ssm_norm" },
  1260. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  1261. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1262. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1263. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1264. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1265. },
  1266. },
  1267. {
  1268. LLM_ARCH_XVERSE,
  1269. {
  1270. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1271. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1272. { LLM_TENSOR_OUTPUT, "output" },
  1273. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1274. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1275. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1276. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1277. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1278. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1279. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  1280. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1281. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1282. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1283. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1284. },
  1285. },
  1286. {
  1287. LLM_ARCH_COMMAND_R,
  1288. {
  1289. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1290. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1291. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1292. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1293. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1294. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1295. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1296. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1297. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1298. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1299. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1300. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1301. },
  1302. },
  1303. {
  1304. LLM_ARCH_COHERE2,
  1305. {
  1306. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1307. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1308. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1309. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1310. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1311. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1312. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1313. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1314. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1315. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1316. },
  1317. },
  1318. {
  1319. LLM_ARCH_DBRX,
  1320. {
  1321. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1322. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1323. { LLM_TENSOR_OUTPUT, "output" },
  1324. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  1325. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1326. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1327. { LLM_TENSOR_ATTN_OUT_NORM, "blk.%d.attn_output_norm" },
  1328. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1329. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1330. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1331. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1332. },
  1333. },
  1334. {
  1335. LLM_ARCH_OLMO,
  1336. {
  1337. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1338. { LLM_TENSOR_OUTPUT, "output" },
  1339. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1340. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1341. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1342. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1343. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1344. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1345. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1346. },
  1347. },
  1348. {
  1349. LLM_ARCH_OLMO2,
  1350. {
  1351. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1352. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1353. { LLM_TENSOR_OUTPUT, "output" },
  1354. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1355. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1356. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1357. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1358. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1359. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1360. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1361. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1362. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1363. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1364. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1365. },
  1366. },
  1367. {
  1368. LLM_ARCH_OLMOE,
  1369. {
  1370. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1371. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1372. { LLM_TENSOR_OUTPUT, "output" },
  1373. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1374. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1375. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1376. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1377. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1378. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1379. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1380. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1381. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1382. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1383. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1384. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1385. },
  1386. },
  1387. {
  1388. LLM_ARCH_OPENELM,
  1389. {
  1390. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1391. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1392. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1393. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  1394. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1395. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1396. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1397. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1398. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1399. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1400. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1401. },
  1402. },
  1403. {
  1404. LLM_ARCH_ARCTIC,
  1405. {
  1406. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1407. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1408. { LLM_TENSOR_OUTPUT, "output" },
  1409. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1410. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1411. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1412. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1413. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1414. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1415. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1416. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1417. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1418. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1419. { LLM_TENSOR_FFN_NORM_EXPS, "blk.%d.ffn_norm_exps" },
  1420. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1421. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1422. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1423. },
  1424. },
  1425. {
  1426. LLM_ARCH_DEEPSEEK,
  1427. {
  1428. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1429. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1430. { LLM_TENSOR_OUTPUT, "output" },
  1431. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1432. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1433. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1434. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1435. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1436. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1437. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  1438. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1439. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1440. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1441. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1442. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1443. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1444. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1445. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1446. { LLM_TENSOR_FFN_GATE_INP_SHEXP, "blk.%d.ffn_gate_inp_shexp" },
  1447. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  1448. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  1449. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  1450. },
  1451. },
  1452. {
  1453. LLM_ARCH_DEEPSEEK2,
  1454. {
  1455. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1456. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1457. { LLM_TENSOR_OUTPUT, "output" },
  1458. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1459. { LLM_TENSOR_ATTN_Q_A_NORM, "blk.%d.attn_q_a_norm" },
  1460. { LLM_TENSOR_ATTN_KV_A_NORM, "blk.%d.attn_kv_a_norm" },
  1461. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1462. { LLM_TENSOR_ATTN_Q_A, "blk.%d.attn_q_a" },
  1463. { LLM_TENSOR_ATTN_Q_B, "blk.%d.attn_q_b" },
  1464. { LLM_TENSOR_ATTN_KV_A_MQA, "blk.%d.attn_kv_a_mqa" },
  1465. { LLM_TENSOR_ATTN_KV_B, "blk.%d.attn_kv_b" },
  1466. { LLM_TENSOR_ATTN_K_B, "blk.%d.attn_k_b" },
  1467. { LLM_TENSOR_ATTN_V_B, "blk.%d.attn_v_b" },
  1468. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1469. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1470. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1471. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1472. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1473. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1474. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1475. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1476. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1477. { LLM_TENSOR_FFN_GATE_INP_SHEXP, "blk.%d.ffn_gate_inp_shexp" },
  1478. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  1479. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  1480. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  1481. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  1482. },
  1483. },
  1484. {
  1485. LLM_ARCH_PLM,
  1486. {
  1487. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1488. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1489. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1490. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1491. { LLM_TENSOR_ATTN_KV_A_MQA, "blk.%d.attn_kv_a_mqa" },
  1492. { LLM_TENSOR_ATTN_KV_A_NORM, "blk.%d.attn_kv_a_norm" },
  1493. { LLM_TENSOR_ATTN_KV_B, "blk.%d.attn_kv_b" },
  1494. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1495. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1496. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1497. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1498. },
  1499. },
  1500. {
  1501. LLM_ARCH_CHATGLM,
  1502. {
  1503. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1504. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1505. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1506. { LLM_TENSOR_OUTPUT, "output" },
  1507. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1508. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  1509. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1510. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1511. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1512. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1513. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1514. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1515. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1516. },
  1517. },
  1518. {
  1519. LLM_ARCH_GLM4,
  1520. {
  1521. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1522. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1523. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1524. { LLM_TENSOR_OUTPUT, "output" },
  1525. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1526. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1527. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1528. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1529. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1530. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1531. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1532. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1533. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1534. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1535. },
  1536. },
  1537. {
  1538. LLM_ARCH_GLM4_MOE,
  1539. {
  1540. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1541. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1542. { LLM_TENSOR_OUTPUT, "output" },
  1543. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1544. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1545. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1546. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1547. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1548. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1549. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1550. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1551. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1552. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1553. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1554. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1555. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1556. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1557. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1558. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  1559. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  1560. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  1561. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  1562. // NextN/MTP tensors - preserved but unused (in final layer, dynamic layer number)
  1563. { LLM_TENSOR_NEXTN_EH_PROJ, "blk.%d.nextn.eh_proj" },
  1564. { LLM_TENSOR_NEXTN_EMBED_TOKENS, "blk.%d.nextn.embed_tokens" },
  1565. { LLM_TENSOR_NEXTN_ENORM, "blk.%d.nextn.enorm" },
  1566. { LLM_TENSOR_NEXTN_HNORM, "blk.%d.nextn.hnorm" },
  1567. { LLM_TENSOR_NEXTN_SHARED_HEAD_HEAD, "blk.%d.nextn.shared_head_head" },
  1568. { LLM_TENSOR_NEXTN_SHARED_HEAD_NORM, "blk.%d.nextn.shared_head_norm" },
  1569. },
  1570. },
  1571. {
  1572. LLM_ARCH_BITNET,
  1573. {
  1574. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1575. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1576. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1577. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1578. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1579. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1580. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1581. { LLM_TENSOR_ATTN_SUB_NORM, "blk.%d.attn_sub_norm" },
  1582. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1583. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1584. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1585. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1586. { LLM_TENSOR_FFN_SUB_NORM, "blk.%d.ffn_sub_norm" },
  1587. },
  1588. },
  1589. {
  1590. LLM_ARCH_T5,
  1591. {
  1592. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1593. { LLM_TENSOR_OUTPUT, "output" },
  1594. { LLM_TENSOR_DEC_OUTPUT_NORM, "dec.output_norm" },
  1595. { LLM_TENSOR_DEC_ATTN_NORM, "dec.blk.%d.attn_norm" },
  1596. { LLM_TENSOR_DEC_ATTN_Q, "dec.blk.%d.attn_q" },
  1597. { LLM_TENSOR_DEC_ATTN_K, "dec.blk.%d.attn_k" },
  1598. { LLM_TENSOR_DEC_ATTN_V, "dec.blk.%d.attn_v" },
  1599. { LLM_TENSOR_DEC_ATTN_OUT, "dec.blk.%d.attn_o" },
  1600. { LLM_TENSOR_DEC_ATTN_REL_B, "dec.blk.%d.attn_rel_b" },
  1601. { LLM_TENSOR_DEC_CROSS_ATTN_NORM, "dec.blk.%d.cross_attn_norm" },
  1602. { LLM_TENSOR_DEC_CROSS_ATTN_Q, "dec.blk.%d.cross_attn_q" },
  1603. { LLM_TENSOR_DEC_CROSS_ATTN_K, "dec.blk.%d.cross_attn_k" },
  1604. { LLM_TENSOR_DEC_CROSS_ATTN_V, "dec.blk.%d.cross_attn_v" },
  1605. { LLM_TENSOR_DEC_CROSS_ATTN_OUT, "dec.blk.%d.cross_attn_o" },
  1606. { LLM_TENSOR_DEC_CROSS_ATTN_REL_B, "dec.blk.%d.cross_attn_rel_b" },
  1607. { LLM_TENSOR_DEC_FFN_NORM, "dec.blk.%d.ffn_norm" },
  1608. { LLM_TENSOR_DEC_FFN_GATE, "dec.blk.%d.ffn_gate" },
  1609. { LLM_TENSOR_DEC_FFN_DOWN, "dec.blk.%d.ffn_down" },
  1610. { LLM_TENSOR_DEC_FFN_UP, "dec.blk.%d.ffn_up" },
  1611. { LLM_TENSOR_ENC_OUTPUT_NORM, "enc.output_norm" },
  1612. { LLM_TENSOR_ENC_ATTN_NORM, "enc.blk.%d.attn_norm" },
  1613. { LLM_TENSOR_ENC_ATTN_Q, "enc.blk.%d.attn_q" },
  1614. { LLM_TENSOR_ENC_ATTN_K, "enc.blk.%d.attn_k" },
  1615. { LLM_TENSOR_ENC_ATTN_V, "enc.blk.%d.attn_v" },
  1616. { LLM_TENSOR_ENC_ATTN_OUT, "enc.blk.%d.attn_o" },
  1617. { LLM_TENSOR_ENC_ATTN_REL_B, "enc.blk.%d.attn_rel_b" },
  1618. { LLM_TENSOR_ENC_FFN_NORM, "enc.blk.%d.ffn_norm" },
  1619. { LLM_TENSOR_ENC_FFN_GATE, "enc.blk.%d.ffn_gate" },
  1620. { LLM_TENSOR_ENC_FFN_DOWN, "enc.blk.%d.ffn_down" },
  1621. { LLM_TENSOR_ENC_FFN_UP, "enc.blk.%d.ffn_up" },
  1622. },
  1623. },
  1624. {
  1625. LLM_ARCH_T5ENCODER,
  1626. {
  1627. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1628. { LLM_TENSOR_OUTPUT, "output" },
  1629. { LLM_TENSOR_ENC_OUTPUT_NORM, "enc.output_norm" },
  1630. { LLM_TENSOR_ENC_ATTN_NORM, "enc.blk.%d.attn_norm" },
  1631. { LLM_TENSOR_ENC_ATTN_Q, "enc.blk.%d.attn_q" },
  1632. { LLM_TENSOR_ENC_ATTN_K, "enc.blk.%d.attn_k" },
  1633. { LLM_TENSOR_ENC_ATTN_V, "enc.blk.%d.attn_v" },
  1634. { LLM_TENSOR_ENC_ATTN_OUT, "enc.blk.%d.attn_o" },
  1635. { LLM_TENSOR_ENC_ATTN_REL_B, "enc.blk.%d.attn_rel_b" },
  1636. { LLM_TENSOR_ENC_FFN_NORM, "enc.blk.%d.ffn_norm" },
  1637. { LLM_TENSOR_ENC_FFN_GATE, "enc.blk.%d.ffn_gate" },
  1638. { LLM_TENSOR_ENC_FFN_DOWN, "enc.blk.%d.ffn_down" },
  1639. { LLM_TENSOR_ENC_FFN_UP, "enc.blk.%d.ffn_up" },
  1640. },
  1641. },
  1642. {
  1643. LLM_ARCH_JAIS,
  1644. {
  1645. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1646. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1647. { LLM_TENSOR_OUTPUT, "output" },
  1648. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1649. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  1650. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1651. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1652. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1653. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1654. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1655. },
  1656. },
  1657. {
  1658. LLM_ARCH_NEMOTRON,
  1659. {
  1660. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1661. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1662. { LLM_TENSOR_OUTPUT, "output" },
  1663. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1664. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1665. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1666. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1667. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1668. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1669. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  1670. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1671. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1672. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1673. },
  1674. },
  1675. {
  1676. LLM_ARCH_NEMOTRON_H,
  1677. {
  1678. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1679. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1680. { LLM_TENSOR_OUTPUT, "output" },
  1681. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1682. // mamba(2) ssm layers
  1683. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  1684. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  1685. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  1686. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  1687. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  1688. { LLM_TENSOR_SSM_NORM, "blk.%d.ssm_norm" },
  1689. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  1690. // attention layers
  1691. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1692. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1693. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1694. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1695. // dense FFN
  1696. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1697. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1698. },
  1699. },
  1700. {
  1701. LLM_ARCH_EXAONE,
  1702. {
  1703. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1704. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1705. { LLM_TENSOR_OUTPUT, "output" },
  1706. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1707. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1708. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1709. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1710. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1711. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1712. { LLM_TENSOR_ATTN_ROT_EMBD, "blk.%d.attn_rot_embd" },
  1713. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1714. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1715. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1716. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1717. },
  1718. },
  1719. {
  1720. LLM_ARCH_EXAONE4,
  1721. {
  1722. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1723. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1724. { LLM_TENSOR_OUTPUT, "output" },
  1725. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1726. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1727. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1728. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1729. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1730. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1731. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1732. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  1733. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1734. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1735. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1736. { LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
  1737. }
  1738. },
  1739. {
  1740. LLM_ARCH_RWKV6,
  1741. {
  1742. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1743. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  1744. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1745. { LLM_TENSOR_OUTPUT, "output" },
  1746. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1747. { LLM_TENSOR_ATTN_NORM_2, "blk.%d.attn_norm_2" },
  1748. { LLM_TENSOR_TIME_MIX_W1, "blk.%d.time_mix_w1" },
  1749. { LLM_TENSOR_TIME_MIX_W2, "blk.%d.time_mix_w2" },
  1750. { LLM_TENSOR_TIME_MIX_LERP_X, "blk.%d.time_mix_lerp_x" },
  1751. { LLM_TENSOR_TIME_MIX_LERP_W, "blk.%d.time_mix_lerp_w" },
  1752. { LLM_TENSOR_TIME_MIX_LERP_K, "blk.%d.time_mix_lerp_k" },
  1753. { LLM_TENSOR_TIME_MIX_LERP_V, "blk.%d.time_mix_lerp_v" },
  1754. { LLM_TENSOR_TIME_MIX_LERP_R, "blk.%d.time_mix_lerp_r" },
  1755. { LLM_TENSOR_TIME_MIX_LERP_G, "blk.%d.time_mix_lerp_g" },
  1756. { LLM_TENSOR_TIME_MIX_LERP_FUSED, "blk.%d.time_mix_lerp_fused" },
  1757. { LLM_TENSOR_TIME_MIX_FIRST, "blk.%d.time_mix_first" },
  1758. { LLM_TENSOR_TIME_MIX_DECAY, "blk.%d.time_mix_decay" },
  1759. { LLM_TENSOR_TIME_MIX_DECAY_W1, "blk.%d.time_mix_decay_w1" },
  1760. { LLM_TENSOR_TIME_MIX_DECAY_W2, "blk.%d.time_mix_decay_w2" },
  1761. { LLM_TENSOR_TIME_MIX_KEY, "blk.%d.time_mix_key" },
  1762. { LLM_TENSOR_TIME_MIX_VALUE, "blk.%d.time_mix_value" },
  1763. { LLM_TENSOR_TIME_MIX_RECEPTANCE, "blk.%d.time_mix_receptance" },
  1764. { LLM_TENSOR_TIME_MIX_GATE, "blk.%d.time_mix_gate" },
  1765. { LLM_TENSOR_TIME_MIX_LN, "blk.%d.time_mix_ln" },
  1766. { LLM_TENSOR_TIME_MIX_OUTPUT, "blk.%d.time_mix_output" },
  1767. { LLM_TENSOR_CHANNEL_MIX_LERP_K, "blk.%d.channel_mix_lerp_k" },
  1768. { LLM_TENSOR_CHANNEL_MIX_LERP_R, "blk.%d.channel_mix_lerp_r" },
  1769. { LLM_TENSOR_CHANNEL_MIX_KEY, "blk.%d.channel_mix_key" },
  1770. { LLM_TENSOR_CHANNEL_MIX_VALUE, "blk.%d.channel_mix_value" },
  1771. { LLM_TENSOR_CHANNEL_MIX_RECEPTANCE, "blk.%d.channel_mix_receptance" },
  1772. },
  1773. },
  1774. {
  1775. LLM_ARCH_RWKV6QWEN2,
  1776. {
  1777. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1778. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1779. { LLM_TENSOR_OUTPUT, "output" },
  1780. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1781. { LLM_TENSOR_TIME_MIX_W1, "blk.%d.time_mix_w1" },
  1782. { LLM_TENSOR_TIME_MIX_W2, "blk.%d.time_mix_w2" },
  1783. { LLM_TENSOR_TIME_MIX_LERP_X, "blk.%d.time_mix_lerp_x" },
  1784. { LLM_TENSOR_TIME_MIX_LERP_FUSED, "blk.%d.time_mix_lerp_fused" },
  1785. { LLM_TENSOR_TIME_MIX_FIRST, "blk.%d.time_mix_first" },
  1786. { LLM_TENSOR_TIME_MIX_DECAY, "blk.%d.time_mix_decay" },
  1787. { LLM_TENSOR_TIME_MIX_DECAY_W1, "blk.%d.time_mix_decay_w1" },
  1788. { LLM_TENSOR_TIME_MIX_DECAY_W2, "blk.%d.time_mix_decay_w2" },
  1789. { LLM_TENSOR_TIME_MIX_KEY, "blk.%d.time_mix_key" },
  1790. { LLM_TENSOR_TIME_MIX_VALUE, "blk.%d.time_mix_value" },
  1791. { LLM_TENSOR_TIME_MIX_RECEPTANCE, "blk.%d.time_mix_receptance" },
  1792. { LLM_TENSOR_TIME_MIX_GATE, "blk.%d.time_mix_gate" },
  1793. { LLM_TENSOR_TIME_MIX_OUTPUT, "blk.%d.time_mix_output" },
  1794. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1795. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1796. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1797. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1798. },
  1799. },
  1800. {
  1801. LLM_ARCH_RWKV7,
  1802. {
  1803. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1804. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  1805. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1806. { LLM_TENSOR_OUTPUT, "output" },
  1807. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1808. { LLM_TENSOR_ATTN_NORM_2, "blk.%d.attn_norm_2" },
  1809. { LLM_TENSOR_TIME_MIX_W0, "blk.%d.time_mix_w0" },
  1810. { LLM_TENSOR_TIME_MIX_W1, "blk.%d.time_mix_w1" },
  1811. { LLM_TENSOR_TIME_MIX_W2, "blk.%d.time_mix_w2" },
  1812. { LLM_TENSOR_TIME_MIX_A0, "blk.%d.time_mix_a0" },
  1813. { LLM_TENSOR_TIME_MIX_A1, "blk.%d.time_mix_a1" },
  1814. { LLM_TENSOR_TIME_MIX_A2, "blk.%d.time_mix_a2" },
  1815. { LLM_TENSOR_TIME_MIX_V0, "blk.%d.time_mix_v0" },
  1816. { LLM_TENSOR_TIME_MIX_V1, "blk.%d.time_mix_v1" },
  1817. { LLM_TENSOR_TIME_MIX_V2, "blk.%d.time_mix_v2" },
  1818. { LLM_TENSOR_TIME_MIX_G1, "blk.%d.time_mix_g1" },
  1819. { LLM_TENSOR_TIME_MIX_G2, "blk.%d.time_mix_g2" },
  1820. { LLM_TENSOR_TIME_MIX_K_K, "blk.%d.time_mix_k_k" },
  1821. { LLM_TENSOR_TIME_MIX_K_A, "blk.%d.time_mix_k_a" },
  1822. { LLM_TENSOR_TIME_MIX_R_K, "blk.%d.time_mix_r_k" },
  1823. { LLM_TENSOR_TIME_MIX_LERP_FUSED, "blk.%d.time_mix_lerp_fused" },
  1824. { LLM_TENSOR_TIME_MIX_KEY, "blk.%d.time_mix_key" },
  1825. { LLM_TENSOR_TIME_MIX_VALUE, "blk.%d.time_mix_value" },
  1826. { LLM_TENSOR_TIME_MIX_RECEPTANCE, "blk.%d.time_mix_receptance" },
  1827. { LLM_TENSOR_TIME_MIX_LN, "blk.%d.time_mix_ln" },
  1828. { LLM_TENSOR_TIME_MIX_OUTPUT, "blk.%d.time_mix_output" },
  1829. { LLM_TENSOR_CHANNEL_MIX_LERP_K, "blk.%d.channel_mix_lerp_k" },
  1830. { LLM_TENSOR_CHANNEL_MIX_KEY, "blk.%d.channel_mix_key" },
  1831. { LLM_TENSOR_CHANNEL_MIX_VALUE, "blk.%d.channel_mix_value" },
  1832. },
  1833. },
  1834. {
  1835. LLM_ARCH_ARWKV7,
  1836. {
  1837. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1838. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  1839. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1840. { LLM_TENSOR_OUTPUT, "output" },
  1841. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1842. { LLM_TENSOR_TIME_MIX_W0, "blk.%d.time_mix_w0" },
  1843. { LLM_TENSOR_TIME_MIX_W1, "blk.%d.time_mix_w1" },
  1844. { LLM_TENSOR_TIME_MIX_W2, "blk.%d.time_mix_w2" },
  1845. { LLM_TENSOR_TIME_MIX_A0, "blk.%d.time_mix_a0" },
  1846. { LLM_TENSOR_TIME_MIX_A1, "blk.%d.time_mix_a1" },
  1847. { LLM_TENSOR_TIME_MIX_A2, "blk.%d.time_mix_a2" },
  1848. { LLM_TENSOR_TIME_MIX_V0, "blk.%d.time_mix_v0" },
  1849. { LLM_TENSOR_TIME_MIX_V1, "blk.%d.time_mix_v1" },
  1850. { LLM_TENSOR_TIME_MIX_V2, "blk.%d.time_mix_v2" },
  1851. { LLM_TENSOR_TIME_MIX_G1, "blk.%d.time_mix_g1" },
  1852. { LLM_TENSOR_TIME_MIX_G2, "blk.%d.time_mix_g2" },
  1853. { LLM_TENSOR_TIME_MIX_K_K, "blk.%d.time_mix_k_k" },
  1854. { LLM_TENSOR_TIME_MIX_K_A, "blk.%d.time_mix_k_a" },
  1855. { LLM_TENSOR_TIME_MIX_R_K, "blk.%d.time_mix_r_k" },
  1856. { LLM_TENSOR_TIME_MIX_LERP_FUSED, "blk.%d.time_mix_lerp_fused" },
  1857. { LLM_TENSOR_TIME_MIX_KEY, "blk.%d.time_mix_key" },
  1858. { LLM_TENSOR_TIME_MIX_VALUE, "blk.%d.time_mix_value" },
  1859. { LLM_TENSOR_TIME_MIX_RECEPTANCE, "blk.%d.time_mix_receptance" },
  1860. { LLM_TENSOR_TIME_MIX_LN, "blk.%d.time_mix_ln" },
  1861. { LLM_TENSOR_TIME_MIX_OUTPUT, "blk.%d.time_mix_output" },
  1862. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1863. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1864. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1865. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1866. },
  1867. },
  1868. {
  1869. LLM_ARCH_GRANITE,
  1870. {
  1871. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1872. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1873. { LLM_TENSOR_OUTPUT, "output" },
  1874. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1875. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1876. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1877. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1878. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1879. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1880. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1881. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1882. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1883. },
  1884. },
  1885. {
  1886. LLM_ARCH_GRANITE_MOE,
  1887. {
  1888. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1889. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1890. { LLM_TENSOR_OUTPUT, "output" },
  1891. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1892. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1893. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1894. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1895. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1896. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1897. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1898. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1899. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1900. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1901. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  1902. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  1903. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  1904. },
  1905. },
  1906. {
  1907. LLM_ARCH_GRANITE_HYBRID,
  1908. {
  1909. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1910. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1911. { LLM_TENSOR_OUTPUT, "output" },
  1912. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1913. // mamba(2) ssm layers
  1914. { LLM_TENSOR_SSM_IN, "blk.%d.ssm_in" },
  1915. { LLM_TENSOR_SSM_CONV1D, "blk.%d.ssm_conv1d" },
  1916. { LLM_TENSOR_SSM_DT, "blk.%d.ssm_dt" },
  1917. { LLM_TENSOR_SSM_A, "blk.%d.ssm_a" },
  1918. { LLM_TENSOR_SSM_D, "blk.%d.ssm_d" },
  1919. { LLM_TENSOR_SSM_NORM, "blk.%d.ssm_norm" },
  1920. { LLM_TENSOR_SSM_OUT, "blk.%d.ssm_out" },
  1921. // attention layers
  1922. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1923. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1924. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1925. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1926. // dense FFN
  1927. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1928. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1929. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1930. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1931. // moe FFN
  1932. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1933. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  1934. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  1935. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  1936. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  1937. // shared expert
  1938. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  1939. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  1940. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  1941. },
  1942. },
  1943. {
  1944. LLM_ARCH_CHAMELEON,
  1945. {
  1946. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1947. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1948. { LLM_TENSOR_OUTPUT, "output" },
  1949. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1950. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1951. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1952. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1953. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1954. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  1955. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  1956. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  1957. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  1958. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  1959. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  1960. },
  1961. },
  1962. {
  1963. LLM_ARCH_WAVTOKENIZER_DEC,
  1964. {
  1965. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1966. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  1967. { LLM_TENSOR_CONV1D, "conv1d" },
  1968. { LLM_TENSOR_CONVNEXT_DW, "convnext.%d.dw" },
  1969. { LLM_TENSOR_CONVNEXT_NORM, "convnext.%d.norm" },
  1970. { LLM_TENSOR_CONVNEXT_PW1, "convnext.%d.pw1" },
  1971. { LLM_TENSOR_CONVNEXT_PW2, "convnext.%d.pw2" },
  1972. { LLM_TENSOR_CONVNEXT_GAMMA, "convnext.%d.gamma" },
  1973. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1974. { LLM_TENSOR_OUTPUT, "output" },
  1975. { LLM_TENSOR_POS_NET_CONV1, "posnet.%d.conv1" },
  1976. { LLM_TENSOR_POS_NET_CONV2, "posnet.%d.conv2" },
  1977. { LLM_TENSOR_POS_NET_NORM, "posnet.%d.norm" },
  1978. { LLM_TENSOR_POS_NET_NORM1, "posnet.%d.norm1" },
  1979. { LLM_TENSOR_POS_NET_NORM2, "posnet.%d.norm2" },
  1980. { LLM_TENSOR_POS_NET_ATTN_NORM, "posnet.%d.attn_norm" },
  1981. { LLM_TENSOR_POS_NET_ATTN_Q, "posnet.%d.attn_q" },
  1982. { LLM_TENSOR_POS_NET_ATTN_K, "posnet.%d.attn_k" },
  1983. { LLM_TENSOR_POS_NET_ATTN_V, "posnet.%d.attn_v" },
  1984. { LLM_TENSOR_POS_NET_ATTN_OUT, "posnet.%d.attn_output" },
  1985. },
  1986. },
  1987. {
  1988. LLM_ARCH_BAILINGMOE,
  1989. {
  1990. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  1991. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  1992. { LLM_TENSOR_OUTPUT, "output" },
  1993. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  1994. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  1995. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  1996. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  1997. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  1998. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  1999. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2000. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2001. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2002. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2003. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2004. { LLM_TENSOR_FFN_GATE_INP_SHEXP, "blk.%d.ffn_gate_inp_shexp" },
  2005. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  2006. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  2007. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  2008. },
  2009. },
  2010. {
  2011. LLM_ARCH_BAILINGMOE2,
  2012. {
  2013. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2014. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2015. { LLM_TENSOR_OUTPUT, "output" },
  2016. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2017. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2018. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2019. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  2020. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2021. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2022. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  2023. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2024. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2025. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2026. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2027. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2028. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2029. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2030. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  2031. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  2032. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  2033. { LLM_TENSOR_NEXTN_EH_PROJ, "blk.%d.nextn.eh_proj" },
  2034. { LLM_TENSOR_NEXTN_EMBED_TOKENS, "blk.%d.nextn.embed_tokens" },
  2035. { LLM_TENSOR_NEXTN_ENORM, "blk.%d.nextn.enorm" },
  2036. { LLM_TENSOR_NEXTN_HNORM, "blk.%d.nextn.hnorm" },
  2037. { LLM_TENSOR_NEXTN_SHARED_HEAD_HEAD, "blk.%d.nextn.shared_head_head" },
  2038. { LLM_TENSOR_NEXTN_SHARED_HEAD_NORM, "blk.%d.nextn.shared_head_norm" },
  2039. { LLM_TENSOR_LAYER_OUT_NORM, "blk.%d.layer_output_norm" },
  2040. },
  2041. },
  2042. {
  2043. LLM_ARCH_DOTS1,
  2044. {
  2045. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2046. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2047. { LLM_TENSOR_OUTPUT, "output" },
  2048. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2049. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2050. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2051. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2052. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2053. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2054. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2055. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2056. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2057. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2058. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2059. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2060. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2061. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2062. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2063. { LLM_TENSOR_FFN_GATE_INP_SHEXP, "blk.%d.ffn_gate_inp_shexp" },
  2064. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  2065. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  2066. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  2067. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  2068. }
  2069. },
  2070. {
  2071. LLM_ARCH_ERNIE4_5,
  2072. {
  2073. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2074. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2075. { LLM_TENSOR_OUTPUT, "output" },
  2076. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2077. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2078. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2079. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2080. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2081. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2082. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2083. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2084. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2085. },
  2086. },
  2087. {
  2088. LLM_ARCH_ERNIE4_5_MOE,
  2089. {
  2090. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2091. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2092. { LLM_TENSOR_OUTPUT, "output" },
  2093. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2094. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2095. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2096. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2097. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2098. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2099. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2100. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2101. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2102. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2103. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  2104. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  2105. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  2106. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2107. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2108. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2109. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  2110. },
  2111. },
  2112. {
  2113. LLM_ARCH_HUNYUAN_MOE,
  2114. {
  2115. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2116. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2117. { LLM_TENSOR_OUTPUT, "output" },
  2118. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2119. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2120. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2121. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2122. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2123. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2124. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2125. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2126. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2127. { LLM_TENSOR_FFN_GATE_SHEXP, "blk.%d.ffn_gate_shexp" },
  2128. { LLM_TENSOR_FFN_DOWN_SHEXP, "blk.%d.ffn_down_shexp" },
  2129. { LLM_TENSOR_FFN_UP_SHEXP, "blk.%d.ffn_up_shexp" },
  2130. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2131. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2132. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2133. },
  2134. },
  2135. {
  2136. LLM_ARCH_HUNYUAN_DENSE,
  2137. {
  2138. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2139. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2140. { LLM_TENSOR_OUTPUT, "output" },
  2141. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2142. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2143. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2144. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2145. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2146. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2147. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2148. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2149. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2150. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2151. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2152. },
  2153. },
  2154. {
  2155. LLM_ARCH_SMOLLM3,
  2156. {
  2157. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2158. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2159. { LLM_TENSOR_OUTPUT, "output" },
  2160. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2161. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2162. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2163. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2164. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2165. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2166. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2167. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2168. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2169. },
  2170. },
  2171. {
  2172. LLM_ARCH_OPENAI_MOE,
  2173. {
  2174. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2175. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2176. { LLM_TENSOR_OUTPUT, "output" },
  2177. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2178. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  2179. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2180. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2181. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2182. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2183. { LLM_TENSOR_ATTN_SINKS, "blk.%d.attn_sinks" },
  2184. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2185. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2186. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2187. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2188. },
  2189. },
  2190. {
  2191. LLM_ARCH_LFM2,
  2192. {
  2193. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2194. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2195. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2196. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2197. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2198. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2199. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2200. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2201. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2202. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2203. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2204. { LLM_TENSOR_SHORTCONV_CONV, "blk.%d.shortconv.conv" },
  2205. { LLM_TENSOR_SHORTCONV_INPROJ, "blk.%d.shortconv.in_proj" },
  2206. { LLM_TENSOR_SHORTCONV_OUTPROJ, "blk.%d.shortconv.out_proj" },
  2207. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2208. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  2209. { LLM_TENSOR_OUTPUT, "output" },
  2210. }
  2211. },
  2212. {
  2213. LLM_ARCH_LFM2MOE,
  2214. {
  2215. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2216. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2217. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2218. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2219. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2220. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2221. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2222. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2223. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2224. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2225. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2226. { LLM_TENSOR_SHORTCONV_CONV, "blk.%d.shortconv.conv" },
  2227. { LLM_TENSOR_SHORTCONV_INPROJ, "blk.%d.shortconv.in_proj" },
  2228. { LLM_TENSOR_SHORTCONV_OUTPROJ, "blk.%d.shortconv.out_proj" },
  2229. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2230. { LLM_TENSOR_TOKEN_EMBD_NORM, "token_embd_norm" },
  2231. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2232. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2233. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2234. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2235. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  2236. }
  2237. },
  2238. {
  2239. LLM_ARCH_SMALLTHINKER,
  2240. {
  2241. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2242. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2243. { LLM_TENSOR_OUTPUT, "output" },
  2244. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2245. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2246. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2247. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2248. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2249. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2250. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2251. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2252. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2253. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2254. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2255. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2256. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" }
  2257. },
  2258. },
  2259. {
  2260. LLM_ARCH_APERTUS,
  2261. {
  2262. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2263. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2264. { LLM_TENSOR_OUTPUT, "output" },
  2265. { LLM_TENSOR_ROPE_FREQS, "rope_freqs" },
  2266. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2267. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2268. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2269. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2270. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2271. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2272. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2273. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2274. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2275. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2276. },
  2277. },
  2278. {
  2279. LLM_ARCH_DREAM,
  2280. {
  2281. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2282. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2283. { LLM_TENSOR_OUTPUT, "output" },
  2284. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2285. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2286. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2287. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2288. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2289. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2290. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2291. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2292. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2293. },
  2294. },
  2295. {
  2296. LLM_ARCH_LLADA,
  2297. {
  2298. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2299. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2300. { LLM_TENSOR_OUTPUT, "output" },
  2301. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2302. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2303. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2304. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2305. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2306. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2307. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2308. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2309. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2310. },
  2311. },
  2312. {
  2313. LLM_ARCH_LLADA_MOE,
  2314. {
  2315. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2316. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2317. { LLM_TENSOR_OUTPUT, "output" },
  2318. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2319. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2320. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2321. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2322. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2323. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2324. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2325. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2326. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2327. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2328. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2329. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2330. },
  2331. },
  2332. {
  2333. LLM_ARCH_SEED_OSS,
  2334. {
  2335. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2336. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2337. { LLM_TENSOR_OUTPUT, "output" },
  2338. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2339. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2340. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2341. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2342. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2343. { LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
  2344. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2345. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2346. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2347. },
  2348. },
  2349. {
  2350. LLM_ARCH_GROVEMOE,
  2351. {
  2352. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2353. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2354. { LLM_TENSOR_OUTPUT, "output" },
  2355. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2356. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2357. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2358. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2359. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2360. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2361. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2362. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2363. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2364. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2365. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2366. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2367. { LLM_TENSOR_FFN_GATE_CHEXPS, "blk.%d.ffn_gate_chexps" },
  2368. { LLM_TENSOR_FFN_DOWN_CHEXPS, "blk.%d.ffn_down_chexps" },
  2369. { LLM_TENSOR_FFN_UP_CHEXPS, "blk.%d.ffn_up_chexps" },
  2370. },
  2371. },
  2372. {
  2373. LLM_ARCH_MINIMAX_M2,
  2374. {
  2375. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2376. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2377. { LLM_TENSOR_OUTPUT, "output" },
  2378. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2379. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2380. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2381. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2382. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2383. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2384. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2385. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2386. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2387. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2388. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2389. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2390. { LLM_TENSOR_FFN_EXP_PROBS_B, "blk.%d.exp_probs_b" },
  2391. },
  2392. },
  2393. {
  2394. LLM_ARCH_PANGU_EMBED,
  2395. {
  2396. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2397. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2398. { LLM_TENSOR_OUTPUT, "output" },
  2399. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2400. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2401. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2402. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2403. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2404. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2405. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2406. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2407. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2408. },
  2409. },
  2410. {
  2411. LLM_ARCH_COGVLM,
  2412. {
  2413. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2414. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2415. { LLM_TENSOR_OUTPUT, "output" },
  2416. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2417. { LLM_TENSOR_ATTN_QKV, "blk.%d.attn_qkv" },
  2418. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2419. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2420. { LLM_TENSOR_FFN_GATE, "blk.%d.ffn_gate" },
  2421. { LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
  2422. { LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
  2423. { LLM_TENSOR_VISEXP_ATTN_QKV, "blk.%d.vis_attn_qkv" },
  2424. { LLM_TENSOR_VISEXP_ATTN_OUT, "blk.%d.vis_attn_output" },
  2425. { LLM_TENSOR_VISEXP_FFN_GATE, "blk.%d.vis_gate" },
  2426. { LLM_TENSOR_VISEXP_FFN_DOWN, "blk.%d.vis_down" },
  2427. { LLM_TENSOR_VISEXP_FFN_UP, "blk.%d.vis_up" },
  2428. },
  2429. },
  2430. {
  2431. LLM_ARCH_RND1,
  2432. {
  2433. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2434. { LLM_TENSOR_OUTPUT_NORM, "output_norm" },
  2435. { LLM_TENSOR_OUTPUT, "output" },
  2436. { LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
  2437. { LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
  2438. { LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
  2439. { LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
  2440. { LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
  2441. { LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },
  2442. { LLM_TENSOR_ATTN_OUT, "blk.%d.attn_output" },
  2443. { LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
  2444. { LLM_TENSOR_FFN_GATE_INP, "blk.%d.ffn_gate_inp" },
  2445. { LLM_TENSOR_FFN_GATE_EXPS, "blk.%d.ffn_gate_exps" },
  2446. { LLM_TENSOR_FFN_DOWN_EXPS, "blk.%d.ffn_down_exps" },
  2447. { LLM_TENSOR_FFN_UP_EXPS, "blk.%d.ffn_up_exps" },
  2448. },
  2449. },
  2450. {
  2451. LLM_ARCH_UNKNOWN,
  2452. {
  2453. { LLM_TENSOR_TOKEN_EMBD, "token_embd" },
  2454. },
  2455. },
  2456. };
  2457. static const std::map<llm_tensor, llm_tensor_info> LLM_TENSOR_INFOS = {
  2458. {LLM_TENSOR_TOKEN_EMBD, {LLM_TENSOR_LAYER_INPUT, GGML_OP_GET_ROWS}},
  2459. {LLM_TENSOR_POS_EMBD, {LLM_TENSOR_LAYER_INPUT, GGML_OP_GET_ROWS}},
  2460. {LLM_TENSOR_TOKEN_EMBD_NORM, {LLM_TENSOR_LAYER_INPUT, GGML_OP_GET_ROWS}},
  2461. {LLM_TENSOR_TOKEN_TYPES, {LLM_TENSOR_LAYER_INPUT, GGML_OP_GET_ROWS}},
  2462. {LLM_TENSOR_OUTPUT, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2463. {LLM_TENSOR_CLS, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2464. {LLM_TENSOR_CLS_OUT, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2465. {LLM_TENSOR_DENSE_2_OUT, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}}, // Dense layer output
  2466. {LLM_TENSOR_DENSE_3_OUT, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}}, // Dense layer output
  2467. {LLM_TENSOR_OUTPUT_NORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL}},
  2468. {LLM_TENSOR_DEC_OUTPUT_NORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL}},
  2469. {LLM_TENSOR_ENC_OUTPUT_NORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL}},
  2470. {LLM_TENSOR_ROPE_FREQS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ROPE}},
  2471. {LLM_TENSOR_ROPE_FACTORS_LONG, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ROPE}},
  2472. {LLM_TENSOR_ROPE_FACTORS_SHORT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ROPE}},
  2473. {LLM_TENSOR_ATTN_Q, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2474. {LLM_TENSOR_ATTN_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2475. {LLM_TENSOR_ATTN_V, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2476. {LLM_TENSOR_ATTN_QKV, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2477. {LLM_TENSOR_ATTN_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2478. {LLM_TENSOR_ATTN_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2479. {LLM_TENSOR_FFN_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2480. {LLM_TENSOR_FFN_DOWN, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2481. {LLM_TENSOR_FFN_UP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2482. {LLM_TENSOR_FFN_DOWN_SHEXP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2483. {LLM_TENSOR_FFN_GATE_SHEXP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2484. {LLM_TENSOR_FFN_UP_SHEXP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2485. {LLM_TENSOR_ATTN_Q_A, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2486. {LLM_TENSOR_ATTN_Q_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2487. {LLM_TENSOR_ATTN_KV_A_MQA, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2488. {LLM_TENSOR_ATTN_KV_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2489. {LLM_TENSOR_ATTN_K_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2490. {LLM_TENSOR_ATTN_V_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2491. {LLM_TENSOR_ATTN_SINKS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_SCALE}},
  2492. {LLM_TENSOR_DEC_ATTN_Q, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2493. {LLM_TENSOR_DEC_ATTN_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2494. {LLM_TENSOR_DEC_ATTN_V, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2495. {LLM_TENSOR_DEC_ATTN_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2496. {LLM_TENSOR_DEC_CROSS_ATTN_Q, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2497. {LLM_TENSOR_DEC_CROSS_ATTN_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2498. {LLM_TENSOR_DEC_CROSS_ATTN_V, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2499. {LLM_TENSOR_DEC_CROSS_ATTN_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2500. {LLM_TENSOR_DEC_FFN_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2501. {LLM_TENSOR_DEC_FFN_DOWN, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2502. {LLM_TENSOR_DEC_FFN_UP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2503. {LLM_TENSOR_ENC_ATTN_Q, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2504. {LLM_TENSOR_ENC_ATTN_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2505. {LLM_TENSOR_ENC_ATTN_V, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2506. {LLM_TENSOR_ENC_ATTN_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2507. {LLM_TENSOR_ENC_FFN_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2508. {LLM_TENSOR_ENC_FFN_DOWN, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2509. {LLM_TENSOR_ENC_FFN_UP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2510. {LLM_TENSOR_FFN_GATE_INP_SHEXP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2511. {LLM_TENSOR_FFN_GATE_INP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2512. {LLM_TENSOR_SSM_IN, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2513. {LLM_TENSOR_SSM_X, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2514. {LLM_TENSOR_SSM_DT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2515. {LLM_TENSOR_SSM_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2516. {LLM_TENSOR_TIME_MIX_W1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2517. {LLM_TENSOR_TIME_MIX_W2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2518. {LLM_TENSOR_TIME_MIX_A1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2519. {LLM_TENSOR_TIME_MIX_A2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2520. {LLM_TENSOR_TIME_MIX_V1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2521. {LLM_TENSOR_TIME_MIX_V2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2522. {LLM_TENSOR_TIME_MIX_G1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2523. {LLM_TENSOR_TIME_MIX_G2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2524. {LLM_TENSOR_TIME_MIX_DECAY_W1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2525. {LLM_TENSOR_TIME_MIX_DECAY_W2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2526. {LLM_TENSOR_TIME_MIX_KEY, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2527. {LLM_TENSOR_TIME_MIX_VALUE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2528. {LLM_TENSOR_TIME_MIX_RECEPTANCE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2529. {LLM_TENSOR_TIME_MIX_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2530. {LLM_TENSOR_TIME_MIX_OUTPUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2531. {LLM_TENSOR_CHANNEL_MIX_KEY, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2532. {LLM_TENSOR_CHANNEL_MIX_RECEPTANCE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2533. {LLM_TENSOR_CHANNEL_MIX_VALUE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2534. {LLM_TENSOR_FFN_ACT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_DIV}},
  2535. {LLM_TENSOR_SSM_CONV1D, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_SSM_CONV}},
  2536. {LLM_TENSOR_SSM_A, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_SSM_SCAN}},
  2537. {LLM_TENSOR_SSM_DT_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2538. {LLM_TENSOR_SSM_B_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2539. {LLM_TENSOR_SSM_C_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2540. {LLM_TENSOR_SSM_D, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2541. {LLM_TENSOR_SSM_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2542. {LLM_TENSOR_TIME_MIX_LERP_X, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2543. {LLM_TENSOR_TIME_MIX_LN, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2544. {LLM_TENSOR_CHANNEL_MIX_LERP_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2545. {LLM_TENSOR_CHANNEL_MIX_LERP_R, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2546. {LLM_TENSOR_TIME_MIX_K_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2547. {LLM_TENSOR_TIME_MIX_K_A, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2548. {LLM_TENSOR_TIME_MIX_R_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2549. {LLM_TENSOR_TIME_MIX_LERP_W, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2550. {LLM_TENSOR_TIME_MIX_LERP_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2551. {LLM_TENSOR_TIME_MIX_LERP_V, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2552. {LLM_TENSOR_TIME_MIX_LERP_R, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2553. {LLM_TENSOR_TIME_MIX_LERP_G, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2554. {LLM_TENSOR_TIME_MIX_LERP_FUSED, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2555. {LLM_TENSOR_TIME_MIX_DECAY, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2556. {LLM_TENSOR_TIME_MIX_W0, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2557. {LLM_TENSOR_TIME_MIX_A0, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2558. {LLM_TENSOR_TIME_MIX_V0, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2559. {LLM_TENSOR_TIME_MIX_FIRST, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_RWKV_WKV6}},
  2560. {LLM_TENSOR_ATTN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2561. {LLM_TENSOR_ATTN_NORM_2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2562. {LLM_TENSOR_ATTN_OUT_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2563. {LLM_TENSOR_ATTN_POST_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2564. {LLM_TENSOR_FFN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2565. {LLM_TENSOR_FFN_POST_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2566. {LLM_TENSOR_FFN_NORM_EXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2567. {LLM_TENSOR_ATTN_Q_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2568. {LLM_TENSOR_ATTN_K_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2569. {LLM_TENSOR_LAYER_OUT_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2570. {LLM_TENSOR_ATTN_Q_A_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2571. {LLM_TENSOR_ATTN_KV_A_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2572. {LLM_TENSOR_ATTN_SUB_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2573. {LLM_TENSOR_FFN_SUB_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2574. {LLM_TENSOR_DEC_ATTN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2575. {LLM_TENSOR_DEC_CROSS_ATTN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2576. {LLM_TENSOR_DEC_FFN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2577. {LLM_TENSOR_ENC_ATTN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2578. {LLM_TENSOR_ENC_FFN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2579. {LLM_TENSOR_DEC_ATTN_REL_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_GET_ROWS}},
  2580. {LLM_TENSOR_ENC_ATTN_REL_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_GET_ROWS}},
  2581. {LLM_TENSOR_FFN_DOWN_EXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT_ID}},
  2582. {LLM_TENSOR_FFN_GATE_EXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT_ID}},
  2583. {LLM_TENSOR_FFN_UP_EXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT_ID}},
  2584. {LLM_TENSOR_FFN_DOWN_CHEXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT_ID}},
  2585. {LLM_TENSOR_FFN_GATE_CHEXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT_ID}},
  2586. {LLM_TENSOR_FFN_UP_CHEXPS, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT_ID}},
  2587. {LLM_TENSOR_FFN_EXP_PROBS_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_ADD}},
  2588. // altup / laurel (gemma 3n)
  2589. {LLM_TENSOR_PER_LAYER_TOKEN_EMBD, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_GET_ROWS}},
  2590. {LLM_TENSOR_PER_LAYER_MODEL_PROJ, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2591. {LLM_TENSOR_PER_LAYER_PROJ_NORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL}},
  2592. {LLM_TENSOR_ALTUP_PROJ, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2593. {LLM_TENSOR_ALTUP_UNEMBD_PROJ, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2594. {LLM_TENSOR_PER_LAYER_INP_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2595. {LLM_TENSOR_PER_LAYER_PROJ, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2596. {LLM_TENSOR_PER_LAYER_POST_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2597. {LLM_TENSOR_ALTUP_CORRECT_COEF, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2598. {LLM_TENSOR_ALTUP_CORRECT_SCALE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2599. {LLM_TENSOR_ALTUP_PREDICT_COEF, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2600. {LLM_TENSOR_ALTUP_ROUTER, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2601. {LLM_TENSOR_ALTUP_ROUTER_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2602. {LLM_TENSOR_LAUREL_L, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2603. {LLM_TENSOR_LAUREL_R, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2604. {LLM_TENSOR_LAUREL_POST_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2605. // this tensor is loaded for T5, but never used
  2606. {LLM_TENSOR_DEC_CROSS_ATTN_REL_B, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_NONE}},
  2607. {LLM_TENSOR_CONV1D, {LLM_TENSOR_LAYER_INPUT, GGML_OP_IM2COL}},
  2608. {LLM_TENSOR_POS_NET_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2609. {LLM_TENSOR_POS_NET_NORM1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2610. {LLM_TENSOR_POS_NET_NORM2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2611. {LLM_TENSOR_POS_NET_CONV1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_IM2COL}},
  2612. {LLM_TENSOR_POS_NET_CONV2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_IM2COL}},
  2613. {LLM_TENSOR_POS_NET_ATTN_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2614. {LLM_TENSOR_POS_NET_ATTN_Q, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2615. {LLM_TENSOR_POS_NET_ATTN_K, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2616. {LLM_TENSOR_POS_NET_ATTN_V, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2617. {LLM_TENSOR_POS_NET_ATTN_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2618. {LLM_TENSOR_CONVNEXT_DW, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_IM2COL}},
  2619. {LLM_TENSOR_CONVNEXT_NORM, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2620. {LLM_TENSOR_CONVNEXT_PW1, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2621. {LLM_TENSOR_CONVNEXT_PW2, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2622. {LLM_TENSOR_CONVNEXT_GAMMA, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL}},
  2623. {LLM_TENSOR_SHORTCONV_CONV, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_SSM_CONV}},
  2624. {LLM_TENSOR_SHORTCONV_INPROJ, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2625. {LLM_TENSOR_SHORTCONV_OUTPROJ, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2626. {LLM_TENSOR_VISEXP_ATTN_QKV, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2627. {LLM_TENSOR_VISEXP_ATTN_OUT, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2628. {LLM_TENSOR_VISEXP_FFN_GATE, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2629. {LLM_TENSOR_VISEXP_FFN_DOWN, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2630. {LLM_TENSOR_VISEXP_FFN_UP, {LLM_TENSOR_LAYER_REPEATING, GGML_OP_MUL_MAT}},
  2631. // NextN/MTP tensors are currently ignored (reserved for future MTP support)
  2632. // These tensors only exist in the last layer(s) and are treated as output tensors
  2633. {LLM_TENSOR_NEXTN_EH_PROJ, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2634. {LLM_TENSOR_NEXTN_EMBED_TOKENS, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_GET_ROWS}},
  2635. {LLM_TENSOR_NEXTN_ENORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_GET_ROWS}},
  2636. {LLM_TENSOR_NEXTN_HNORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL}},
  2637. {LLM_TENSOR_NEXTN_SHARED_HEAD_HEAD, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL_MAT}},
  2638. {LLM_TENSOR_NEXTN_SHARED_HEAD_NORM, {LLM_TENSOR_LAYER_OUTPUT, GGML_OP_MUL}},
  2639. };
  2640. LLM_KV::LLM_KV(llm_arch arch, const char * suffix) : arch(arch), suffix(suffix) {}
  2641. std::string LLM_KV::operator()(llm_kv kv) const {
  2642. std::string name = ::format(LLM_KV_NAMES.at(kv), LLM_ARCH_NAMES.at(arch));
  2643. if (suffix != nullptr) {
  2644. name += ".";
  2645. name += suffix;
  2646. }
  2647. return name;
  2648. }
  2649. std::string LLM_TN_IMPL::str() const {
  2650. if (LLM_TENSOR_NAMES.at(arch).find(tensor) == LLM_TENSOR_NAMES.at(arch).end()) {
  2651. return "__missing__";
  2652. }
  2653. std::string name = ::format(LLM_TENSOR_NAMES.at(arch).at(tensor), bid, xid);
  2654. if (suffix != nullptr) {
  2655. name += ".";
  2656. name += suffix;
  2657. }
  2658. return name;
  2659. }
  2660. const char * llm_arch_name(llm_arch arch) {
  2661. auto it = LLM_ARCH_NAMES.find(arch);
  2662. if (it == LLM_ARCH_NAMES.end()) {
  2663. return "unknown";
  2664. }
  2665. return it->second;
  2666. }
  2667. llm_arch llm_arch_from_string(const std::string & name) {
  2668. for (const auto & kv : LLM_ARCH_NAMES) { // NOLINT
  2669. if (kv.second == name) {
  2670. return kv.first;
  2671. }
  2672. }
  2673. return LLM_ARCH_UNKNOWN;
  2674. }
  2675. const llm_tensor_info & llm_tensor_info_for(llm_tensor tensor) {
  2676. return LLM_TENSOR_INFOS.at(tensor);
  2677. }
  2678. bool llm_arch_is_recurrent(const llm_arch & arch) {
  2679. switch (arch) {
  2680. case LLM_ARCH_MAMBA:
  2681. case LLM_ARCH_MAMBA2:
  2682. case LLM_ARCH_RWKV6:
  2683. case LLM_ARCH_RWKV6QWEN2:
  2684. case LLM_ARCH_RWKV7:
  2685. case LLM_ARCH_ARWKV7:
  2686. return true;
  2687. default:
  2688. return false;
  2689. }
  2690. }
  2691. bool llm_arch_is_hybrid(const llm_arch & arch) {
  2692. switch (arch) {
  2693. case LLM_ARCH_JAMBA:
  2694. case LLM_ARCH_FALCON_H1:
  2695. case LLM_ARCH_PLAMO2:
  2696. case LLM_ARCH_GRANITE_HYBRID:
  2697. case LLM_ARCH_LFM2:
  2698. case LLM_ARCH_LFM2MOE:
  2699. case LLM_ARCH_NEMOTRON_H:
  2700. return true;
  2701. default:
  2702. return false;
  2703. }
  2704. }
  2705. bool llm_arch_is_diffusion(const llm_arch & arch) {
  2706. switch (arch) {
  2707. case LLM_ARCH_DREAM:
  2708. case LLM_ARCH_LLADA:
  2709. case LLM_ARCH_LLADA_MOE:
  2710. case LLM_ARCH_RND1:
  2711. return true;
  2712. default:
  2713. return false;
  2714. }
  2715. }