clip.cpp 135 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206
  1. // NOTE: This is modified from clip.cpp only for LLaVA,
  2. // so there might be still unnecessary artifacts hanging around
  3. // I'll gradually clean and extend it
  4. // Note: Even when using identical normalized image inputs (see normalize_image_u8_to_f32()) we have a significant difference in resulting embeddings compared to pytorch
  5. #include "clip.h"
  6. #include "ggml.h"
  7. #include "ggml-cpp.h"
  8. #include "ggml-cpu.h"
  9. #include "ggml-alloc.h"
  10. #include "ggml-backend.h"
  11. #include "gguf.h"
  12. #define STB_IMAGE_IMPLEMENTATION
  13. #include "stb_image.h"
  14. #include <cassert>
  15. #include <cmath>
  16. #include <cstdlib>
  17. #include <cstring>
  18. #include <fstream>
  19. #include <map>
  20. #include <regex>
  21. #include <stdexcept>
  22. #include <unordered_set>
  23. #include <vector>
  24. #include <sstream>
  25. #include <cinttypes>
  26. #include <limits>
  27. #if defined(LLAVA_LOG_OFF)
  28. # define LOG_INF(...)
  29. # define LOG_WRN(...)
  30. # define LOG_ERR(...)
  31. # define LOG_DBG(...)
  32. #else // defined(LLAVA_LOG_OFF)
  33. # define LOG_INF(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
  34. # define LOG_WRN(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
  35. # define LOG_ERR(...) do { fprintf(stderr, __VA_ARGS__); } while (0)
  36. # define LOG_DBG(...) do { fprintf(stdout, __VA_ARGS__); } while (0)
  37. #endif // defined(LLAVA_LOG_OFF)
  38. //#define CLIP_DEBUG_FUNCTIONS
  39. // RGB uint8 image
  40. struct clip_image_u8 {
  41. int nx;
  42. int ny;
  43. std::vector<uint8_t> buf;
  44. };
  45. // RGB float32 image (NHWC)
  46. // Memory layout: RGBRGBRGB...
  47. struct clip_image_f32 {
  48. int nx;
  49. int ny;
  50. std::vector<float> buf;
  51. };
  52. static std::string format(const char * fmt, ...) {
  53. va_list ap;
  54. va_list ap2;
  55. va_start(ap, fmt);
  56. va_copy(ap2, ap);
  57. int size = vsnprintf(NULL, 0, fmt, ap);
  58. GGML_ASSERT(size >= 0 && size < INT_MAX); // NOLINT
  59. std::vector<char> buf(size + 1);
  60. int size2 = vsnprintf(buf.data(), size + 1, fmt, ap2);
  61. GGML_ASSERT(size2 == size);
  62. va_end(ap2);
  63. va_end(ap);
  64. return std::string(buf.data(), buf.size());
  65. }
  66. //
  67. // key constants
  68. //
  69. #define KEY_FTYPE "general.file_type"
  70. #define KEY_NAME "general.name"
  71. #define KEY_DESCRIPTION "general.description"
  72. #define KEY_HAS_TEXT_ENC "clip.has_text_encoder"
  73. #define KEY_HAS_VIS_ENC "clip.has_vision_encoder"
  74. #define KEY_HAS_LLAVA_PROJ "clip.has_llava_projector"
  75. #define KEY_HAS_MINICPMV_PROJ "clip.has_minicpmv_projector"
  76. #define KEY_HAS_GLM_PROJ "clip.has_glm_projector"
  77. #define KEY_MINICPMV_VERSION "clip.minicpmv_version"
  78. #define KEY_HAS_QWEN2VL_MERGER "clip.has_qwen2vl_merger"
  79. #define KEY_USE_GELU "clip.use_gelu"
  80. #define KEY_USE_SILU "clip.use_silu"
  81. #define KEY_N_EMBD "clip.%s.embedding_length"
  82. #define KEY_N_FF "clip.%s.feed_forward_length"
  83. #define KEY_N_BLOCK "clip.%s.block_count"
  84. #define KEY_N_HEAD "clip.%s.attention.head_count"
  85. #define KEY_LAYER_NORM_EPS "clip.%s.attention.layer_norm_epsilon"
  86. #define KEY_PROJ_DIM "clip.%s.projection_dim"
  87. #define KEY_TOKENS "tokenizer.ggml.tokens"
  88. #define KEY_N_POSITIONS "clip.text.context_length"
  89. #define KEY_IMAGE_SIZE "clip.vision.image_size"
  90. #define KEY_PATCH_SIZE "clip.vision.patch_size"
  91. #define KEY_IMAGE_MEAN "clip.vision.image_mean"
  92. #define KEY_IMAGE_STD "clip.vision.image_std"
  93. #define KEY_PROJ_TYPE "clip.projector_type"
  94. #define KEY_FEATURE_LAYER "clip.vision.feature_layer"
  95. #define KEY_MM_PATCH_MERGE_TYPE "clip.vision.mm_patch_merge_type"
  96. #define KEY_IMAGE_GRID_PINPOINTS "clip.vision.image_grid_pinpoints"
  97. #define KEY_IMAGE_CROP_RESOLUTION "clip.vision.image_crop_resolution"
  98. //
  99. // tensor name constants
  100. //
  101. #define TN_TOKEN_EMBD "%s.token_embd.weight"
  102. #define TN_POS_EMBD "%s.position_embd.weight"
  103. #define TN_CLASS_EMBD "v.class_embd"
  104. #define TN_PATCH_EMBD "v.patch_embd.weight" // not rename tensor with ".0" postfix for backwrad compat
  105. #define TN_PATCH_EMBD_1 "v.patch_embd.weight.1"
  106. #define TN_PATCH_BIAS "v.patch_embd.bias"
  107. #define TN_ATTN_K "%s.blk.%d.attn_k.%s"
  108. #define TN_ATTN_Q "%s.blk.%d.attn_q.%s"
  109. #define TN_ATTN_V "%s.blk.%d.attn_v.%s"
  110. #define TN_ATTN_OUTPUT "%s.blk.%d.attn_out.%s"
  111. #define TN_FFN_DOWN "%s.blk.%d.ffn_down.%s"
  112. #define TN_FFN_UP "%s.blk.%d.ffn_up.%s"
  113. #define TN_LN_1 "%s.blk.%d.ln1.%s"
  114. #define TN_LN_2 "%s.blk.%d.ln2.%s"
  115. #define TN_LN_PRE "%s.pre_ln.%s"
  116. #define TN_LN_POST "%s.post_ln.%s"
  117. #define TN_TEXT_PROJ "text_projection.weight"
  118. #define TN_VIS_PROJ "visual_projection.weight"
  119. #define TN_LLAVA_PROJ "mm.%d.%s"
  120. #define TN_MVLM_PROJ_MLP "mm.model.mlp.%d.%s"
  121. #define TN_MVLM_PROJ_BLOCK "mm.model.mb_block.%d.block.%d.%s"
  122. #define TN_MVLM_PROJ_PEG "mm.model.peg.%d.%s"
  123. #define TN_IMAGE_NEWLINE "model.image_newline"
  124. #define TN_MM_INP_PROJ "mm.input_projection.weight" // gemma3
  125. #define TN_MM_SOFT_EMB_N "mm.soft_emb_norm.weight" // gemma3
  126. #define TN_MINICPMV_POS_EMBD_K "resampler.pos_embed_k"
  127. #define TN_MINICPMV_QUERY "resampler.query"
  128. #define TN_MINICPMV_PROJ "resampler.proj.weight"
  129. #define TN_MINICPMV_KV_PROJ "resampler.kv.weight"
  130. #define TN_MINICPMV_ATTN "resampler.attn.%s.%s"
  131. #define TN_MINICPMV_LN "resampler.ln_%s.%s"
  132. #define TN_GLM_ADAPER_CONV "adapter.conv.%s"
  133. #define TN_GLM_ADAPTER_LINEAR "adapter.linear.linear.%s"
  134. #define TN_GLM_ADAPTER_NORM_1 "adapter.linear.norm1.%s"
  135. #define TN_GLM_ADAPTER_D_H_2_4H "adapter.linear.dense_h_to_4h.%s"
  136. #define TN_GLM_ADAPTER_GATE "adapter.linear.gate.%s"
  137. #define TN_GLM_ADAPTER_D_4H_2_H "adapter.linear.dense_4h_to_h.%s"
  138. #define TN_GLM_BOI_W "adapter.boi"
  139. #define TN_GLM_EOI_W "adapter.eoi"
  140. enum projector_type {
  141. PROJECTOR_TYPE_MLP,
  142. PROJECTOR_TYPE_MLP_NORM,
  143. PROJECTOR_TYPE_LDP,
  144. PROJECTOR_TYPE_LDPV2,
  145. PROJECTOR_TYPE_RESAMPLER,
  146. PROJECTOR_TYPE_GLM_EDGE,
  147. PROJECTOR_TYPE_MERGER,
  148. PROJECTOR_TYPE_GEMMA3,
  149. PROJECTOR_TYPE_UNKNOWN,
  150. };
  151. static std::map<projector_type, std::string> PROJECTOR_TYPE_NAMES = {
  152. { PROJECTOR_TYPE_MLP, "mlp" },
  153. { PROJECTOR_TYPE_LDP, "ldp" },
  154. { PROJECTOR_TYPE_LDPV2, "ldpv2"},
  155. { PROJECTOR_TYPE_RESAMPLER, "resampler"},
  156. { PROJECTOR_TYPE_GLM_EDGE, "adapter"},
  157. { PROJECTOR_TYPE_MERGER, "qwen2vl_merger"},
  158. { PROJECTOR_TYPE_GEMMA3, "gemma3"},
  159. };
  160. //
  161. // utilities to get data from a gguf file
  162. //
  163. static int get_key_idx(const gguf_context * ctx, const char * key) {
  164. int i = gguf_find_key(ctx, key);
  165. if (i == -1) {
  166. LOG_ERR("key %s not found in file\n", key);
  167. throw std::runtime_error(format("Missing required key: %s", key));
  168. }
  169. return i;
  170. }
  171. static uint32_t get_u32(const gguf_context * ctx, const std::string & key) {
  172. const int i = get_key_idx(ctx, key.c_str());
  173. return gguf_get_val_u32(ctx, i);
  174. }
  175. static float get_f32(const gguf_context * ctx, const std::string & key) {
  176. const int i = get_key_idx(ctx, key.c_str());
  177. return gguf_get_val_f32(ctx, i);
  178. }
  179. static struct ggml_tensor * get_tensor(struct ggml_context * ctx, const std::string & name) {
  180. struct ggml_tensor * cur = ggml_get_tensor(ctx, name.c_str());
  181. if (!cur) {
  182. throw std::runtime_error(format("%s: unable to find tensor %s\n", __func__, name.c_str()));
  183. }
  184. return cur;
  185. }
  186. static std::string get_ftype(int ftype) {
  187. return ggml_type_name(static_cast<ggml_type>(ftype));
  188. }
  189. static std::string gguf_data_to_str(enum gguf_type type, const void * data, int i) {
  190. switch (type) {
  191. case GGUF_TYPE_UINT8: return std::to_string(((const uint8_t *)data)[i]);
  192. case GGUF_TYPE_INT8: return std::to_string(((const int8_t *)data)[i]);
  193. case GGUF_TYPE_UINT16: return std::to_string(((const uint16_t *)data)[i]);
  194. case GGUF_TYPE_INT16: return std::to_string(((const int16_t *)data)[i]);
  195. case GGUF_TYPE_UINT32: return std::to_string(((const uint32_t *)data)[i]);
  196. case GGUF_TYPE_INT32: return std::to_string(((const int32_t *)data)[i]);
  197. case GGUF_TYPE_UINT64: return std::to_string(((const uint64_t *)data)[i]);
  198. case GGUF_TYPE_INT64: return std::to_string(((const int64_t *)data)[i]);
  199. case GGUF_TYPE_FLOAT32: return std::to_string(((const float *)data)[i]);
  200. case GGUF_TYPE_FLOAT64: return std::to_string(((const double *)data)[i]);
  201. case GGUF_TYPE_BOOL: return ((const bool *)data)[i] ? "true" : "false";
  202. default: return format("unknown type %d", type);
  203. }
  204. }
  205. static void replace_all(std::string & s, const std::string & search, const std::string & replace) {
  206. if (search.empty()) {
  207. return;
  208. }
  209. std::string builder;
  210. builder.reserve(s.length());
  211. size_t pos = 0;
  212. size_t last_pos = 0;
  213. while ((pos = s.find(search, last_pos)) != std::string::npos) {
  214. builder.append(s, last_pos, pos - last_pos);
  215. builder.append(replace);
  216. last_pos = pos + search.length();
  217. }
  218. builder.append(s, last_pos, std::string::npos);
  219. s = std::move(builder);
  220. }
  221. static std::string gguf_kv_to_str(const struct gguf_context * ctx_gguf, int i) {
  222. const enum gguf_type type = gguf_get_kv_type(ctx_gguf, i);
  223. switch (type) {
  224. case GGUF_TYPE_STRING:
  225. return gguf_get_val_str(ctx_gguf, i);
  226. case GGUF_TYPE_ARRAY:
  227. {
  228. const enum gguf_type arr_type = gguf_get_arr_type(ctx_gguf, i);
  229. int arr_n = gguf_get_arr_n(ctx_gguf, i);
  230. const void * data = arr_type == GGUF_TYPE_STRING ? nullptr : gguf_get_arr_data(ctx_gguf, i);
  231. std::stringstream ss;
  232. ss << "[";
  233. for (int j = 0; j < arr_n; j++) {
  234. if (arr_type == GGUF_TYPE_STRING) {
  235. std::string val = gguf_get_arr_str(ctx_gguf, i, j);
  236. // escape quotes
  237. replace_all(val, "\\", "\\\\");
  238. replace_all(val, "\"", "\\\"");
  239. ss << '"' << val << '"';
  240. } else if (arr_type == GGUF_TYPE_ARRAY) {
  241. ss << "???";
  242. } else {
  243. ss << gguf_data_to_str(arr_type, data, j);
  244. }
  245. if (j < arr_n - 1) {
  246. ss << ", ";
  247. }
  248. }
  249. ss << "]";
  250. return ss.str();
  251. }
  252. default:
  253. return gguf_data_to_str(type, gguf_get_val_data(ctx_gguf, i), 0);
  254. }
  255. }
  256. static void print_tensor_info(const ggml_tensor * tensor, const char * prefix = "") {
  257. size_t tensor_size = ggml_nbytes(tensor);
  258. LOG_INF("%s: n_dims = %d, name = %s, tensor_size=%zu, shape:[%" PRId64 ", %" PRId64 ", %" PRId64 ", %" PRId64 "], type = %s\n",
  259. prefix, ggml_n_dims(tensor), tensor->name, tensor_size,
  260. tensor->ne[0], tensor->ne[1], tensor->ne[2], tensor->ne[3], ggml_type_name(tensor->type));
  261. }
  262. static projector_type clip_projector_type_from_string(const std::string & name) {
  263. for (const auto & kv : PROJECTOR_TYPE_NAMES) { // NOLINT
  264. if (kv.second == name) {
  265. return kv.first;
  266. }
  267. }
  268. throw std::runtime_error(format("Unknown projector type: %s", name.c_str()));
  269. }
  270. #ifdef CLIP_DEBUG_FUNCTIONS
  271. static void clip_image_write_image_to_ppm(const clip_image_u8& img, const std::string& filename) {
  272. std::ofstream file(filename, std::ios::binary);
  273. if (!file.is_open()) {
  274. LOG_ERR("Failed to open file for writing: %s\n", filename.c_str());
  275. return;
  276. }
  277. // PPM header: P6 format, width, height, and max color value
  278. file << "P6\n" << img.nx << " " << img.ny << "\n255\n";
  279. // Write pixel data
  280. for (size_t i = 0; i < img.buf.size(); i += 3) {
  281. // PPM expects binary data in RGB format, which matches our image buffer
  282. file.write(reinterpret_cast<const char*>(&img.buf[i]), 3);
  283. }
  284. file.close();
  285. }
  286. static void clip_image_save_to_bmp(const clip_image_u8& img, const std::string& filename) {
  287. std::ofstream file(filename, std::ios::binary);
  288. if (!file.is_open()) {
  289. LOG_ERR("Failed to open file for writing: %s\n", filename.c_str());
  290. return;
  291. }
  292. int fileSize = 54 + 3 * img.nx * img.ny; // File header + info header + pixel data
  293. int bytesPerPixel = 3;
  294. int widthInBytes = img.nx * bytesPerPixel;
  295. int paddingAmount = (4 - (widthInBytes % 4)) % 4;
  296. int stride = widthInBytes + paddingAmount;
  297. // Bitmap file header
  298. unsigned char fileHeader[14] = {
  299. 'B','M', // Signature
  300. 0,0,0,0, // Image file size in bytes
  301. 0,0,0,0, // Reserved
  302. 54,0,0,0 // Start of pixel array
  303. };
  304. // Total file size
  305. fileSize = 54 + (stride * img.ny);
  306. fileHeader[2] = (unsigned char)(fileSize);
  307. fileHeader[3] = (unsigned char)(fileSize >> 8);
  308. fileHeader[4] = (unsigned char)(fileSize >> 16);
  309. fileHeader[5] = (unsigned char)(fileSize >> 24);
  310. // Bitmap information header (BITMAPINFOHEADER)
  311. unsigned char infoHeader[40] = {
  312. 40,0,0,0, // Size of this header (40 bytes)
  313. 0,0,0,0, // Image width
  314. 0,0,0,0, // Image height
  315. 1,0, // Number of color planes
  316. 24,0, // Bits per pixel
  317. 0,0,0,0, // No compression
  318. 0,0,0,0, // Image size (can be 0 for no compression)
  319. 0,0,0,0, // X pixels per meter (not specified)
  320. 0,0,0,0, // Y pixels per meter (not specified)
  321. 0,0,0,0, // Total colors (color table not used)
  322. 0,0,0,0 // Important colors (all are important)
  323. };
  324. // Width and height in the information header
  325. infoHeader[4] = (unsigned char)(img.nx);
  326. infoHeader[5] = (unsigned char)(img.nx >> 8);
  327. infoHeader[6] = (unsigned char)(img.nx >> 16);
  328. infoHeader[7] = (unsigned char)(img.nx >> 24);
  329. infoHeader[8] = (unsigned char)(img.ny);
  330. infoHeader[9] = (unsigned char)(img.ny >> 8);
  331. infoHeader[10] = (unsigned char)(img.ny >> 16);
  332. infoHeader[11] = (unsigned char)(img.ny >> 24);
  333. // Write file headers
  334. file.write(reinterpret_cast<char*>(fileHeader), sizeof(fileHeader));
  335. file.write(reinterpret_cast<char*>(infoHeader), sizeof(infoHeader));
  336. // Pixel data
  337. std::vector<unsigned char> padding(3, 0); // Max padding size to be added to each row
  338. for (int y = img.ny - 1; y >= 0; --y) { // BMP files are stored bottom-to-top
  339. for (int x = 0; x < img.nx; ++x) {
  340. // Each pixel
  341. size_t pixelIndex = (y * img.nx + x) * 3;
  342. unsigned char pixel[3] = {
  343. img.buf[pixelIndex + 2], // BMP stores pixels in BGR format
  344. img.buf[pixelIndex + 1],
  345. img.buf[pixelIndex]
  346. };
  347. file.write(reinterpret_cast<char*>(pixel), 3);
  348. }
  349. // Write padding for the row
  350. file.write(reinterpret_cast<char*>(padding.data()), paddingAmount);
  351. }
  352. file.close();
  353. }
  354. // debug function to convert f32 to u8
  355. static void clip_image_convert_f32_to_u8(const clip_image_f32& src, clip_image_u8& dst) {
  356. dst.nx = src.nx;
  357. dst.ny = src.ny;
  358. dst.buf.resize(3 * src.nx * src.ny);
  359. for (size_t i = 0; i < src.buf.size(); ++i) {
  360. dst.buf[i] = static_cast<uint8_t>(std::min(std::max(int(src.buf[i] * 255.0f), 0), 255));
  361. }
  362. }
  363. #endif
  364. //
  365. // clip layers
  366. //
  367. struct clip_hparams {
  368. int32_t image_size;
  369. int32_t patch_size;
  370. int32_t hidden_size;
  371. int32_t n_intermediate;
  372. int32_t projection_dim;
  373. int32_t n_head;
  374. int32_t n_layer;
  375. float eps;
  376. char mm_patch_merge_type[32] = "flat"; // spatial_unpad or flat (default)
  377. std::vector<int32_t> image_grid_pinpoints;
  378. int32_t image_crop_resolution;
  379. std::unordered_set<int32_t> vision_feature_layer;
  380. };
  381. struct clip_layer {
  382. // attention
  383. struct ggml_tensor * k_w;
  384. struct ggml_tensor * k_b;
  385. struct ggml_tensor * q_w;
  386. struct ggml_tensor * q_b;
  387. struct ggml_tensor * v_w;
  388. struct ggml_tensor * v_b;
  389. struct ggml_tensor * o_w;
  390. struct ggml_tensor * o_b;
  391. // layernorm 1
  392. struct ggml_tensor * ln_1_w;
  393. struct ggml_tensor * ln_1_b;
  394. // ff
  395. struct ggml_tensor * ff_i_w;
  396. struct ggml_tensor * ff_i_b;
  397. struct ggml_tensor * ff_o_w;
  398. struct ggml_tensor * ff_o_b;
  399. // layernorm 2
  400. struct ggml_tensor * ln_2_w;
  401. struct ggml_tensor * ln_2_b;
  402. };
  403. struct clip_vision_model {
  404. struct clip_hparams hparams;
  405. // embeddings
  406. struct ggml_tensor * class_embedding;
  407. struct ggml_tensor * patch_embeddings_0;
  408. struct ggml_tensor * patch_embeddings_1; // second Conv2D kernel when we decouple Conv3D along temproal dimension (Qwen2VL)
  409. struct ggml_tensor * patch_bias;
  410. struct ggml_tensor * position_embeddings;
  411. struct ggml_tensor * pre_ln_w;
  412. struct ggml_tensor * pre_ln_b;
  413. std::vector<clip_layer> layers;
  414. struct ggml_tensor * post_ln_w;
  415. struct ggml_tensor * post_ln_b;
  416. struct ggml_tensor * projection;
  417. // LLaVA projection
  418. struct ggml_tensor * mm_0_w = NULL;
  419. struct ggml_tensor * mm_0_b = NULL;
  420. struct ggml_tensor * mm_2_w = NULL;
  421. struct ggml_tensor * mm_2_b = NULL;
  422. struct ggml_tensor * image_newline = NULL;
  423. // Yi type models with mlp+normalization projection
  424. struct ggml_tensor * mm_1_w = NULL; // Yi type models have 0, 1, 3, 4
  425. struct ggml_tensor * mm_1_b = NULL;
  426. struct ggml_tensor * mm_3_w = NULL;
  427. struct ggml_tensor * mm_3_b = NULL;
  428. struct ggml_tensor * mm_4_w = NULL;
  429. struct ggml_tensor * mm_4_b = NULL;
  430. //GLMV-Edge projection
  431. struct ggml_tensor * mm_model_adapter_conv_w;
  432. struct ggml_tensor * mm_model_adapter_conv_b;
  433. struct ggml_tensor * boi_w;
  434. struct ggml_tensor * eoi_w;
  435. // MobileVLM projection
  436. struct ggml_tensor * mm_model_mlp_1_w;
  437. struct ggml_tensor * mm_model_mlp_1_b;
  438. struct ggml_tensor * mm_model_mlp_3_w;
  439. struct ggml_tensor * mm_model_mlp_3_b;
  440. struct ggml_tensor * mm_model_block_1_block_0_0_w;
  441. struct ggml_tensor * mm_model_block_1_block_0_1_w;
  442. struct ggml_tensor * mm_model_block_1_block_0_1_b;
  443. struct ggml_tensor * mm_model_block_1_block_1_fc1_w;
  444. struct ggml_tensor * mm_model_block_1_block_1_fc1_b;
  445. struct ggml_tensor * mm_model_block_1_block_1_fc2_w;
  446. struct ggml_tensor * mm_model_block_1_block_1_fc2_b;
  447. struct ggml_tensor * mm_model_block_1_block_2_0_w;
  448. struct ggml_tensor * mm_model_block_1_block_2_1_w;
  449. struct ggml_tensor * mm_model_block_1_block_2_1_b;
  450. struct ggml_tensor * mm_model_block_2_block_0_0_w;
  451. struct ggml_tensor * mm_model_block_2_block_0_1_w;
  452. struct ggml_tensor * mm_model_block_2_block_0_1_b;
  453. struct ggml_tensor * mm_model_block_2_block_1_fc1_w;
  454. struct ggml_tensor * mm_model_block_2_block_1_fc1_b;
  455. struct ggml_tensor * mm_model_block_2_block_1_fc2_w;
  456. struct ggml_tensor * mm_model_block_2_block_1_fc2_b;
  457. struct ggml_tensor * mm_model_block_2_block_2_0_w;
  458. struct ggml_tensor * mm_model_block_2_block_2_1_w;
  459. struct ggml_tensor * mm_model_block_2_block_2_1_b;
  460. // MobileVLM_V2 projection
  461. struct ggml_tensor * mm_model_mlp_0_w;
  462. struct ggml_tensor * mm_model_mlp_0_b;
  463. struct ggml_tensor * mm_model_mlp_2_w;
  464. struct ggml_tensor * mm_model_mlp_2_b;
  465. struct ggml_tensor * mm_model_peg_0_w;
  466. struct ggml_tensor * mm_model_peg_0_b;
  467. // MINICPMV projection
  468. struct ggml_tensor * mm_model_pos_embed_k;
  469. struct ggml_tensor * mm_model_query;
  470. struct ggml_tensor * mm_model_proj;
  471. struct ggml_tensor * mm_model_kv_proj;
  472. struct ggml_tensor * mm_model_attn_q_w;
  473. struct ggml_tensor * mm_model_attn_q_b;
  474. struct ggml_tensor * mm_model_attn_k_w;
  475. struct ggml_tensor * mm_model_attn_k_b;
  476. struct ggml_tensor * mm_model_attn_v_w;
  477. struct ggml_tensor * mm_model_attn_v_b;
  478. struct ggml_tensor * mm_model_attn_o_w;
  479. struct ggml_tensor * mm_model_attn_o_b;
  480. struct ggml_tensor * mm_model_ln_q_w;
  481. struct ggml_tensor * mm_model_ln_q_b;
  482. struct ggml_tensor * mm_model_ln_kv_w;
  483. struct ggml_tensor * mm_model_ln_kv_b;
  484. struct ggml_tensor * mm_model_ln_post_w;
  485. struct ggml_tensor * mm_model_ln_post_b;
  486. // gemma3
  487. struct ggml_tensor * mm_input_proj_w;
  488. struct ggml_tensor * mm_soft_emb_norm_w;
  489. };
  490. struct clip_ctx {
  491. bool has_text_encoder = false;
  492. bool has_vision_encoder = false;
  493. bool has_llava_projector = false;
  494. bool has_minicpmv_projector = false;
  495. bool has_glm_projector = false;
  496. bool has_qwen2vl_merger = false;
  497. int minicpmv_version = 2;
  498. struct clip_vision_model vision_model;
  499. projector_type proj_type = PROJECTOR_TYPE_MLP;
  500. int32_t max_feature_layer; // unused in newer models like gemma3
  501. float image_mean[3];
  502. float image_std[3];
  503. bool use_gelu = false;
  504. bool use_silu = false;
  505. int32_t ftype = 1;
  506. bool has_class_embedding = true;
  507. bool has_pre_norm = true;
  508. bool has_post_norm = false;
  509. bool has_patch_bias = false;
  510. struct gguf_context * ctx_gguf = nullptr;
  511. struct ggml_context * ctx_data = nullptr;
  512. std::vector<uint8_t> buf_compute_meta;
  513. std::vector<ggml_backend_t> backend_ptrs;
  514. std::vector<ggml_backend_buffer_type_t> backend_buft;
  515. ggml_backend_t backend = nullptr;
  516. ggml_backend_t backend_cpu = nullptr;
  517. ggml_backend_buffer_t buf = nullptr;
  518. ggml_backend_sched_ptr sched;
  519. struct clip_image_size * load_image_size = nullptr;
  520. clip_ctx(clip_context_params & ctx_params) {
  521. backend_cpu = ggml_backend_init_by_type(GGML_BACKEND_DEVICE_TYPE_CPU, nullptr);
  522. backend = ctx_params.use_gpu
  523. ? ggml_backend_init_by_type(GGML_BACKEND_DEVICE_TYPE_GPU, nullptr)
  524. : nullptr;
  525. if (backend) {
  526. LOG_INF("%s: CLIP using %s backend\n", __func__, ggml_backend_name(backend));
  527. backend_ptrs.push_back(backend);
  528. backend_buft.push_back(ggml_backend_get_default_buffer_type(backend));
  529. } else {
  530. backend = backend_cpu;
  531. LOG_INF("%s: CLIP using CPU backend\n", __func__);
  532. }
  533. backend_ptrs.push_back(backend_cpu);
  534. backend_buft.push_back(ggml_backend_get_default_buffer_type(backend_cpu));
  535. sched.reset(
  536. ggml_backend_sched_new(backend_ptrs.data(), backend_buft.data(), backend_ptrs.size(), 8192, false)
  537. );
  538. }
  539. ~clip_ctx() {
  540. ggml_free(ctx_data);
  541. gguf_free(ctx_gguf);
  542. ggml_backend_buffer_free(buf);
  543. ggml_backend_free(backend);
  544. if (backend_cpu != backend) {
  545. ggml_backend_free(backend_cpu);
  546. }
  547. }
  548. };
  549. static ggml_cgraph * clip_image_build_graph_siglip(clip_ctx * ctx, const clip_image_f32_batch * imgs) {
  550. const auto & model = ctx->vision_model;
  551. const auto & hparams = model.hparams;
  552. const int image_size = hparams.image_size;
  553. int image_size_width = image_size;
  554. int image_size_height = image_size;
  555. const int patch_size = hparams.patch_size;
  556. const int num_patches = ((image_size_width / patch_size) * (image_size_height / patch_size));
  557. const int hidden_size = hparams.hidden_size;
  558. const int n_head = hparams.n_head;
  559. const int d_head = hidden_size / n_head;
  560. const int n_layer = hparams.n_layer;
  561. const float eps = hparams.eps;
  562. GGML_ASSERT(imgs->size == 1); // batch_size == 1
  563. struct ggml_init_params params = {
  564. /*.mem_size =*/ ctx->buf_compute_meta.size(),
  565. /*.mem_buffer =*/ ctx->buf_compute_meta.data(),
  566. /*.no_alloc =*/ true,
  567. };
  568. struct ggml_context * ctx0 = ggml_init(params);
  569. struct ggml_cgraph * gf = ggml_new_graph(ctx0);
  570. // input raw
  571. struct ggml_tensor * inp_raw = ggml_new_tensor_3d(ctx0, GGML_TYPE_F32, image_size_width, image_size_height, 3);
  572. ggml_set_name(inp_raw, "inp_raw");
  573. ggml_set_input(inp_raw);
  574. struct ggml_tensor * inp = ggml_conv_2d(ctx0, model.patch_embeddings_0, inp_raw, patch_size, patch_size, 0, 0, 1, 1);
  575. inp = ggml_reshape_2d(ctx0, inp, num_patches, hidden_size);
  576. inp = ggml_cont(ctx0, ggml_transpose(ctx0, inp));
  577. inp = ggml_add(ctx0, inp, model.patch_bias);
  578. // position embeddings
  579. struct ggml_tensor * embeddings = ggml_add(ctx0, inp, model.position_embeddings);
  580. // loop over layers
  581. for (int il = 0; il < n_layer; il++) {
  582. struct ggml_tensor * cur = embeddings; // embeddings = residual, cur = hidden_states
  583. // layernorm1
  584. {
  585. cur = ggml_norm(ctx0, cur, eps);
  586. cur = ggml_add(ctx0, ggml_mul(ctx0, cur, model.layers[il].ln_1_w), model.layers[il].ln_1_b);
  587. }
  588. // self-attention
  589. {
  590. struct ggml_tensor * Q =
  591. ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].q_w, cur), model.layers[il].q_b);
  592. Q = ggml_reshape_3d(ctx0, Q, d_head, n_head, num_patches);
  593. Q = ggml_cont(ctx0, ggml_permute(ctx0, Q, 0, 2, 1, 3));
  594. struct ggml_tensor * K =
  595. ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].k_w, cur), model.layers[il].k_b);
  596. K = ggml_reshape_3d(ctx0, K, d_head, n_head, num_patches);
  597. K = ggml_cont(ctx0, ggml_permute(ctx0, K, 0, 2, 1, 3));
  598. struct ggml_tensor * V =
  599. ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].v_w, cur), model.layers[il].v_b);
  600. V = ggml_reshape_3d(ctx0, V, d_head, n_head, num_patches);
  601. V = ggml_cont(ctx0, ggml_permute(ctx0, V, 1, 2, 0, 3));
  602. struct ggml_tensor * KQ = ggml_mul_mat(ctx0, K, Q);
  603. KQ = ggml_scale_inplace(ctx0, KQ, 1.0f / sqrtf((float)d_head));
  604. KQ = ggml_soft_max_inplace(ctx0, KQ);
  605. struct ggml_tensor * KQV = ggml_mul_mat(ctx0, V, KQ);
  606. KQV = ggml_reshape_3d(ctx0, KQV, d_head, num_patches, n_head);
  607. KQV = ggml_permute(ctx0, KQV, 0, 2, 1, 3);
  608. cur = ggml_cont_2d(ctx0, KQV, hidden_size, num_patches);
  609. }
  610. // attention output
  611. cur = ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].o_w, cur), model.layers[il].o_b);
  612. // re-add the layer input, e.g., residual
  613. cur = ggml_add(ctx0, cur, embeddings);
  614. embeddings = cur; // embeddings = residual, cur = hidden_states
  615. // layernorm2
  616. {
  617. cur = ggml_norm(ctx0, cur, eps);
  618. cur = ggml_add(ctx0, ggml_mul(ctx0, cur, model.layers[il].ln_2_w), model.layers[il].ln_2_b);
  619. }
  620. cur = ggml_mul_mat(ctx0, model.layers[il].ff_i_w, cur);
  621. cur = ggml_add(ctx0, cur, model.layers[il].ff_i_b);
  622. // siglip uses gelu
  623. cur = ggml_gelu(ctx0, cur);
  624. cur = ggml_mul_mat(ctx0, model.layers[il].ff_o_w, cur);
  625. cur = ggml_add(ctx0, cur, model.layers[il].ff_o_b);
  626. // residual 2
  627. cur = ggml_add(ctx0, embeddings, cur);
  628. embeddings = cur;
  629. }
  630. // post-layernorm
  631. if (ctx->has_post_norm) {
  632. embeddings = ggml_norm(ctx0, embeddings, eps);
  633. ggml_set_name(embeddings, "post_ln");
  634. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.post_ln_w), model.post_ln_b);
  635. }
  636. if (ctx->proj_type == PROJECTOR_TYPE_GEMMA3) {
  637. const int batch_size = 1;
  638. const int mm_tokens_per_image = 256; // default value for gemma3
  639. const int tokens_per_side = sqrt(mm_tokens_per_image);
  640. const int patches_per_image = sqrt(num_patches);
  641. const int kernel_size = patches_per_image / tokens_per_side;
  642. embeddings = ggml_cont(ctx0, ggml_transpose(ctx0, embeddings));
  643. embeddings = ggml_reshape_4d(ctx0, embeddings, patches_per_image, patches_per_image, hidden_size, batch_size);
  644. // doing a pool2d to reduce the number of output tokens to 256
  645. embeddings = ggml_pool_2d(ctx0, embeddings, GGML_OP_POOL_AVG, kernel_size, kernel_size, kernel_size, kernel_size, 0, 0);
  646. embeddings = ggml_reshape_3d(ctx0, embeddings, embeddings->ne[0] * embeddings->ne[0], hidden_size, batch_size);
  647. embeddings = ggml_cont(ctx0, ggml_transpose(ctx0, embeddings));
  648. // apply norm before projection
  649. embeddings = ggml_rms_norm(ctx0, embeddings, eps);
  650. embeddings = ggml_mul(ctx0, embeddings, model.mm_soft_emb_norm_w);
  651. // apply projection
  652. embeddings = ggml_mul_mat(ctx0,
  653. ggml_cont(ctx0, ggml_transpose(ctx0, model.mm_input_proj_w)),
  654. embeddings);
  655. }
  656. // build the graph
  657. ggml_build_forward_expand(gf, embeddings);
  658. ggml_free(ctx0);
  659. return gf;
  660. }
  661. static ggml_cgraph * clip_image_build_graph_legacy(clip_ctx * ctx, const clip_image_f32_batch * imgs, struct clip_image_size * load_image_size, bool is_inf = false) {
  662. if (!ctx->has_vision_encoder) {
  663. LOG_ERR("This gguf file seems to have no vision encoder\n");
  664. return nullptr;
  665. }
  666. const auto & model = ctx->vision_model;
  667. const auto & hparams = model.hparams;
  668. const int image_size = hparams.image_size;
  669. int image_size_width = image_size;
  670. int image_size_height = image_size;
  671. if (ctx->has_minicpmv_projector) {
  672. if (load_image_size == nullptr) {
  673. load_image_size = clip_image_size_init();
  674. }
  675. LOG_DBG("%s: %d %d\n", __func__, load_image_size->width, load_image_size->height);
  676. image_size_width = load_image_size->width;
  677. image_size_height = load_image_size->height;
  678. if (is_inf) {
  679. image_size_width = imgs->data->nx;
  680. image_size_height = imgs->data->ny;
  681. }
  682. }
  683. else if (ctx->has_qwen2vl_merger) {
  684. // use the image's native resolution when image is avaible
  685. if (is_inf) {
  686. // if (imgs->data->nx && imgs->data->ny) {
  687. image_size_width = imgs->data->nx;
  688. image_size_height = imgs->data->ny;
  689. }
  690. }
  691. const int patch_size = hparams.patch_size;
  692. const int num_patches = ((image_size_width / patch_size) * (image_size_height / patch_size));
  693. const int patches_w = image_size_width / patch_size;
  694. const int patches_h = image_size_height / patch_size;
  695. const int num_positions = num_patches + (ctx->has_class_embedding ? 1 : 0);
  696. const int num_position_ids = ctx->has_qwen2vl_merger ? num_positions * 4 : num_positions;
  697. const int hidden_size = hparams.hidden_size;
  698. const int n_head = hparams.n_head;
  699. const int d_head = hidden_size / n_head;
  700. const float eps = hparams.eps;
  701. int mrope_sections[4] = {d_head/4, d_head/4, d_head/4, d_head/4};
  702. const int batch_size = imgs->size;
  703. if (ctx->has_llava_projector || ctx->has_minicpmv_projector || ctx->has_glm_projector) {
  704. GGML_ASSERT(batch_size == 1);
  705. }
  706. struct ggml_init_params params = {
  707. /*.mem_size =*/ ctx->buf_compute_meta.size(),
  708. /*.mem_buffer =*/ ctx->buf_compute_meta.data(),
  709. /*.no_alloc =*/ true,
  710. };
  711. struct ggml_context * ctx0 = ggml_init(params);
  712. struct ggml_cgraph * gf = ggml_new_graph(ctx0);
  713. struct ggml_tensor * inp_raw = ggml_new_tensor_4d(ctx0, GGML_TYPE_F32, image_size_width, image_size_height, 3, batch_size);
  714. ggml_set_name(inp_raw, "inp_raw");
  715. ggml_set_input(inp_raw);
  716. struct ggml_tensor * inp = ggml_conv_2d(ctx0, model.patch_embeddings_0, inp_raw, patch_size, patch_size, 0, 0, 1, 1);
  717. if (ctx->has_qwen2vl_merger) {
  718. GGML_ASSERT(image_size_width % (patch_size * 2) == 0);
  719. GGML_ASSERT(image_size_height % (patch_size * 2) == 0);
  720. auto inp_1 = ggml_conv_2d(ctx0, model.patch_embeddings_1, inp_raw, patch_size, patch_size, 0, 0, 1, 1);
  721. inp = ggml_add(ctx0, inp, inp_1);
  722. inp = ggml_cont(ctx0, ggml_permute(ctx0, inp, 1, 2, 0, 3)); // [w, h, c, b] -> [c, w, h, b]
  723. inp = ggml_reshape_4d(
  724. ctx0, inp,
  725. hidden_size * 2, patches_w / 2, patches_h, batch_size);
  726. inp = ggml_reshape_4d(
  727. ctx0, inp,
  728. hidden_size * 2, patches_w / 2, 2, batch_size * (patches_h / 2));
  729. inp = ggml_cont(ctx0, ggml_permute(ctx0, inp, 0, 2, 1, 3));
  730. inp = ggml_reshape_3d(
  731. ctx0, inp,
  732. hidden_size, patches_w * patches_h, batch_size);
  733. }
  734. else {
  735. inp = ggml_reshape_3d(ctx0, inp, num_patches, hidden_size, batch_size);
  736. inp = ggml_cont(ctx0, ggml_permute(ctx0, inp, 1, 0, 2, 3));
  737. }
  738. if (ctx->has_patch_bias) {
  739. // inp = ggml_add(ctx0, inp, ggml_repeat(ctx0, model.patch_bias, inp));
  740. inp = ggml_add(ctx0, inp, model.patch_bias);
  741. }
  742. struct ggml_tensor * embeddings = inp;
  743. struct ggml_tensor * pos_embed = nullptr;
  744. if (ctx->has_llava_projector) {
  745. // concat class_embeddings and patch_embeddings
  746. if (ctx->has_class_embedding) {
  747. embeddings = ggml_new_tensor_3d(ctx0, GGML_TYPE_F32, hidden_size, num_positions, batch_size);
  748. ggml_set_name(embeddings, "embeddings");
  749. ggml_set_input(embeddings);
  750. embeddings = ggml_acc(ctx0, embeddings, model.class_embedding,
  751. embeddings->nb[1], embeddings->nb[2], embeddings->nb[3], 0);
  752. embeddings = ggml_acc(ctx0, embeddings, inp,
  753. embeddings->nb[1], embeddings->nb[2], embeddings->nb[3], model.class_embedding->nb[1]);
  754. }
  755. }
  756. struct ggml_tensor * positions = ggml_new_tensor_1d(ctx0, GGML_TYPE_I32, num_position_ids);
  757. ggml_set_name(positions, "positions");
  758. ggml_set_input(positions);
  759. if (!ctx->has_qwen2vl_merger) { // qwen2vl use rope position embedding
  760. embeddings =
  761. ggml_add(ctx0, embeddings, ggml_get_rows(ctx0, model.position_embeddings, positions));
  762. }
  763. if (ctx->has_minicpmv_projector) {
  764. int pos_w = image_size_width/patch_size;
  765. int pos_h = image_size_height/patch_size;
  766. if (ctx->minicpmv_version == 2) {
  767. pos_embed = ggml_new_tensor_3d(ctx0, GGML_TYPE_F32, 4096, pos_w * pos_h, 1);
  768. }
  769. else if (ctx->minicpmv_version == 3) {
  770. pos_embed = ggml_new_tensor_3d(ctx0, GGML_TYPE_F32, 3584, pos_w * pos_h, 1);
  771. }
  772. else if (ctx->minicpmv_version == 4) {
  773. pos_embed = ggml_new_tensor_3d(ctx0, GGML_TYPE_F32, 3584, pos_w * pos_h, 1);
  774. }
  775. ggml_set_name(pos_embed, "pos_embed");
  776. ggml_set_input(pos_embed);
  777. }
  778. // pre-layernorm
  779. if (ctx->has_pre_norm) {
  780. embeddings = ggml_norm(ctx0, embeddings, eps);
  781. ggml_set_name(embeddings, "pre_ln");
  782. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.pre_ln_w), model.pre_ln_b);
  783. }
  784. std::vector<struct ggml_tensor *> embedding_stack;
  785. const auto & vision_feature_layer = hparams.vision_feature_layer;
  786. // loop over layers
  787. for (int il = 0; il < ctx->max_feature_layer; il++) {
  788. struct ggml_tensor * cur = embeddings; // embeddings = residual, cur = hidden_states
  789. // If this is an embedding feature layer, save the output.
  790. // NOTE: 0 index here refers to the input to the encoder.
  791. if (vision_feature_layer.find(il) != vision_feature_layer.end()) {
  792. embedding_stack.push_back(embeddings);
  793. }
  794. //const size_t nb_q_w = model.layers[il].q_w->nb[0];
  795. // layernorm1
  796. {
  797. cur = ggml_norm(ctx0, cur, eps);
  798. cur = ggml_add(ctx0, ggml_mul(ctx0, cur, model.layers[il].ln_1_w),
  799. model.layers[il].ln_1_b);
  800. }
  801. // self-attention
  802. {
  803. struct ggml_tensor * Q =
  804. ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].q_w, cur), model.layers[il].q_b);
  805. Q = ggml_reshape_4d(ctx0, Q, d_head, n_head, num_positions, batch_size);
  806. if (ctx->has_qwen2vl_merger) {
  807. Q = ggml_rope_multi(
  808. ctx0, Q, positions, nullptr,
  809. d_head/2, mrope_sections, GGML_ROPE_TYPE_VISION, 32768, 10000, 1, 0, 1, 32, 1);
  810. }
  811. Q = ggml_scale_inplace(ctx0, Q, 1.0f / sqrt((float)d_head));
  812. Q = ggml_cont(ctx0, ggml_permute(ctx0, Q, 0, 2, 1, 3));
  813. Q = ggml_reshape_3d(ctx0, Q, d_head, num_positions, n_head * batch_size);
  814. struct ggml_tensor * K =
  815. ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].k_w, cur), model.layers[il].k_b);
  816. K = ggml_reshape_4d(ctx0, K, d_head, n_head, num_positions, batch_size);
  817. if (ctx->has_qwen2vl_merger) {
  818. K = ggml_rope_multi(
  819. ctx0, K, positions, nullptr,
  820. d_head/2, mrope_sections, GGML_ROPE_TYPE_VISION, 32768, 10000, 1, 0, 1, 32, 1);
  821. }
  822. K = ggml_cont(ctx0, ggml_permute(ctx0, K, 0, 2, 1, 3));
  823. K = ggml_reshape_3d(ctx0, K, d_head, num_positions, n_head * batch_size);
  824. struct ggml_tensor * V =
  825. ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].v_w, cur), model.layers[il].v_b);
  826. V = ggml_reshape_4d(ctx0, V, d_head, n_head, num_positions, batch_size);
  827. V = ggml_cont(ctx0, ggml_permute(ctx0, V, 1, 2, 0, 3));
  828. V = ggml_reshape_3d(ctx0, V, num_positions, d_head, n_head * batch_size);
  829. struct ggml_tensor * KQ = ggml_mul_mat(ctx0, K, Q);
  830. KQ = ggml_soft_max_inplace(ctx0, KQ);
  831. struct ggml_tensor * KQV = ggml_mul_mat(ctx0, V, KQ);
  832. KQV = ggml_reshape_4d(ctx0, KQV, d_head, num_positions, n_head, batch_size);
  833. KQV = ggml_permute(ctx0, KQV, 0, 2, 1, 3);
  834. cur = ggml_cont_3d(ctx0, KQV, hidden_size, num_positions, batch_size);
  835. }
  836. // attention output
  837. cur = ggml_add(ctx0, ggml_mul_mat(ctx0, model.layers[il].o_w, cur), model.layers[il].o_b);
  838. // re-add the layer input, e.g., residual
  839. cur = ggml_add(ctx0, cur, embeddings);
  840. embeddings = cur; // embeddings = residual, cur = hidden_states
  841. // layernorm2
  842. {
  843. cur = ggml_norm(ctx0, cur, eps);
  844. cur = ggml_add(ctx0, ggml_mul(ctx0, cur, model.layers[il].ln_2_w), model.layers[il].ln_2_b);
  845. }
  846. cur = ggml_mul_mat(ctx0, model.layers[il].ff_i_w, cur);
  847. cur = ggml_add(ctx0, cur, model.layers[il].ff_i_b);
  848. if (ctx->use_gelu) {
  849. cur = ggml_gelu_inplace(ctx0, cur);
  850. } else if (ctx->use_silu) {
  851. cur = ggml_silu_inplace(ctx0, cur);
  852. } else {
  853. cur = ggml_gelu_quick_inplace(ctx0, cur);
  854. }
  855. cur = ggml_mul_mat(ctx0, model.layers[il].ff_o_w, cur);
  856. cur = ggml_add(ctx0, cur, model.layers[il].ff_o_b);
  857. // residual 2
  858. cur = ggml_add(ctx0, embeddings, cur);
  859. embeddings = cur;
  860. }
  861. // post-layernorm
  862. if (ctx->has_post_norm) {
  863. embeddings = ggml_norm(ctx0, embeddings, eps);
  864. ggml_set_name(embeddings, "post_ln");
  865. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.post_ln_w), model.post_ln_b);
  866. }
  867. // final layer is a vision feature layer
  868. if (vision_feature_layer.find(ctx->max_feature_layer) != vision_feature_layer.end()) {
  869. embedding_stack.push_back(embeddings);
  870. }
  871. // If feature layers are explicitly set, stack them (if we have multiple)
  872. if (!embedding_stack.empty()) {
  873. embeddings = embedding_stack[0];
  874. for (size_t i = 1; i < embedding_stack.size(); i++) {
  875. embeddings = ggml_concat(ctx0, embeddings, embedding_stack[i], 0);
  876. }
  877. }
  878. // llava projector
  879. if (ctx->has_llava_projector) {
  880. embeddings = ggml_reshape_2d(ctx0, embeddings, embeddings->ne[0], embeddings->ne[1]);
  881. struct ggml_tensor * patches = ggml_new_tensor_1d(ctx0, GGML_TYPE_I32, num_patches);
  882. ggml_set_name(patches, "patches");
  883. ggml_set_input(patches);
  884. // shape [1, 576, 1024]
  885. // ne is whcn, ne = [1024, 576, 1, 1]
  886. embeddings = ggml_get_rows(ctx0, embeddings, patches);
  887. // print_tensor_info(embeddings, "embeddings");
  888. // llava projector
  889. if (ctx->proj_type == PROJECTOR_TYPE_MLP) {
  890. embeddings = ggml_mul_mat(ctx0, model.mm_0_w, embeddings);
  891. embeddings = ggml_add(ctx0, embeddings, model.mm_0_b);
  892. embeddings = ggml_gelu(ctx0, embeddings);
  893. embeddings = ggml_mul_mat(ctx0, model.mm_2_w, embeddings);
  894. embeddings = ggml_add(ctx0, embeddings, model.mm_2_b);
  895. }
  896. else if (ctx->proj_type == PROJECTOR_TYPE_MLP_NORM) {
  897. embeddings = ggml_mul_mat(ctx0, model.mm_0_w, embeddings);
  898. embeddings = ggml_add(ctx0, embeddings, model.mm_0_b);
  899. // ggml_tensor_printf(embeddings, "mm_0_w",0,true,false);
  900. // First LayerNorm
  901. embeddings = ggml_norm(ctx0, embeddings, eps);
  902. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.mm_1_w),
  903. model.mm_1_b);
  904. // GELU activation
  905. embeddings = ggml_gelu(ctx0, embeddings);
  906. // Second linear layer
  907. embeddings = ggml_mul_mat(ctx0, model.mm_3_w, embeddings);
  908. embeddings = ggml_add(ctx0, embeddings, model.mm_3_b);
  909. // Second LayerNorm
  910. embeddings = ggml_norm(ctx0, embeddings, eps);
  911. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.mm_4_w),
  912. model.mm_4_b);
  913. }
  914. else if (ctx->proj_type == PROJECTOR_TYPE_LDP) {
  915. // MobileVLM projector
  916. int n_patch = 24;
  917. struct ggml_tensor * mlp_1 = ggml_mul_mat(ctx0, model.mm_model_mlp_1_w, embeddings);
  918. mlp_1 = ggml_add(ctx0, mlp_1, model.mm_model_mlp_1_b);
  919. mlp_1 = ggml_gelu(ctx0, mlp_1);
  920. struct ggml_tensor * mlp_3 = ggml_mul_mat(ctx0, model.mm_model_mlp_3_w, mlp_1);
  921. mlp_3 = ggml_add(ctx0, mlp_3, model.mm_model_mlp_3_b);
  922. // mlp_3 shape = [1, 576, 2048], ne = [2048, 576, 1, 1]
  923. // block 1
  924. struct ggml_tensor * block_1 = nullptr;
  925. {
  926. // transpose from [1, 576, 2048] --> [1, 2048, 576] --> [1, 2048, 24, 24]
  927. mlp_3 = ggml_cont(ctx0, ggml_permute(ctx0, mlp_3, 1, 0, 2, 3));
  928. mlp_3 = ggml_reshape_4d(ctx0, mlp_3, n_patch, n_patch, mlp_3->ne[1], mlp_3->ne[2]);
  929. // stride = 1, padding = 1, bias is nullptr
  930. block_1 = ggml_conv_2d_dw(ctx0, model.mm_model_block_1_block_0_0_w, mlp_3, 1, 1, 1, 1, 1, 1);
  931. // layer norm
  932. // // block_1 shape = [1, 2048, 24, 24], ne = [24, 24, 2048, 1]
  933. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 1, 2, 0, 3));
  934. // block_1 shape = [1, 24, 24, 2048], ne = [2048, 24, 24, 1]
  935. block_1 = ggml_norm(ctx0, block_1, eps);
  936. block_1 = ggml_add(ctx0, ggml_mul(ctx0, block_1, model.mm_model_block_1_block_0_1_w), model.mm_model_block_1_block_0_1_b);
  937. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 2, 0, 1, 3));
  938. // block_1 shape = [1, 2048, 24, 24], ne = [24, 24, 2048, 1]
  939. // hardswish
  940. struct ggml_tensor * block_1_hw = ggml_hardswish(ctx0, block_1);
  941. block_1 = ggml_pool_2d(ctx0, block_1_hw, GGML_OP_POOL_AVG, block_1_hw->ne[0], block_1_hw->ne[1], block_1_hw->ne[0], block_1_hw->ne[1], 0, 0);
  942. // block_1 shape = [1, 2048, 1, 1], ne = [1, 1, 2048, 1]
  943. // pointwise conv
  944. block_1 = ggml_reshape_2d(ctx0, block_1, block_1->ne[0]*block_1->ne[1]*block_1->ne[2], block_1->ne[3]);
  945. block_1 = ggml_mul_mat(ctx0, model.mm_model_block_1_block_1_fc1_w, block_1);
  946. block_1 = ggml_add(ctx0, block_1, model.mm_model_block_1_block_1_fc1_b);
  947. block_1 = ggml_relu(ctx0, block_1);
  948. block_1 = ggml_mul_mat(ctx0, model.mm_model_block_1_block_1_fc2_w, block_1);
  949. block_1 = ggml_add(ctx0, block_1, model.mm_model_block_1_block_1_fc2_b);
  950. block_1 = ggml_hardsigmoid(ctx0, block_1);
  951. // block_1_hw shape = [1, 2048, 24, 24], ne = [24, 24, 2048, 1], block_1 shape = [1, 2048], ne = [2048, 1, 1, 1]
  952. block_1 = ggml_reshape_4d(ctx0, block_1, 1, 1, block_1->ne[0], block_1->ne[1]);
  953. block_1 = ggml_mul(ctx0, block_1_hw, block_1);
  954. int w = block_1->ne[0], h = block_1->ne[1];
  955. block_1 = ggml_reshape_3d(ctx0, block_1, w*h, block_1->ne[2], block_1->ne[3]);
  956. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 1, 0, 2, 3));
  957. // block_1 shape = [1, 24*24, 2048], ne = [24*24, 2048, 1]
  958. block_1 = ggml_mul_mat(ctx0, model.mm_model_block_1_block_2_0_w, block_1);
  959. block_1 = ggml_reshape_4d(ctx0, block_1, block_1->ne[0], w, h, block_1->ne[3]);
  960. // block_1 shape = [1, 24, 24, 2048], ne = [2048, 24, 24, 1]
  961. block_1 = ggml_norm(ctx0, block_1, eps);
  962. block_1 = ggml_add(ctx0, ggml_mul(ctx0, block_1, model.mm_model_block_1_block_2_1_w), model.mm_model_block_1_block_2_1_b);
  963. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 2, 0, 1, 3));
  964. // block1 shape = [1, 2048, 24, 24], ne = [24, 24, 2048, 1]
  965. // residual
  966. block_1 = ggml_add(ctx0, mlp_3, block_1);
  967. }
  968. // block_2
  969. {
  970. // stride = 2
  971. block_1 = ggml_conv_2d_dw(ctx0, model.mm_model_block_2_block_0_0_w, block_1, 2, 2, 1, 1, 1, 1);
  972. // block_1 shape = [1, 2048, 12, 12], ne = [12, 12, 2048, 1]
  973. // layer norm
  974. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 1, 2, 0, 3));
  975. // block_1 shape = [1, 12, 12, 2048], ne = [2048, 12, 12, 1]
  976. block_1 = ggml_norm(ctx0, block_1, eps);
  977. block_1 = ggml_add(ctx0, ggml_mul(ctx0, block_1, model.mm_model_block_2_block_0_1_w), model.mm_model_block_2_block_0_1_b);
  978. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 2, 0, 1, 3));
  979. // block_1 shape = [1, 2048, 12, 12], ne = [12, 12, 2048, 1]
  980. // hardswish
  981. struct ggml_tensor * block_1_hw = ggml_hardswish(ctx0, block_1);
  982. // not sure the parameters is right for globalAvgPooling
  983. block_1 = ggml_pool_2d(ctx0, block_1_hw, GGML_OP_POOL_AVG, block_1_hw->ne[0], block_1_hw->ne[1], block_1_hw->ne[0], block_1_hw->ne[1], 0, 0);
  984. // block_1 shape = [1, 2048, 1, 1], ne = [1, 1, 2048, 1]
  985. // pointwise conv
  986. block_1 = ggml_reshape_2d(ctx0, block_1, block_1->ne[0]*block_1->ne[1]*block_1->ne[2], block_1->ne[3]);
  987. block_1 = ggml_mul_mat(ctx0, model.mm_model_block_2_block_1_fc1_w, block_1);
  988. block_1 = ggml_add(ctx0, block_1, model.mm_model_block_2_block_1_fc1_b);
  989. block_1 = ggml_relu(ctx0, block_1);
  990. block_1 = ggml_mul_mat(ctx0, model.mm_model_block_2_block_1_fc2_w, block_1);
  991. block_1 = ggml_add(ctx0, block_1, model.mm_model_block_2_block_1_fc2_b);
  992. block_1 = ggml_hardsigmoid(ctx0, block_1);
  993. // block_1_hw shape = [1, 2048, 12, 12], ne = [12, 12, 2048, 1], block_1 shape = [1, 2048, 1, 1], ne = [1, 1, 2048, 1]
  994. block_1 = ggml_reshape_4d(ctx0, block_1, 1, 1, block_1->ne[0], block_1->ne[1]);
  995. block_1 = ggml_mul(ctx0, block_1_hw, block_1);
  996. int w = block_1->ne[0], h = block_1->ne[1];
  997. block_1 = ggml_reshape_3d(ctx0, block_1, w*h, block_1->ne[2], block_1->ne[3]);
  998. block_1 = ggml_cont(ctx0, ggml_permute(ctx0, block_1, 1, 0, 2, 3));
  999. // block_1 shape = [1, 24*24, 2048], ne = [24*24, 2048, 1]
  1000. block_1 = ggml_mul_mat(ctx0, model.mm_model_block_2_block_2_0_w, block_1);
  1001. block_1 = ggml_reshape_4d(ctx0, block_1, block_1->ne[0], w, h, block_1->ne[3]);
  1002. // block_1 shape = [1, 12, 12, 2048], ne = [2048, 12, 12, 1]
  1003. block_1 = ggml_norm(ctx0, block_1, eps);
  1004. block_1 = ggml_add(ctx0, ggml_mul(ctx0, block_1, model.mm_model_block_2_block_2_1_w), model.mm_model_block_2_block_2_1_b);
  1005. block_1 = ggml_reshape_3d(ctx0, block_1, block_1->ne[0], block_1->ne[1] * block_1->ne[2], block_1->ne[3]);
  1006. // block_1 shape = [1, 144, 2048], ne = [2048, 144, 1]
  1007. }
  1008. embeddings = block_1;
  1009. }
  1010. else if (ctx->proj_type == PROJECTOR_TYPE_LDPV2)
  1011. {
  1012. int n_patch = 24;
  1013. struct ggml_tensor * mlp_0 = ggml_mul_mat(ctx0, model.mm_model_mlp_0_w, embeddings);
  1014. mlp_0 = ggml_add(ctx0, mlp_0, model.mm_model_mlp_0_b);
  1015. mlp_0 = ggml_gelu(ctx0, mlp_0);
  1016. struct ggml_tensor * mlp_2 = ggml_mul_mat(ctx0, model.mm_model_mlp_2_w, mlp_0);
  1017. mlp_2 = ggml_add(ctx0, mlp_2, model.mm_model_mlp_2_b);
  1018. // mlp_2 ne = [2048, 576, 1, 1]
  1019. // // AVG Pool Layer 2*2, strides = 2
  1020. mlp_2 = ggml_cont(ctx0, ggml_permute(ctx0, mlp_2, 1, 0, 2, 3));
  1021. // mlp_2 ne = [576, 2048, 1, 1]
  1022. mlp_2 = ggml_reshape_4d(ctx0, mlp_2, n_patch, n_patch, mlp_2->ne[1], mlp_2->ne[2]);
  1023. // mlp_2 ne [24, 24, 2048, 1]
  1024. mlp_2 = ggml_pool_2d(ctx0, mlp_2, GGML_OP_POOL_AVG, 2, 2, 2, 2, 0, 0);
  1025. // weight ne = [3, 3, 2048, 1]
  1026. struct ggml_tensor * peg_0 = ggml_conv_2d_dw(ctx0, model.mm_model_peg_0_w, mlp_2, 1, 1, 1, 1, 1, 1);
  1027. peg_0 = ggml_cont(ctx0, ggml_permute(ctx0, peg_0, 1, 2, 0, 3));
  1028. peg_0 = ggml_add(ctx0, peg_0, model.mm_model_peg_0_b);
  1029. mlp_2 = ggml_cont(ctx0, ggml_permute(ctx0, mlp_2, 1, 2, 0, 3));
  1030. peg_0 = ggml_add(ctx0, peg_0, mlp_2);
  1031. peg_0 = ggml_reshape_3d(ctx0, peg_0, peg_0->ne[0], peg_0->ne[1] * peg_0->ne[2], peg_0->ne[3]);
  1032. embeddings = peg_0;
  1033. }
  1034. else {
  1035. GGML_ABORT("fatal error");
  1036. }
  1037. }
  1038. // minicpmv projector
  1039. else if (ctx->has_minicpmv_projector)
  1040. {
  1041. if (ctx->proj_type == PROJECTOR_TYPE_RESAMPLER) {
  1042. struct ggml_tensor * q = model.mm_model_query;
  1043. { // layernorm
  1044. q = ggml_norm(ctx0, q, eps);
  1045. q = ggml_add(ctx0, ggml_mul(ctx0, q, model.mm_model_ln_q_w), model.mm_model_ln_q_b);
  1046. }
  1047. struct ggml_tensor * v = ggml_mul_mat(ctx0, model.mm_model_kv_proj, embeddings);
  1048. { // layernorm
  1049. v = ggml_norm(ctx0, v, eps);
  1050. v = ggml_add(ctx0, ggml_mul(ctx0, v, model.mm_model_ln_kv_w), model.mm_model_ln_kv_b);
  1051. }
  1052. struct ggml_tensor * k;
  1053. { // position
  1054. // q = ggml_add(ctx0, q, model.mm_model_pos_embed);
  1055. k = ggml_add(ctx0, v, pos_embed);
  1056. }
  1057. { // attention
  1058. int hidden_size = 4096;
  1059. const int d_head = 128;
  1060. int n_head = hidden_size/d_head;
  1061. int num_query = 96;
  1062. if (ctx->minicpmv_version == 2) {
  1063. hidden_size = 4096;
  1064. n_head = hidden_size/d_head;
  1065. num_query = 96;
  1066. }
  1067. else if (ctx->minicpmv_version == 3) {
  1068. hidden_size = 3584;
  1069. n_head = hidden_size/d_head;
  1070. num_query = 64;
  1071. }
  1072. else if (ctx->minicpmv_version == 4) {
  1073. hidden_size = 3584;
  1074. n_head = hidden_size/d_head;
  1075. num_query = 64;
  1076. }
  1077. struct ggml_tensor * Q = ggml_add(ctx0, ggml_mul_mat(ctx0, model.mm_model_attn_q_w, q), model.mm_model_attn_q_b);
  1078. Q = ggml_scale_inplace(ctx0, Q, 1.0f / sqrt((float)d_head));
  1079. struct ggml_tensor * K = ggml_add(ctx0, ggml_mul_mat(ctx0, model.mm_model_attn_k_w, k), model.mm_model_attn_k_b);
  1080. struct ggml_tensor * V = ggml_add(ctx0, ggml_mul_mat(ctx0, model.mm_model_attn_v_w, v), model.mm_model_attn_v_b);
  1081. // permute
  1082. Q = ggml_reshape_4d(ctx0, Q, d_head, n_head, num_query, batch_size);
  1083. Q = ggml_cont(ctx0, ggml_permute(ctx0, Q, 0, 2, 1, 3));
  1084. Q = ggml_reshape_3d(ctx0, Q, d_head, num_query, n_head * batch_size);
  1085. K = ggml_reshape_4d(ctx0, K, d_head, n_head, num_positions, batch_size);
  1086. K = ggml_cont(ctx0, ggml_permute(ctx0, K, 0, 2, 1, 3));
  1087. K = ggml_reshape_3d(ctx0, K, d_head, num_positions, n_head * batch_size);
  1088. V = ggml_reshape_4d(ctx0, V, d_head, n_head, num_positions, batch_size);
  1089. V = ggml_cont(ctx0, ggml_permute(ctx0, V, 1, 2, 0, 3));
  1090. V = ggml_reshape_3d(ctx0, V, num_positions, d_head, n_head * batch_size);
  1091. struct ggml_tensor * KQ = ggml_mul_mat(ctx0, K, Q);
  1092. KQ = ggml_soft_max_inplace(ctx0, KQ);
  1093. struct ggml_tensor * KQV = ggml_mul_mat(ctx0, V, KQ);
  1094. KQV = ggml_reshape_4d(ctx0, KQV, d_head, num_query, n_head, batch_size);
  1095. KQV = ggml_permute(ctx0, KQV, 0, 2, 1, 3);
  1096. KQV = ggml_cont_3d(ctx0, KQV, hidden_size, num_query, batch_size);
  1097. embeddings = ggml_add(ctx0, ggml_mul_mat(ctx0, model.mm_model_attn_o_w, KQV), model.mm_model_attn_o_b);
  1098. }
  1099. { // layernorm
  1100. embeddings = ggml_norm(ctx0, embeddings, eps);
  1101. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.mm_model_ln_post_w), model.mm_model_ln_post_b);
  1102. }
  1103. embeddings = ggml_mul_mat(ctx0, model.mm_model_proj, embeddings);
  1104. }
  1105. else {
  1106. GGML_ASSERT(false);
  1107. }
  1108. }
  1109. // glm projector
  1110. else if (ctx->has_glm_projector) {
  1111. if (ctx->proj_type == PROJECTOR_TYPE_GLM_EDGE) {
  1112. size_t gridsz = (size_t)sqrt(embeddings->ne[1]);
  1113. embeddings = ggml_cont(ctx0, ggml_permute(ctx0,embeddings,1,0,2,3));
  1114. embeddings = ggml_reshape_3d(ctx0, embeddings, gridsz, gridsz, embeddings->ne[1]);
  1115. embeddings = ggml_conv_2d(ctx0, model.mm_model_adapter_conv_w, embeddings, 2, 2, 0, 0, 1, 1);
  1116. embeddings = ggml_reshape_3d(ctx0, embeddings,embeddings->ne[0]*embeddings->ne[1] , embeddings->ne[2], batch_size);
  1117. embeddings = ggml_cont(ctx0, ggml_permute(ctx0,embeddings, 1, 0, 2, 3));
  1118. embeddings = ggml_add(ctx0, embeddings, model.mm_model_adapter_conv_b);
  1119. //GLU
  1120. {
  1121. embeddings = ggml_mul_mat(ctx0, model.mm_model_mlp_0_w, embeddings);
  1122. embeddings = ggml_norm(ctx0, embeddings, eps);
  1123. embeddings = ggml_add(ctx0, ggml_mul(ctx0, embeddings, model.mm_model_ln_q_w), model.mm_model_ln_q_b);
  1124. embeddings = ggml_gelu_inplace(ctx0, embeddings);
  1125. struct ggml_tensor * x = embeddings;
  1126. embeddings = ggml_mul_mat(ctx0, model.mm_model_mlp_2_w, embeddings);
  1127. x = ggml_mul_mat(ctx0, model.mm_model_mlp_1_w,x);
  1128. embeddings = ggml_silu_inplace(ctx0, embeddings);
  1129. embeddings = ggml_mul(ctx0, embeddings,x);
  1130. embeddings = ggml_mul_mat(ctx0, model.mm_model_mlp_3_w, embeddings);
  1131. }
  1132. } else {
  1133. GGML_ABORT("fatel error");
  1134. }
  1135. }
  1136. else if (ctx->proj_type == PROJECTOR_TYPE_MERGER) {
  1137. embeddings = ggml_reshape_3d(ctx0, embeddings, hidden_size * 4, num_positions / 4, batch_size);
  1138. embeddings = ggml_mul_mat(ctx0, model.mm_0_w, embeddings);
  1139. embeddings = ggml_add(ctx0, embeddings, model.mm_0_b);
  1140. // GELU activation
  1141. embeddings = ggml_gelu(ctx0, embeddings);
  1142. // Second linear layer
  1143. embeddings = ggml_mul_mat(ctx0, model.mm_1_w, embeddings);
  1144. embeddings = ggml_add(ctx0, embeddings, model.mm_1_b);
  1145. }
  1146. // build the graph
  1147. ggml_build_forward_expand(gf, embeddings);
  1148. ggml_free(ctx0);
  1149. return gf;
  1150. }
  1151. static ggml_cgraph * clip_image_build_graph(clip_ctx * ctx, const clip_image_f32_batch * imgs, struct clip_image_size * load_image_size, bool is_inf = false) {
  1152. if (ctx->proj_type == PROJECTOR_TYPE_GEMMA3) {
  1153. return clip_image_build_graph_siglip(ctx, imgs);
  1154. } else {
  1155. // TODO: we should have one build_* function per model
  1156. return clip_image_build_graph_legacy(ctx, imgs, load_image_size, is_inf);
  1157. }
  1158. }
  1159. // read and create ggml_context containing the tensors and their data
  1160. struct clip_ctx * clip_model_load(const char * fname, const int verbosity = 1) {
  1161. return clip_init(fname, clip_context_params{
  1162. /* use_gpu */ true,
  1163. /* verbosity */ verbosity,
  1164. });
  1165. }
  1166. struct clip_ctx * clip_init(const char * fname, struct clip_context_params ctx_params) {
  1167. int verbosity = ctx_params.verbosity;
  1168. struct ggml_context * meta = NULL;
  1169. struct gguf_init_params params = {
  1170. /*.no_alloc = */ true,
  1171. /*.ctx = */ &meta,
  1172. };
  1173. struct gguf_context * ctx = gguf_init_from_file(fname, params);
  1174. if (!ctx) {
  1175. throw std::runtime_error(format("%s: failed to load CLIP model from %s. Does this file exist?\n", __func__, fname));
  1176. }
  1177. if (verbosity >= 1) {
  1178. const int n_tensors = gguf_get_n_tensors(ctx);
  1179. const int n_kv = gguf_get_n_kv(ctx);
  1180. const int ftype = get_u32(ctx, KEY_FTYPE);
  1181. const std::string ftype_str = get_ftype(ftype);
  1182. const int idx_desc = get_key_idx(ctx, KEY_DESCRIPTION);
  1183. const std::string description = gguf_get_val_str(ctx, idx_desc);
  1184. const int idx_name = gguf_find_key(ctx, KEY_NAME);
  1185. if (idx_name != -1) { // make name optional temporarily as some of the uploaded models missing it due to a bug
  1186. const std::string name = gguf_get_val_str(ctx, idx_name);
  1187. LOG_INF("%s: model name: %s\n", __func__, name.c_str());
  1188. }
  1189. LOG_INF("%s: description: %s\n", __func__, description.c_str());
  1190. LOG_INF("%s: GGUF version: %d\n", __func__, gguf_get_version(ctx));
  1191. LOG_INF("%s: alignment: %zu\n", __func__, gguf_get_alignment(ctx));
  1192. LOG_INF("%s: n_tensors: %d\n", __func__, n_tensors);
  1193. LOG_INF("%s: n_kv: %d\n", __func__, n_kv);
  1194. LOG_INF("%s: ftype: %s\n", __func__, ftype_str.c_str());
  1195. LOG_INF("\n");
  1196. }
  1197. const int n_tensors = gguf_get_n_tensors(ctx);
  1198. // kv
  1199. const int n_kv = gguf_get_n_kv(ctx);
  1200. LOG_INF("%s: loaded meta data with %d key-value pairs and %d tensors from %s\n",
  1201. __func__, n_kv, n_tensors, fname);
  1202. {
  1203. std::map<enum ggml_type, uint32_t> n_type;
  1204. for (int i = 0; i < n_tensors; i++) {
  1205. enum ggml_type type = gguf_get_tensor_type(ctx, i);
  1206. n_type[type]++;
  1207. }
  1208. LOG_INF("%s: Dumping metadata keys/values. Note: KV overrides do not apply in this output.\n", __func__);
  1209. for (int i = 0; i < n_kv; i++) {
  1210. const char * name = gguf_get_key(ctx, i);
  1211. const enum gguf_type type = gguf_get_kv_type(ctx, i);
  1212. const std::string type_name =
  1213. type == GGUF_TYPE_ARRAY
  1214. ? format("%s[%s,%d]", gguf_type_name(type), gguf_type_name(gguf_get_arr_type(ctx, i)), gguf_get_arr_n(ctx, i))
  1215. : gguf_type_name(type);
  1216. std::string value = gguf_kv_to_str(ctx, i);
  1217. const size_t MAX_VALUE_LEN = 40;
  1218. if (value.size() > MAX_VALUE_LEN) {
  1219. value = format("%s...", value.substr(0, MAX_VALUE_LEN - 3).c_str());
  1220. }
  1221. replace_all(value, "\n", "\\n");
  1222. LOG_INF("%s: - kv %3d: %42s %-16s = %s\n", __func__, i, name, type_name.c_str(), value.c_str());
  1223. }
  1224. // print type counts
  1225. for (auto & kv : n_type) {
  1226. if (kv.second == 0) {
  1227. continue;
  1228. }
  1229. LOG_INF("%s: - type %4s: %4d tensors\n", __func__, ggml_type_name(kv.first), kv.second);
  1230. }
  1231. }
  1232. // data
  1233. size_t model_size = 0;
  1234. {
  1235. for (int i = 0; i < n_tensors; ++i) {
  1236. const char * name = gguf_get_tensor_name(ctx, i);
  1237. const size_t offset = gguf_get_tensor_offset(ctx, i);
  1238. enum ggml_type type = gguf_get_tensor_type(ctx, i);
  1239. struct ggml_tensor * cur = ggml_get_tensor(meta, name);
  1240. size_t tensor_size = ggml_nbytes(cur);
  1241. model_size += tensor_size;
  1242. if (verbosity >= 3) {
  1243. LOG_INF("%s: tensor[%d]: n_dims = %d, name = %s, tensor_size=%zu, offset=%zu, shape:[%" PRIu64 ", %" PRIu64 ", %" PRIu64 ", %" PRIu64 "], type = %s\n",
  1244. __func__, i, ggml_n_dims(cur), cur->name, tensor_size, offset, cur->ne[0], cur->ne[1], cur->ne[2], cur->ne[3], ggml_type_name(type));
  1245. }
  1246. }
  1247. }
  1248. clip_ctx * new_clip = new clip_ctx(ctx_params);
  1249. // update projector type
  1250. {
  1251. int idx = gguf_find_key(ctx, KEY_PROJ_TYPE);
  1252. if (idx != -1) {
  1253. const std::string proj_type = gguf_get_val_str(ctx, idx);
  1254. new_clip->proj_type = clip_projector_type_from_string(proj_type);
  1255. } else {
  1256. new_clip->proj_type = PROJECTOR_TYPE_MLP;
  1257. }
  1258. if (new_clip->proj_type == PROJECTOR_TYPE_MLP) {
  1259. if (gguf_find_tensor(ctx, format(TN_LLAVA_PROJ, 3, "weight").c_str()) != -1) {
  1260. new_clip->proj_type = PROJECTOR_TYPE_MLP_NORM;
  1261. }
  1262. }
  1263. }
  1264. // model size and capabilities
  1265. {
  1266. int idx = get_key_idx(ctx, KEY_HAS_TEXT_ENC);
  1267. new_clip->has_text_encoder = gguf_get_val_bool(ctx, idx);
  1268. idx = get_key_idx(ctx, KEY_HAS_VIS_ENC);
  1269. new_clip->has_vision_encoder = gguf_get_val_bool(ctx, idx);
  1270. idx = gguf_find_key(ctx, KEY_HAS_LLAVA_PROJ);
  1271. if (idx != -1) {
  1272. new_clip->has_llava_projector = gguf_get_val_bool(ctx, idx);
  1273. }
  1274. idx = gguf_find_key(ctx, KEY_HAS_MINICPMV_PROJ);
  1275. if (idx != -1) {
  1276. new_clip->has_minicpmv_projector = gguf_get_val_bool(ctx, idx);
  1277. }
  1278. idx = gguf_find_key(ctx, KEY_MINICPMV_VERSION);
  1279. if (idx != -1) {
  1280. new_clip->minicpmv_version = gguf_get_val_i32(ctx, idx);
  1281. }
  1282. idx = gguf_find_key(ctx, KEY_HAS_GLM_PROJ);
  1283. if (idx != -1) {
  1284. new_clip->has_glm_projector = gguf_get_val_bool(ctx, idx);
  1285. }
  1286. idx = gguf_find_key(ctx, KEY_HAS_QWEN2VL_MERGER);
  1287. if (idx != -1) {
  1288. new_clip->has_qwen2vl_merger = gguf_get_val_bool(ctx, idx);
  1289. }
  1290. // GGML_ASSERT(new_clip->has_llava_projector); // see monatis/clip.cpp for image and/or text encoding for semantic search
  1291. GGML_ASSERT(new_clip->has_vision_encoder);
  1292. GGML_ASSERT(!new_clip->has_text_encoder);
  1293. try {
  1294. idx = get_key_idx(ctx, KEY_USE_GELU);
  1295. new_clip->use_gelu = gguf_get_val_bool(ctx, idx);
  1296. } catch (std::runtime_error & /*e*/) {
  1297. new_clip->use_gelu = false;
  1298. }
  1299. try {
  1300. idx = get_key_idx(ctx, KEY_USE_SILU);
  1301. new_clip->use_silu = gguf_get_val_bool(ctx, idx);
  1302. } catch (std::runtime_error & /*e*/) {
  1303. new_clip->use_silu = false;
  1304. }
  1305. if (verbosity >= 1) {
  1306. LOG_INF("%s: text_encoder: %d\n", __func__, new_clip->has_text_encoder);
  1307. LOG_INF("%s: vision_encoder: %d\n", __func__, new_clip->has_vision_encoder);
  1308. LOG_INF("%s: llava_projector: %d\n", __func__, new_clip->has_llava_projector);
  1309. LOG_INF("%s: minicpmv_projector: %d\n", __func__, new_clip->has_minicpmv_projector);
  1310. LOG_INF("%s: minicpmv_version: %d\n", __func__, new_clip->minicpmv_version);
  1311. LOG_INF("%s: glm_projector: %d\n", __func__, new_clip->has_glm_projector);
  1312. LOG_INF("%s: model size: %.2f MB\n", __func__, model_size / 1024.0 / 1024.0);
  1313. LOG_INF("%s: metadata size: %.2f MB\n", __func__, ggml_get_mem_size(meta) / 1024.0 / 1024.0);
  1314. }
  1315. }
  1316. LOG_INF("%s: params backend buffer size = % 6.2f MB (%i tensors)\n", __func__, model_size / (1024.0 * 1024.0), n_tensors);
  1317. // load tensors
  1318. {
  1319. std::vector<uint8_t> read_buf;
  1320. struct ggml_init_params params = {
  1321. /*.mem_size =*/ (n_tensors + 1) * ggml_tensor_overhead(),
  1322. /*.mem_buffer =*/ NULL,
  1323. /*.no_alloc =*/ true,
  1324. };
  1325. new_clip->ctx_data = ggml_init(params);
  1326. if (!new_clip->ctx_data) {
  1327. LOG_ERR("%s: ggml_init() failed\n", __func__);
  1328. clip_free(new_clip);
  1329. gguf_free(ctx);
  1330. return nullptr;
  1331. }
  1332. auto fin = std::ifstream(fname, std::ios::binary);
  1333. if (!fin) {
  1334. LOG_ERR("cannot open model file for loading tensors\n");
  1335. clip_free(new_clip);
  1336. gguf_free(ctx);
  1337. return nullptr;
  1338. }
  1339. // add tensors to context
  1340. for (int i = 0; i < n_tensors; ++i) {
  1341. const char * name = gguf_get_tensor_name(ctx, i);
  1342. struct ggml_tensor * t = ggml_get_tensor(meta, name);
  1343. struct ggml_tensor * cur = ggml_dup_tensor(new_clip->ctx_data, t);
  1344. ggml_set_name(cur, name);
  1345. }
  1346. // alloc memory and offload data
  1347. ggml_backend_buffer_type_t buft = ggml_backend_get_default_buffer_type(new_clip->backend);
  1348. new_clip->buf = ggml_backend_alloc_ctx_tensors_from_buft(new_clip->ctx_data, buft);
  1349. ggml_backend_buffer_set_usage(new_clip->buf, GGML_BACKEND_BUFFER_USAGE_WEIGHTS);
  1350. for (int i = 0; i < n_tensors; ++i) {
  1351. const char * name = gguf_get_tensor_name(ctx, i);
  1352. struct ggml_tensor * cur = ggml_get_tensor(new_clip->ctx_data, name);
  1353. const size_t offset = gguf_get_data_offset(ctx) + gguf_get_tensor_offset(ctx, i);
  1354. fin.seekg(offset, std::ios::beg);
  1355. if (!fin) {
  1356. LOG_ERR("%s: failed to seek for tensor %s\n", __func__, name);
  1357. clip_free(new_clip);
  1358. gguf_free(ctx);
  1359. return nullptr;
  1360. }
  1361. int num_bytes = ggml_nbytes(cur);
  1362. if (ggml_backend_buft_is_host(buft)) {
  1363. // for the CPU and Metal backend, we can read directly into the tensor
  1364. fin.read(reinterpret_cast<char *>(cur->data), num_bytes);
  1365. } else {
  1366. // read into a temporary buffer first, then copy to device memory
  1367. read_buf.resize(num_bytes);
  1368. fin.read(reinterpret_cast<char *>(read_buf.data()), num_bytes);
  1369. ggml_backend_tensor_set(cur, read_buf.data(), 0, num_bytes);
  1370. }
  1371. }
  1372. fin.close();
  1373. }
  1374. // vision model
  1375. if (new_clip->has_vision_encoder) {
  1376. // load vision model
  1377. auto & vision_model = new_clip->vision_model;
  1378. auto & hparams = vision_model.hparams;
  1379. hparams.hidden_size = get_u32(ctx, format(KEY_N_EMBD, "vision"));
  1380. hparams.n_head = get_u32(ctx, format(KEY_N_HEAD, "vision"));
  1381. hparams.n_intermediate = get_u32(ctx, format(KEY_N_FF, "vision"));
  1382. hparams.n_layer = get_u32(ctx, format(KEY_N_BLOCK, "vision"));
  1383. hparams.image_size = get_u32(ctx, KEY_IMAGE_SIZE);
  1384. hparams.patch_size = get_u32(ctx, KEY_PATCH_SIZE);
  1385. hparams.projection_dim = get_u32(ctx, format(KEY_PROJ_DIM, "vision"));
  1386. hparams.eps = get_f32(ctx, format(KEY_LAYER_NORM_EPS, "vision"));
  1387. try {
  1388. int idx = get_key_idx(ctx, KEY_IMAGE_GRID_PINPOINTS);
  1389. int n = gguf_get_arr_n(ctx, idx);
  1390. const int32_t * pinpoints = (const int32_t *)gguf_get_arr_data(ctx, idx);
  1391. for (int i = 0; i < n; ++i) {
  1392. hparams.image_grid_pinpoints.push_back(pinpoints[i]);
  1393. }
  1394. } catch (std::runtime_error & /*e*/) { }
  1395. // Load the vision feature layer indices if they are explicitly provided;
  1396. // if multiple vision feature layers are present, the values will be concatenated
  1397. // to form the final visual features.
  1398. // NOTE: gguf conversions should standardize the values of the vision feature layer to
  1399. // be non-negative, since we use -1 to mark values as unset here.
  1400. try {
  1401. int idx = get_key_idx(ctx, KEY_FEATURE_LAYER);
  1402. int n = gguf_get_arr_n(ctx, idx);
  1403. const int32_t * vision_feature_layer = (const int32_t *)gguf_get_arr_data(ctx, idx);
  1404. for (int i = 0; i < n; ++i) {
  1405. hparams.vision_feature_layer.insert(vision_feature_layer[i]);
  1406. }
  1407. } catch (std::runtime_error & /*e*/) { }
  1408. try {
  1409. int idx = get_key_idx(ctx, KEY_MM_PATCH_MERGE_TYPE);
  1410. strcpy(hparams.mm_patch_merge_type, gguf_get_val_str(ctx, idx));
  1411. } catch (std::runtime_error & /*e*/) {
  1412. strcpy(hparams.mm_patch_merge_type, "flat");
  1413. }
  1414. try {
  1415. hparams.image_crop_resolution = get_u32(ctx, KEY_IMAGE_CROP_RESOLUTION); // llava-1.6
  1416. } catch(const std::exception& /*e*/) {
  1417. hparams.image_crop_resolution = hparams.image_size;
  1418. }
  1419. int idx_mean = get_key_idx(ctx, KEY_IMAGE_MEAN);
  1420. int idx_std = get_key_idx(ctx, KEY_IMAGE_STD);
  1421. const float * mean_data = (const float *)gguf_get_arr_data(ctx, idx_mean);
  1422. const float * std_data = (const float *)gguf_get_arr_data(ctx, idx_std);
  1423. for (int i = 0; i < 3; ++i) {
  1424. new_clip->image_mean[i] = mean_data[i];
  1425. new_clip->image_std[i] = std_data[i];
  1426. }
  1427. // Calculate the deepest feature layer based on hparams and projector type
  1428. new_clip->max_feature_layer = get_deepest_feature_layer(new_clip);
  1429. if (verbosity >= 2) {
  1430. LOG_INF("\n%s: vision model hparams\n", __func__);
  1431. LOG_INF("image_size %d\n", hparams.image_size);
  1432. LOG_INF("patch_size %d\n", hparams.patch_size);
  1433. LOG_INF("v_hidden_size %d\n", hparams.hidden_size);
  1434. LOG_INF("v_n_intermediate %d\n", hparams.n_intermediate);
  1435. LOG_INF("v_projection_dim %d\n", hparams.projection_dim);
  1436. LOG_INF("v_n_head %d\n", hparams.n_head);
  1437. LOG_INF("v_n_layer %d\n", hparams.n_layer);
  1438. LOG_INF("v_eps %f\n", hparams.eps);
  1439. LOG_INF("v_image_mean %f %f %f\n", new_clip->image_mean[0], new_clip->image_mean[1], new_clip->image_mean[2]);
  1440. LOG_INF("v_image_std %f %f %f\n", new_clip->image_std[0], new_clip->image_std[1], new_clip->image_std[2]);
  1441. LOG_INF("v_image_grid_pinpoints: ");
  1442. for (const auto & pp : hparams.image_grid_pinpoints) {
  1443. LOG_INF("%d ", pp);
  1444. }
  1445. LOG_INF("\n");
  1446. LOG_INF("v_vision_feature_layer: ");
  1447. for (const auto & feature_layer: hparams.vision_feature_layer) {
  1448. LOG_INF("%d ", feature_layer);
  1449. }
  1450. LOG_INF("\n");
  1451. LOG_INF("v_mm_patch_merge_type: %s\n", hparams.mm_patch_merge_type);
  1452. }
  1453. try {
  1454. vision_model.class_embedding = get_tensor(new_clip->ctx_data, TN_CLASS_EMBD);
  1455. new_clip->has_class_embedding = true;
  1456. } catch (const std::exception& /*e*/) {
  1457. new_clip->has_class_embedding = false;
  1458. }
  1459. try {
  1460. vision_model.pre_ln_w = get_tensor(new_clip->ctx_data, format(TN_LN_PRE, "v", "weight"));
  1461. vision_model.pre_ln_b = get_tensor(new_clip->ctx_data, format(TN_LN_PRE, "v", "bias"));
  1462. new_clip->has_pre_norm = true;
  1463. } catch (std::exception & /*e*/) {
  1464. new_clip->has_pre_norm = false;
  1465. }
  1466. try {
  1467. vision_model.post_ln_w = get_tensor(new_clip->ctx_data, format(TN_LN_POST, "v", "weight"));
  1468. vision_model.post_ln_b = get_tensor(new_clip->ctx_data, format(TN_LN_POST, "v", "bias"));
  1469. new_clip->has_post_norm = true;
  1470. } catch (std::exception & /*e*/) {
  1471. new_clip->has_post_norm = false;
  1472. }
  1473. try {
  1474. vision_model.patch_bias = get_tensor(new_clip->ctx_data, TN_PATCH_BIAS);
  1475. new_clip->has_patch_bias = true;
  1476. } catch (std::exception & /*e*/) {
  1477. new_clip->has_patch_bias = false;
  1478. }
  1479. try {
  1480. vision_model.patch_embeddings_0 = get_tensor(new_clip->ctx_data, TN_PATCH_EMBD);
  1481. } catch(const std::exception& /*e*/) {
  1482. vision_model.patch_embeddings_0 = nullptr;
  1483. }
  1484. try {
  1485. vision_model.position_embeddings = get_tensor(new_clip->ctx_data, format(TN_POS_EMBD, "v"));
  1486. } catch(const std::exception& /*e*/) {
  1487. vision_model.position_embeddings = nullptr;
  1488. }
  1489. try {
  1490. vision_model.patch_embeddings_1 = get_tensor(new_clip->ctx_data, TN_PATCH_EMBD_1);
  1491. } catch(const std::exception& /*e*/) {
  1492. new_clip->has_qwen2vl_merger = false;
  1493. }
  1494. // LLaVA projection
  1495. if (new_clip->proj_type == PROJECTOR_TYPE_MLP || new_clip->proj_type == PROJECTOR_TYPE_MLP_NORM) {
  1496. vision_model.mm_0_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 0, "weight"));
  1497. vision_model.mm_0_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 0, "bias"));
  1498. try {
  1499. // Yi-type llava
  1500. vision_model.mm_1_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 1, "weight"));
  1501. vision_model.mm_1_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 1, "bias"));
  1502. } catch (std::runtime_error & /*e*/) { }
  1503. try {
  1504. // missing in Yi-type llava
  1505. vision_model.mm_2_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 2, "weight"));
  1506. vision_model.mm_2_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 2, "bias"));
  1507. } catch (std::runtime_error & /*e*/) { }
  1508. try {
  1509. // Yi-type llava
  1510. vision_model.mm_3_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 3, "weight"));
  1511. vision_model.mm_3_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 3, "bias"));
  1512. } catch (std::runtime_error & /*e*/) { }
  1513. try {
  1514. // Yi-type llava
  1515. vision_model.mm_4_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 4, "weight"));
  1516. vision_model.mm_4_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 4, "bias"));
  1517. } catch (std::runtime_error & /*e*/) { }
  1518. try {
  1519. vision_model.image_newline = get_tensor(new_clip->ctx_data, TN_IMAGE_NEWLINE);
  1520. // LOG_INF("%s: image_newline tensor (llava-1.6) found\n", __func__);
  1521. } catch (std::runtime_error & /*e*/) { }
  1522. } else if (new_clip->proj_type == PROJECTOR_TYPE_LDP) {
  1523. // MobileVLM projection
  1524. vision_model.mm_model_mlp_1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 1, "weight"));
  1525. vision_model.mm_model_mlp_1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 1, "bias"));
  1526. vision_model.mm_model_mlp_3_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 3, "weight"));
  1527. vision_model.mm_model_mlp_3_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 3, "bias"));
  1528. vision_model.mm_model_block_1_block_0_0_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 0, "0.weight"));
  1529. vision_model.mm_model_block_1_block_0_1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 0, "1.weight"));
  1530. vision_model.mm_model_block_1_block_0_1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 0, "1.bias"));
  1531. vision_model.mm_model_block_1_block_1_fc1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 1, "fc1.weight"));
  1532. vision_model.mm_model_block_1_block_1_fc1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 1, "fc1.bias"));
  1533. vision_model.mm_model_block_1_block_1_fc2_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 1, "fc2.weight"));
  1534. vision_model.mm_model_block_1_block_1_fc2_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 1, "fc2.bias"));
  1535. vision_model.mm_model_block_1_block_2_0_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 2, "0.weight"));
  1536. vision_model.mm_model_block_1_block_2_1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 2, "1.weight"));
  1537. vision_model.mm_model_block_1_block_2_1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 1, 2, "1.bias"));
  1538. vision_model.mm_model_block_2_block_0_0_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 0, "0.weight"));
  1539. vision_model.mm_model_block_2_block_0_1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 0, "1.weight"));
  1540. vision_model.mm_model_block_2_block_0_1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 0, "1.bias"));
  1541. vision_model.mm_model_block_2_block_1_fc1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 1, "fc1.weight"));
  1542. vision_model.mm_model_block_2_block_1_fc1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 1, "fc1.bias"));
  1543. vision_model.mm_model_block_2_block_1_fc2_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 1, "fc2.weight"));
  1544. vision_model.mm_model_block_2_block_1_fc2_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 1, "fc2.bias"));
  1545. vision_model.mm_model_block_2_block_2_0_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 2, "0.weight"));
  1546. vision_model.mm_model_block_2_block_2_1_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 2, "1.weight"));
  1547. vision_model.mm_model_block_2_block_2_1_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_BLOCK, 2, 2, "1.bias"));
  1548. }
  1549. else if (new_clip->proj_type == PROJECTOR_TYPE_LDPV2)
  1550. {
  1551. // MobilVLM_V2 projection
  1552. vision_model.mm_model_mlp_0_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 0, "weight"));
  1553. vision_model.mm_model_mlp_0_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 0, "bias"));
  1554. vision_model.mm_model_mlp_2_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 2, "weight"));
  1555. vision_model.mm_model_mlp_2_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_MLP, 2, "bias"));
  1556. vision_model.mm_model_peg_0_w = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_PEG, 0, "weight"));
  1557. vision_model.mm_model_peg_0_b = get_tensor(new_clip->ctx_data, format(TN_MVLM_PROJ_PEG, 0, "bias"));
  1558. }
  1559. else if (new_clip->proj_type == PROJECTOR_TYPE_RESAMPLER) {
  1560. // vision_model.mm_model_pos_embed = get_tensor(new_clip->ctx_data, TN_MINICPMV_POS_EMBD);
  1561. vision_model.mm_model_pos_embed_k = get_tensor(new_clip->ctx_data, TN_MINICPMV_POS_EMBD_K);
  1562. vision_model.mm_model_query = get_tensor(new_clip->ctx_data, TN_MINICPMV_QUERY);
  1563. vision_model.mm_model_proj = get_tensor(new_clip->ctx_data, TN_MINICPMV_PROJ);
  1564. vision_model.mm_model_kv_proj = get_tensor(new_clip->ctx_data, TN_MINICPMV_KV_PROJ);
  1565. vision_model.mm_model_attn_q_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "q", "weight"));
  1566. vision_model.mm_model_attn_k_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "k", "weight"));
  1567. vision_model.mm_model_attn_v_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "v", "weight"));
  1568. vision_model.mm_model_attn_q_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "q", "bias"));
  1569. vision_model.mm_model_attn_k_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "k", "bias"));
  1570. vision_model.mm_model_attn_v_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "v", "bias"));
  1571. vision_model.mm_model_attn_o_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "out", "weight"));
  1572. vision_model.mm_model_attn_o_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_ATTN, "out", "bias"));
  1573. vision_model.mm_model_ln_q_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_LN, "q", "weight"));
  1574. vision_model.mm_model_ln_q_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_LN, "q", "bias"));
  1575. vision_model.mm_model_ln_kv_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_LN, "kv", "weight"));
  1576. vision_model.mm_model_ln_kv_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_LN, "kv", "bias"));
  1577. vision_model.mm_model_ln_post_w = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_LN, "post", "weight"));
  1578. vision_model.mm_model_ln_post_b = get_tensor(new_clip->ctx_data, format(TN_MINICPMV_LN, "post", "bias"));
  1579. }
  1580. else if (new_clip->proj_type == PROJECTOR_TYPE_GLM_EDGE) {
  1581. vision_model.mm_model_adapter_conv_w = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPER_CONV, "weight"));
  1582. vision_model.mm_model_adapter_conv_b = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPER_CONV, "bias"));
  1583. vision_model.mm_model_mlp_0_w = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPTER_LINEAR,"weight"));
  1584. vision_model.mm_model_ln_q_w = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPTER_NORM_1,"weight"));
  1585. vision_model.mm_model_ln_q_b = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPTER_NORM_1,"bias"));
  1586. vision_model.mm_model_mlp_1_w = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPTER_D_H_2_4H,"weight"));
  1587. vision_model.mm_model_mlp_2_w = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPTER_GATE,"weight"));
  1588. vision_model.mm_model_mlp_3_w = get_tensor(new_clip->ctx_data, format(TN_GLM_ADAPTER_D_4H_2_H,"weight"));
  1589. vision_model.boi_w = get_tensor(new_clip->ctx_data, TN_GLM_BOI_W);
  1590. vision_model.eoi_w = get_tensor(new_clip->ctx_data, TN_GLM_EOI_W);
  1591. }
  1592. else if (new_clip->proj_type == PROJECTOR_TYPE_MERGER) {
  1593. vision_model.mm_0_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 0, "weight"));
  1594. vision_model.mm_0_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 0, "bias"));
  1595. vision_model.mm_1_w = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 2, "weight"));
  1596. vision_model.mm_1_b = get_tensor(new_clip->ctx_data, format(TN_LLAVA_PROJ, 2, "bias"));
  1597. }
  1598. else if (new_clip->proj_type == PROJECTOR_TYPE_GEMMA3) {
  1599. vision_model.mm_input_proj_w = get_tensor(new_clip->ctx_data, TN_MM_INP_PROJ);
  1600. vision_model.mm_soft_emb_norm_w = get_tensor(new_clip->ctx_data, TN_MM_SOFT_EMB_N);
  1601. }
  1602. else {
  1603. std::string proj_type = PROJECTOR_TYPE_NAMES[new_clip->proj_type];
  1604. throw std::runtime_error(format("%s: don't support projector with: %s currently\n", __func__, proj_type.c_str()));
  1605. }
  1606. vision_model.layers.resize(hparams.n_layer);
  1607. for (int il = 0; il < hparams.n_layer; ++il) {
  1608. auto & layer = vision_model.layers[il];
  1609. layer.k_w = get_tensor(new_clip->ctx_data, format(TN_ATTN_K, "v", il, "weight"));
  1610. layer.q_w = get_tensor(new_clip->ctx_data, format(TN_ATTN_Q, "v", il, "weight"));
  1611. layer.v_w = get_tensor(new_clip->ctx_data, format(TN_ATTN_V, "v", il, "weight"));
  1612. layer.o_w = get_tensor(new_clip->ctx_data, format(TN_ATTN_OUTPUT, "v", il, "weight"));
  1613. layer.ln_1_w = get_tensor(new_clip->ctx_data, format(TN_LN_1, "v", il, "weight"));
  1614. layer.ln_2_w = get_tensor(new_clip->ctx_data, format(TN_LN_2, "v", il, "weight"));
  1615. layer.ff_i_w = get_tensor(new_clip->ctx_data, format(TN_FFN_DOWN, "v", il, "weight"));
  1616. layer.ff_o_w = get_tensor(new_clip->ctx_data, format(TN_FFN_UP, "v", il, "weight"));
  1617. layer.k_b = get_tensor(new_clip->ctx_data, format(TN_ATTN_K, "v", il, "bias"));
  1618. layer.q_b = get_tensor(new_clip->ctx_data, format(TN_ATTN_Q, "v", il, "bias"));
  1619. layer.v_b = get_tensor(new_clip->ctx_data, format(TN_ATTN_V, "v", il, "bias"));
  1620. layer.o_b = get_tensor(new_clip->ctx_data, format(TN_ATTN_OUTPUT, "v", il, "bias"));
  1621. layer.ln_1_b = get_tensor(new_clip->ctx_data, format(TN_LN_1, "v", il, "bias"));
  1622. layer.ln_2_b = get_tensor(new_clip->ctx_data, format(TN_LN_2, "v", il, "bias"));
  1623. layer.ff_i_b = get_tensor(new_clip->ctx_data, format(TN_FFN_DOWN, "v", il, "bias"));
  1624. layer.ff_o_b = get_tensor(new_clip->ctx_data, format(TN_FFN_UP, "v", il, "bias"));
  1625. }
  1626. }
  1627. ggml_free(meta);
  1628. new_clip->ctx_gguf = ctx;
  1629. // measure mem requirement and allocate
  1630. {
  1631. new_clip->buf_compute_meta.resize(GGML_DEFAULT_GRAPH_SIZE * ggml_tensor_overhead() + ggml_graph_overhead());
  1632. clip_image_f32_batch batch;
  1633. batch.size = 1;
  1634. batch.data = nullptr;
  1635. ggml_cgraph * gf = clip_image_build_graph(new_clip, &batch, nullptr, false);
  1636. ggml_backend_sched_reserve(new_clip->sched.get(), gf);
  1637. for (size_t i = 0; i < new_clip->backend_ptrs.size(); ++i) {
  1638. ggml_backend_t backend = new_clip->backend_ptrs[i];
  1639. ggml_backend_buffer_type_t buft = new_clip->backend_buft[i];
  1640. size_t size = ggml_backend_sched_get_buffer_size(new_clip->sched.get(), backend);
  1641. if (size > 1) {
  1642. LOG_INF("%s: %10s compute buffer size = %8.2f MiB\n", __func__,
  1643. ggml_backend_buft_name(buft),
  1644. size / 1024.0 / 1024.0);
  1645. }
  1646. }
  1647. }
  1648. return new_clip;
  1649. }
  1650. void clip_add_load_image_size(struct clip_ctx * ctx_clip, struct clip_image_size * load_image_size) {
  1651. ctx_clip->load_image_size = load_image_size;
  1652. }
  1653. struct clip_image_size * clip_get_load_image_size(struct clip_ctx * ctx_clip) {
  1654. return ctx_clip->load_image_size;
  1655. }
  1656. struct clip_image_size * clip_image_size_init() {
  1657. struct clip_image_size * load_image_size = new struct clip_image_size();
  1658. load_image_size->width = 448;
  1659. load_image_size->height = 448;
  1660. return load_image_size;
  1661. }
  1662. struct clip_image_u8 * clip_image_u8_init() {
  1663. return new clip_image_u8();
  1664. }
  1665. struct clip_image_f32 * clip_image_f32_init() {
  1666. return new clip_image_f32();
  1667. }
  1668. void clip_image_u8_free(struct clip_image_u8 * img) { delete img; }
  1669. void clip_image_f32_free(struct clip_image_f32 * img) { delete img; }
  1670. void clip_image_u8_batch_free(struct clip_image_u8_batch * batch) {
  1671. if (batch->size > 0) {
  1672. delete[] batch->data;
  1673. batch->size = 0;
  1674. }
  1675. }
  1676. void clip_image_f32_batch_free(struct clip_image_f32_batch * batch) {
  1677. if (batch->size > 0) {
  1678. delete[] batch->data;
  1679. batch->size = 0;
  1680. }
  1681. }
  1682. void clip_build_img_from_pixels(const unsigned char * rgb_pixels, int nx, int ny, clip_image_u8 * img) {
  1683. img->nx = nx;
  1684. img->ny = ny;
  1685. img->buf.resize(3 * nx * ny);
  1686. memcpy(img->buf.data(), rgb_pixels, img->buf.size());
  1687. }
  1688. bool clip_image_load_from_file(const char * fname, clip_image_u8 * img) {
  1689. int nx, ny, nc;
  1690. auto * data = stbi_load(fname, &nx, &ny, &nc, 3);
  1691. if (!data) {
  1692. LOG_ERR("%s: failed to load image '%s'\n", __func__, fname);
  1693. return false;
  1694. }
  1695. clip_build_img_from_pixels(data, nx, ny, img);
  1696. stbi_image_free(data);
  1697. return true;
  1698. }
  1699. bool clip_image_load_from_bytes(const unsigned char * bytes, size_t bytes_length, struct clip_image_u8 * img) {
  1700. int nx, ny, nc;
  1701. auto * data = stbi_load_from_memory(bytes, bytes_length, &nx, &ny, &nc, 3);
  1702. if (!data) {
  1703. LOG_ERR("%s: failed to decode image bytes\n", __func__);
  1704. return false;
  1705. }
  1706. clip_build_img_from_pixels(data, nx, ny, img);
  1707. stbi_image_free(data);
  1708. return true;
  1709. }
  1710. // Linear interpolation between two points
  1711. inline float clip_lerp(float s, float e, float t) {
  1712. return s + (e - s) * t;
  1713. }
  1714. // Bilinear resize function
  1715. static void bilinear_resize(const clip_image_u8& src, clip_image_u8& dst, int target_width, int target_height) {
  1716. dst.nx = target_width;
  1717. dst.ny = target_height;
  1718. dst.buf.resize(3 * target_width * target_height);
  1719. float x_ratio = static_cast<float>(src.nx - 1) / target_width;
  1720. float y_ratio = static_cast<float>(src.ny - 1) / target_height;
  1721. for (int y = 0; y < target_height; y++) {
  1722. for (int x = 0; x < target_width; x++) {
  1723. float px = x_ratio * x;
  1724. float py = y_ratio * y;
  1725. int x_floor = static_cast<int>(px);
  1726. int y_floor = static_cast<int>(py);
  1727. float x_lerp = px - x_floor;
  1728. float y_lerp = py - y_floor;
  1729. for (int c = 0; c < 3; c++) {
  1730. float top = clip_lerp(
  1731. static_cast<float>(src.buf[3 * (y_floor * src.nx + x_floor) + c]),
  1732. static_cast<float>(src.buf[3 * (y_floor * src.nx + (x_floor + 1)) + c]),
  1733. x_lerp
  1734. );
  1735. float bottom = clip_lerp(
  1736. static_cast<float>(src.buf[3 * ((y_floor + 1) * src.nx + x_floor) + c]),
  1737. static_cast<float>(src.buf[3 * ((y_floor + 1) * src.nx + (x_floor + 1)) + c]),
  1738. x_lerp
  1739. );
  1740. dst.buf[3 * (y * target_width + x) + c] = static_cast<uint8_t>(clip_lerp(top, bottom, y_lerp));
  1741. }
  1742. }
  1743. }
  1744. }
  1745. // Normalize image to float32 - careful with pytorch .to(model.device, dtype=torch.float16) - this sometimes reduces precision (32>16>32), sometimes not
  1746. static void normalize_image_u8_to_f32(const clip_image_u8* src, clip_image_f32* dst, const float mean[3], const float std[3]) {
  1747. dst->nx = src->nx;
  1748. dst->ny = src->ny;
  1749. dst->buf.resize(src->buf.size());
  1750. for (size_t i = 0; i < src->buf.size(); ++i) {
  1751. int c = i % 3; // rgb
  1752. dst->buf[i] = (static_cast<float>(src->buf[i]) / 255.0f - mean[c]) / std[c];
  1753. }
  1754. }
  1755. inline int clip(int x, int lower, int upper) {
  1756. return std::max(lower, std::min(x, upper));
  1757. }
  1758. static bool bicubic_resize(const clip_image_u8 &img, clip_image_u8 &dst, int target_width, int target_height) {
  1759. const int nx = img.nx;
  1760. const int ny = img.ny;
  1761. dst.nx = target_width;
  1762. dst.ny = target_height;
  1763. dst.buf.resize(3 * target_width * target_height);
  1764. float Cc;
  1765. float C[5];
  1766. float d0, d2, d3, a0, a1, a2, a3;
  1767. int i, j, k, jj;
  1768. int x, y;
  1769. float dx, dy;
  1770. float tx, ty;
  1771. tx = (float)nx / (float)target_width;
  1772. ty = (float)ny / (float)target_height;
  1773. // Bicubic interpolation; adapted from ViT.cpp, inspired from :
  1774. // -> https://github.com/yglukhov/bicubic-interpolation-image-processing/blob/master/libimage.c#L36
  1775. // -> https://en.wikipedia.org/wiki/Bicubic_interpolation
  1776. for (i = 0; i < target_height; i++) {
  1777. for (j = 0; j < target_width; j++) {
  1778. x = (int)(tx * j);
  1779. y = (int)(ty * i);
  1780. dx = tx * j - x;
  1781. dy = ty * i - y;
  1782. for (k = 0; k < 3; k++) {
  1783. for (jj = 0; jj <= 3; jj++) {
  1784. d0 = img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x - 1, 0, nx - 1)) * 3 + k] - img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x, 0, nx - 1)) * 3 + k];
  1785. d2 = img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x + 1, 0, nx - 1)) * 3 + k] - img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x, 0, nx - 1)) * 3 + k];
  1786. d3 = img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x + 2, 0, nx - 1)) * 3 + k] - img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x, 0, nx - 1)) * 3 + k];
  1787. a0 = img.buf[(clip(y - 1 + jj, 0, ny - 1) * nx + clip(x, 0, nx - 1)) * 3 + k];
  1788. a1 = -1.0 / 3 * d0 + d2 - 1.0 / 6 * d3;
  1789. a2 = 1.0 / 2 * d0 + 1.0 / 2 * d2;
  1790. a3 = -1.0 / 6 * d0 - 1.0 / 2 * d2 + 1.0 / 6 * d3;
  1791. C[jj] = a0 + a1 * dx + a2 * dx * dx + a3 * dx * dx * dx;
  1792. d0 = C[0] - C[1];
  1793. d2 = C[2] - C[1];
  1794. d3 = C[3] - C[1];
  1795. a0 = C[1];
  1796. a1 = -1.0 / 3 * d0 + d2 - 1.0 / 6 * d3;
  1797. a2 = 1.0 / 2 * d0 + 1.0 / 2 * d2;
  1798. a3 = -1.0 / 6 * d0 - 1.0 / 2 * d2 + 1.0 / 6 * d3;
  1799. Cc = a0 + a1 * dy + a2 * dy * dy + a3 * dy * dy * dy;
  1800. const uint8_t Cc2 = std::min(std::max(std::round(Cc), 0.0f), 255.0f);
  1801. dst.buf[(i * target_width + j) * 3 + k] = float(Cc2);
  1802. }
  1803. }
  1804. }
  1805. }
  1806. return true;
  1807. }
  1808. // llava-1.6 type of resize_and_pad (black)
  1809. static void resize_and_pad_image(const clip_image_u8& image, clip_image_u8 &image_output, const std::pair<int, int>& target_resolution) {
  1810. int target_width = target_resolution.first;
  1811. int target_height = target_resolution.second;
  1812. float scale_w = static_cast<float>(target_width) / image.nx;
  1813. float scale_h = static_cast<float>(target_height) / image.ny;
  1814. int new_width, new_height;
  1815. if (scale_w < scale_h) {
  1816. new_width = target_width;
  1817. new_height = std::min(static_cast<int>(std::ceil(image.ny * scale_w)), target_height);
  1818. } else {
  1819. new_height = target_height;
  1820. new_width = std::min(static_cast<int>(std::ceil(image.nx * scale_h)), target_width);
  1821. }
  1822. clip_image_u8 resized_image;
  1823. // bilinear_resize(image, resized_image, new_width, new_height);
  1824. bicubic_resize(image, resized_image, new_width, new_height);
  1825. clip_image_u8 padded_image;
  1826. padded_image.nx = target_width;
  1827. padded_image.ny = target_height;
  1828. padded_image.buf.resize(3 * target_width * target_height, 0); // Initialize with black
  1829. // Calculate padding offsets
  1830. int pad_x = (target_width - new_width) / 2;
  1831. int pad_y = (target_height - new_height) / 2;
  1832. // Copy the resized image into the center of the padded buffer
  1833. for (int y = 0; y < new_height; ++y) {
  1834. for (int x = 0; x < new_width; ++x) {
  1835. for (int c = 0; c < 3; ++c) {
  1836. padded_image.buf[3 * ((y + pad_y) * target_width + (x + pad_x)) + c] = resized_image.buf[3 * (y * new_width + x) + c];
  1837. }
  1838. }
  1839. }
  1840. image_output = std::move(padded_image);
  1841. }
  1842. /**
  1843. * Selects the best resolution from a list of possible resolutions based on the original size.
  1844. *
  1845. * @param original_size The original size of the image in the format (width, height).
  1846. * @param possible_resolutions A list of possible resolutions in the format [(width1, height1), (width2, height2), ...].
  1847. * @return The best fit resolution in the format (width, height).
  1848. */
  1849. static std::pair<int, int> select_best_resolution(const std::pair<int, int> & original_size, const std::vector<std::pair<int, int>> & possible_resolutions) {
  1850. int original_width = original_size.first;
  1851. int original_height = original_size.second;
  1852. std::pair<int, int> best_fit;
  1853. int max_effective_resolution = 0;
  1854. int min_wasted_resolution = std::numeric_limits<int>::max();
  1855. for (const auto& resolution : possible_resolutions) {
  1856. int width = resolution.first;
  1857. int height = resolution.second;
  1858. float scale = std::min(static_cast<float>(width) / original_width, static_cast<float>(height) / original_height);
  1859. int downscaled_width = static_cast<int>(original_width * scale);
  1860. int downscaled_height = static_cast<int>(original_height * scale);
  1861. int effective_resolution = std::min(downscaled_width * downscaled_height, original_width * original_height);
  1862. int wasted_resolution = (width * height) - effective_resolution;
  1863. // LOG_INF("resolution: %d %d, scale: %f, downscaled: %d %d, effective: %d, wasted: %d\n", width, height, scale, downscaled_width, downscaled_height, effective_resolution, wasted_resolution);
  1864. if (effective_resolution > max_effective_resolution || (effective_resolution == max_effective_resolution && wasted_resolution < min_wasted_resolution)) {
  1865. max_effective_resolution = effective_resolution;
  1866. min_wasted_resolution = wasted_resolution;
  1867. best_fit = resolution;
  1868. }
  1869. }
  1870. return best_fit;
  1871. }
  1872. static std::vector<clip_image_u8*> divide_to_patches_u8(const clip_image_u8 & image, int patch_size) {
  1873. std::vector<clip_image_u8*> patches;
  1874. int width = image.nx;
  1875. int height = image.ny;
  1876. for (int i = 0; i < height; i += patch_size) {
  1877. for (int j = 0; j < width; j += patch_size) {
  1878. clip_image_u8 *patch = clip_image_u8_init();
  1879. patch->nx = std::min(patch_size, width - j);
  1880. patch->ny = std::min(patch_size, height - i);
  1881. patch->buf.resize(3 * patch->nx * patch->ny);
  1882. for (int y = 0; y < patch->ny; ++y) {
  1883. for (int x = 0; x < patch->nx; ++x) {
  1884. for (int c = 0; c < 3; ++c) {
  1885. patch->buf[3 * (y * patch->nx + x) + c] = image.buf[3 * ((i + y) * width + (j + x)) + c];
  1886. }
  1887. }
  1888. }
  1889. patches.push_back(patch);
  1890. }
  1891. }
  1892. return patches;
  1893. }
  1894. static int ensure_divide(int length, int patch_size) {
  1895. return std::max(static_cast<int>(std::round(static_cast<float>(length) / patch_size) * patch_size), patch_size);
  1896. }
  1897. static std::pair<int, int> uhd_find_best_resize(std::pair<int, int> original_size, int scale_resolution, int patch_size, bool allow_upscale = false) {
  1898. int width = original_size.first;
  1899. int height = original_size.second;
  1900. if ((width * height > scale_resolution * scale_resolution) || allow_upscale) {
  1901. float r = static_cast<float>(width) / height;
  1902. height = static_cast<int>(scale_resolution / std::sqrt(r));
  1903. width = static_cast<int>(height * r);
  1904. }
  1905. int best_width = ensure_divide(width, patch_size);
  1906. int best_height = ensure_divide(height, patch_size);
  1907. return std::make_pair(best_width, best_height);
  1908. }
  1909. static std::pair<int, int> uhd_get_refine_size(std::pair<int, int> original_size, std::pair<int, int> grid, int scale_resolution, int patch_size, bool allow_upscale = false) {
  1910. int width, height;
  1911. std::tie(width, height) = original_size;
  1912. int grid_x, grid_y;
  1913. std::tie(grid_x, grid_y) = grid;
  1914. int refine_width = ensure_divide(width, grid_x);
  1915. int refine_height = ensure_divide(height, grid_y);
  1916. int grid_width = refine_width / grid_x;
  1917. int grid_height = refine_height / grid_y;
  1918. // auto best_grid_size = find_best_resize(std::make_tuple(grid_width, grid_height), scale_resolution, patch_size, allow_upscale); (old line)
  1919. auto best_grid_size = uhd_find_best_resize(std::make_pair(grid_width, grid_height), scale_resolution, patch_size, allow_upscale); // (new line) => fixes conversion for make_tuple to make_pair
  1920. int best_grid_width, best_grid_height;
  1921. std::tie(best_grid_width, best_grid_height) = best_grid_size;
  1922. // std::pair<int, int> refine_size = std::make_tuple(best_grid_width * grid_x, best_grid_height * grid_y); (old line)
  1923. std::pair<int, int> refine_size = std::make_pair(best_grid_width * grid_x, best_grid_height * grid_y); // (new line)
  1924. return refine_size;
  1925. }
  1926. static std::pair<int, int> uhd_best_grid(const int max_slice_nums, const int multiple, const float log_ratio) {
  1927. std::vector<int> candidate_split_grids_nums;
  1928. for (int i : {multiple - 1, multiple, multiple + 1}) {
  1929. if (i == 1 || i > max_slice_nums) {
  1930. continue;
  1931. }
  1932. candidate_split_grids_nums.push_back(i);
  1933. }
  1934. std::vector<std::pair<int, int>> candidate_grids;
  1935. for (int split_grids_nums : candidate_split_grids_nums) {
  1936. int m = 1;
  1937. while (m <= split_grids_nums) {
  1938. if (split_grids_nums % m == 0) {
  1939. candidate_grids.emplace_back(m, split_grids_nums / m);
  1940. }
  1941. ++m;
  1942. }
  1943. }
  1944. std::pair<int, int> best_grid{1, 1};
  1945. float min_error = std::numeric_limits<float>::infinity();
  1946. for (const auto& grid : candidate_grids) {
  1947. float error = std::abs(log_ratio - std::log(1.0 * grid.first / grid.second));
  1948. if (error < min_error) {
  1949. best_grid = grid;
  1950. min_error = error;
  1951. }
  1952. }
  1953. return best_grid;
  1954. }
  1955. // inspired from LLaVA-UHD:
  1956. // -> https://arxiv.org/pdf/2403.11703
  1957. // -> https://github.com/thunlp/LLaVA-UHD
  1958. // -> https://github.com/thunlp/LLaVA-UHD/blob/302301bc2175f7e717fb8548516188e89f649753/llava_uhd/train/llava-uhd/slice_logic.py#L118
  1959. static std::vector<std::vector<clip_image_u8 *>> uhd_slice_image(const clip_image_u8 * img, const int max_slice_nums=9, const int scale_resolution=448, const int patch_size=14) {
  1960. const std::pair<int, int> original_size={img->nx,img->ny};
  1961. const int original_width = img->nx;
  1962. const int original_height = img->ny;
  1963. const float log_ratio = log(1.0*original_width/original_height);
  1964. const float ratio = 1.0 * original_width * original_height/ (scale_resolution * scale_resolution);
  1965. const int multiple = fmin(ceil(ratio), max_slice_nums);
  1966. std::vector<std::vector<clip_image_u8 *>> images;
  1967. LOG_INF("%s: multiple %d\n", __func__, multiple);
  1968. images.push_back(std::vector<clip_image_u8 *>());
  1969. if (multiple <= 1) {
  1970. auto best_size = uhd_find_best_resize(original_size, scale_resolution, patch_size, true);
  1971. clip_image_u8 * source_image = clip_image_u8_init();
  1972. bicubic_resize(*img, *source_image, best_size.first, best_size.second);
  1973. // source_image = image.resize(best_size, Image.Resampling.BICUBIC)
  1974. images[images.size()-1].push_back(source_image);
  1975. }
  1976. else if (multiple > 1) {
  1977. auto best_size = uhd_find_best_resize(original_size, scale_resolution, patch_size);
  1978. clip_image_u8 * source_image = clip_image_u8_init();
  1979. bicubic_resize(*img, *source_image, best_size.first, best_size.second);
  1980. // source_image = image.copy().resize(best_resize, Image.Resampling.BICUBIC)
  1981. LOG_INF("%s: image_size: %d %d; source_image size: %d %d\n", __func__, img->nx, img->ny, best_size.first, best_size.second);
  1982. images[images.size()-1].push_back(source_image);
  1983. std::pair<int, int> best_grid = uhd_best_grid(max_slice_nums, multiple, log_ratio);
  1984. LOG_INF("%s: image_size: %d %d; best_grid: %d %d\n", __func__, img->nx, img->ny, best_grid.first, best_grid.second);
  1985. auto refine_size = uhd_get_refine_size(original_size, best_grid, scale_resolution, patch_size, true);
  1986. clip_image_u8 * refine_image = clip_image_u8_init();
  1987. bicubic_resize(*img, *refine_image, refine_size.first, refine_size.second);
  1988. LOG_INF("%s: refine_image_size: %d %d; refine_size: %d %d\n", __func__, refine_image->nx, refine_image->ny, refine_size.first, refine_size.second);
  1989. // split_to_patches
  1990. int width = refine_image->nx;
  1991. int height = refine_image->ny;
  1992. int grid_x = int(width / best_grid.first);
  1993. int grid_y = int(height / best_grid.second);
  1994. for (int patches_i = 0, ic = 0; patches_i < height && ic < best_grid.second; patches_i += grid_y, ic += 1){
  1995. images.push_back(std::vector<clip_image_u8 *>());
  1996. for(int patches_j = 0, jc = 0; patches_j < width && jc < best_grid.first; patches_j += grid_x, jc += 1){
  1997. clip_image_u8 * patch = clip_image_u8_init();
  1998. patch->nx = grid_x;
  1999. patch->ny = grid_y;
  2000. patch->buf.resize(3 * patch->nx * patch->ny);
  2001. for (int y = patches_i; y < patches_i + grid_y; ++y) {
  2002. for (int x = patches_j; x < patches_j + grid_x; ++x) {
  2003. const int i = 3 * (y * refine_image->nx + x);
  2004. const int j = 3 * ((y-patches_i) * patch->nx + (x-patches_j));
  2005. patch->buf[j] = refine_image->buf[i];
  2006. patch->buf[j+1] = refine_image->buf[i+1];
  2007. patch->buf[j+2] = refine_image->buf[i+2];
  2008. }
  2009. }
  2010. images[images.size()-1].push_back(patch);
  2011. }
  2012. }
  2013. clip_image_u8_free(refine_image);
  2014. }
  2015. return images;
  2016. }
  2017. int clip_uhd_num_image_embeds_col(struct clip_ctx * ctx_clip) {
  2018. const int max_slice_nums=9;
  2019. const int scale_resolution=448;
  2020. const int original_width = ctx_clip->load_image_size->width;
  2021. const int original_height = ctx_clip->load_image_size->height;
  2022. const float log_ratio = log(1.0*original_width/original_height);
  2023. const float ratio = 1.0 * original_width * original_height/ (scale_resolution * scale_resolution);
  2024. const int multiple = fmin(ceil(ratio), max_slice_nums);
  2025. std::pair<int, int> best_grid = uhd_best_grid(max_slice_nums, multiple, log_ratio);
  2026. return best_grid.first;
  2027. }
  2028. // returns the normalized float tensor for llava-1.5, for spatial_unpad with anyres processing for llava-1.6 it returns the normalized image patch tensors as a vector
  2029. // res_imgs memory is being allocated here, previous allocations will be freed if found
  2030. bool clip_image_preprocess(struct clip_ctx * ctx, const clip_image_u8 * img, clip_image_f32_batch * res_imgs) {
  2031. if(clip_is_minicpmv(ctx)){
  2032. int max_slice_nums = 9;
  2033. std::vector<std::vector<clip_image_u8 *>> imgs = uhd_slice_image(img, max_slice_nums);
  2034. res_imgs->size = 0;
  2035. for (size_t i = 0; i < imgs.size(); ++i){
  2036. res_imgs->size += imgs[i].size();
  2037. }
  2038. res_imgs->data = new clip_image_f32[res_imgs->size];
  2039. int idx = 0;
  2040. for (size_t i = 0; i < imgs.size(); ++i) {
  2041. for (size_t j = 0; j < imgs[i].size(); ++j) {
  2042. LOG_DBG("%s: %d %d\n", __func__,imgs[i][j]->nx,imgs[i][j]->ny);
  2043. clip_image_f32 * res = clip_image_f32_init();
  2044. normalize_image_u8_to_f32(imgs[i][j], res, ctx->image_mean, ctx->image_std);
  2045. res_imgs->data[idx++] = *res;
  2046. clip_image_f32_free(res);
  2047. }
  2048. }
  2049. for (size_t i = 0; i < imgs.size(); ++i) {
  2050. for (size_t j = 0; j < imgs[i].size(); ++j) {
  2051. if (imgs[i][j] != nullptr) {
  2052. clip_image_u8_free(imgs[i][j]);
  2053. }
  2054. }
  2055. }
  2056. return true;
  2057. }
  2058. else if (ctx->has_qwen2vl_merger) {
  2059. clip_image_u8 * resized = clip_image_u8_init();
  2060. auto patch_size = clip_patch_size(ctx) * 2;
  2061. int nx = ceil((float)img->nx / patch_size) * patch_size;
  2062. int ny = ceil((float)img->ny / patch_size) * patch_size;
  2063. bicubic_resize(*img, *resized, nx, ny);
  2064. res_imgs->data = new clip_image_f32[1];
  2065. // clip_image_f32 * res = clip_image_f32_init();
  2066. normalize_image_u8_to_f32(resized, res_imgs->data, ctx->image_mean, ctx->image_std);
  2067. // res_imgs->data[0] = *res;
  2068. res_imgs->size = 1;
  2069. // clip_image_f32_free(res);
  2070. clip_image_u8_free(resized);
  2071. return true;
  2072. }
  2073. if (ctx->has_glm_projector || ctx->proj_type == PROJECTOR_TYPE_GEMMA3) {
  2074. res_imgs->size = 1;
  2075. res_imgs->data = new clip_image_f32[res_imgs->size];
  2076. clip_image_u8 resized_image;
  2077. int32_t sz=ctx->vision_model.hparams.image_size;
  2078. bicubic_resize(*img, resized_image,sz,sz);
  2079. clip_image_f32 * res = clip_image_f32_init();
  2080. //clip_image_save_to_bmp(resized_image, "resized.bmp");
  2081. normalize_image_u8_to_f32(&resized_image, res, ctx->image_mean, ctx->image_std);
  2082. res_imgs->data[0] = *res;
  2083. clip_image_f32_free(res);
  2084. return true;
  2085. }
  2086. bool pad_to_square = true;
  2087. if (!ctx->has_vision_encoder) {
  2088. LOG_ERR("This gguf file seems to have no vision encoder\n");
  2089. return false;
  2090. }
  2091. auto & params = ctx->vision_model.hparams;
  2092. // The model config actually contains all we need to decide on how to preprocess, here we automatically switch to the new llava-1.6 preprocessing
  2093. if (strcmp(params.mm_patch_merge_type, "spatial_unpad") == 0) {
  2094. pad_to_square = false;
  2095. }
  2096. // free the previous res_imgs if any set
  2097. if (res_imgs->size > 0) {
  2098. clip_image_f32_batch_free(res_imgs);
  2099. }
  2100. res_imgs->data = nullptr;
  2101. res_imgs->size = 0;
  2102. // the logic below is to pad the shorter side to the longer side with a background color: rgb(122, 116, 104)
  2103. // see https://github.com/haotian-liu/LLaVA/blob/e854a2bf85118c504f6f16bf5c3c7c92f8fa8c6b/llava/conversation.py#L113-L156
  2104. clip_image_u8 * temp = clip_image_u8_init(); // we will keep the input image data here temporarily
  2105. if (pad_to_square && img->nx != img->ny) {
  2106. int longer_side = std::max(img->nx, img->ny);
  2107. temp->nx = longer_side;
  2108. temp->ny = longer_side;
  2109. temp->buf.resize(3 * longer_side * longer_side);
  2110. const uint8_t bc[3] = {122, 116, 104}; // background color in RGB from LLaVA (this is the mean rgb color * 255)
  2111. // fill with background color
  2112. for (size_t i = 0; i < temp->buf.size(); i++) {
  2113. temp->buf[i] = bc[i % 3];
  2114. }
  2115. // copy from the input image
  2116. for (int y = 0; y < img->ny; y++) {
  2117. for (int x = 0; x < img->nx; x++) {
  2118. const int i = 3 * (y * img->nx + x);
  2119. const int j = 3 * (y * temp->nx + x);
  2120. temp->buf[j] = img->buf[i];
  2121. temp->buf[j+1] = img->buf[i+1];
  2122. temp->buf[j+2] = img->buf[i+2];
  2123. }
  2124. }
  2125. } else {
  2126. if (!params.image_grid_pinpoints.empty()) {
  2127. // "spatial_unpad" with "anyres" processing for llava-1.6
  2128. std::vector<std::pair<int, int>> possible_resolutions;
  2129. for (size_t i = 0; i < params.image_grid_pinpoints.size(); i+=2) {
  2130. possible_resolutions.push_back({params.image_grid_pinpoints[i], params.image_grid_pinpoints[i+1]});
  2131. }
  2132. std::pair<int, int> best_resolution = select_best_resolution({img->nx, img->ny}, possible_resolutions);
  2133. // clip_image_save_to_bmp(*img, "input.bmp");
  2134. resize_and_pad_image(*img, *temp, best_resolution); // we do not pad with mean-bg color anymore in llava-1.6
  2135. // clip_image_save_to_bmp(*temp, "resized.bmp");
  2136. // visually verify normalized image:
  2137. // normalize_image_u8_to_f32(*temp, *res, ctx->image_mean, ctx->image_std);
  2138. // {
  2139. // clip_image_u8 * temp2 = clip_image_u8_init();
  2140. // clip_image_convert_f32_to_u8(*res, *temp2);
  2141. // clip_image_save_to_bmp(*temp2, "resized_normalized_f32.bmp");
  2142. // clip_image_u8_free(temp2);
  2143. // }
  2144. std::vector<clip_image_u8 *> patches = divide_to_patches_u8(*temp, params.image_size); // prepare spatial sorted main patches of image_size each (336 in llava-1.6)
  2145. clip_image_u8 *image_original_resize = clip_image_u8_init();
  2146. // bilinear_resize(*img, *image_original_resize, params.image_size, params.image_size); // in python this is "shortest_edge", but all CLIP are square
  2147. bicubic_resize(*img, *image_original_resize, params.image_size, params.image_size); // in python this is "shortest_edge", but all CLIP are square
  2148. patches.insert(patches.begin(), image_original_resize);
  2149. // clip_image_f32_batch_init(patches.size());
  2150. res_imgs->size = patches.size();
  2151. res_imgs->data = new clip_image_f32[res_imgs->size];
  2152. int num=0;
  2153. for (auto& patch : patches) {
  2154. normalize_image_u8_to_f32(patch, &res_imgs->data[num], ctx->image_mean, ctx->image_std);
  2155. num++;
  2156. }
  2157. for (size_t i = 0; i < patches.size(); i++) {
  2158. // LOG_DBG("patch %d: %d %d\n", i, patches[i]->nx, patches[i]->ny);
  2159. clip_image_u8_free(patches[i]);
  2160. }
  2161. clip_image_u8_free(temp);
  2162. return true;
  2163. } else {
  2164. temp->nx = img->nx;
  2165. temp->ny = img->ny;
  2166. temp->buf.resize(img->buf.size());
  2167. memcpy(temp->buf.data(), img->buf.data(), temp->buf.size());
  2168. }
  2169. }
  2170. const int nx = temp->nx;
  2171. const int ny = temp->ny;
  2172. // clip_image_save_to_bmp(*temp, "resized_vanilla.bmp");
  2173. const int nx2 = ctx->vision_model.hparams.image_size;
  2174. const int ny2 = ctx->vision_model.hparams.image_size;
  2175. clip_image_f32 * res = clip_image_f32_init();
  2176. res->nx = nx2;
  2177. res->ny = ny2;
  2178. res->buf.resize(3 * nx2 * ny2);
  2179. const float scale = std::max(nx, ny) / (float)ctx->vision_model.hparams.image_size;
  2180. const int nx3 = int(nx / scale + 0.5f);
  2181. const int ny3 = int(ny / scale + 0.5f);
  2182. const auto & m3 = ctx->image_mean; // {0.48145466f, 0.4578275f, 0.40821073f};
  2183. const auto & s3 = ctx->image_std; // {0.26862954f, 0.26130258f, 0.27577711f};
  2184. for (int y = 0; y < ny3; y++) {
  2185. for (int x = 0; x < nx3; x++) {
  2186. for (int c = 0; c < 3; c++) {
  2187. // linear interpolation
  2188. const float sx = (x + 0.5f) * scale - 0.5f;
  2189. const float sy = (y + 0.5f) * scale - 0.5f;
  2190. const int x0 = std::max(0, (int)std::floor(sx));
  2191. const int y0 = std::max(0, (int)std::floor(sy));
  2192. const int x1 = std::min(x0 + 1, nx - 1);
  2193. const int y1 = std::min(y0 + 1, ny - 1);
  2194. const float dx = sx - x0;
  2195. const float dy = sy - y0;
  2196. const int j00 = 3 * (y0 * nx + x0) + c;
  2197. const int j01 = 3 * (y0 * nx + x1) + c;
  2198. const int j10 = 3 * (y1 * nx + x0) + c;
  2199. const int j11 = 3 * (y1 * nx + x1) + c;
  2200. const float v00 = temp->buf[j00];
  2201. const float v01 = temp->buf[j01];
  2202. const float v10 = temp->buf[j10];
  2203. const float v11 = temp->buf[j11];
  2204. const float v0 = v00 * (1.0f - dx) + v01 * dx;
  2205. const float v1 = v10 * (1.0f - dx) + v11 * dx;
  2206. const float v = v0 * (1.0f - dy) + v1 * dy;
  2207. const uint8_t v2 = std::min(std::max(std::round(v), 0.0f), 255.0f);
  2208. const int i = 3 * (y * nx3 + x) + c;
  2209. res->buf[i] = ((float(v2) / 255.0f) - m3[c]) / s3[c];
  2210. }
  2211. }
  2212. }
  2213. clip_image_u8_free(temp);
  2214. // {
  2215. // clip_image_u8 * temp2 = clip_image_u8_init();
  2216. // clip_image_convert_f32_to_u8(*res, *temp2);
  2217. // clip_image_save_to_bmp(*temp2, "resized_normalized_f32_vanilla.bmp");
  2218. // clip_image_u8_free(temp2);
  2219. // }
  2220. // res_imgs.push_back(res);
  2221. res_imgs->size = 1;
  2222. res_imgs->data = new clip_image_f32[res_imgs->size];
  2223. res_imgs->data[0] = *res;
  2224. clip_image_f32_free(res);
  2225. return true;
  2226. }
  2227. ggml_tensor * clip_get_newline_tensor(const struct clip_ctx * ctx) {
  2228. return ctx->vision_model.image_newline;
  2229. }
  2230. void clip_free(clip_ctx * ctx) {
  2231. delete ctx;
  2232. }
  2233. size_t clip_embd_nbytes(const struct clip_ctx * ctx) {
  2234. int extra_tokens = ctx->has_glm_projector ? 2 : 0;
  2235. return (clip_n_patches(ctx) + extra_tokens) * clip_n_mmproj_embd(ctx) * sizeof(float);
  2236. }
  2237. size_t clip_embd_nbytes_by_img(const struct clip_ctx * ctx, int img_h, int img_w) {
  2238. clip_image_f32 img;
  2239. img.nx = img_w;
  2240. img.ny = img_h;
  2241. return clip_n_patches_by_img(ctx, &img) * clip_n_mmproj_embd(ctx) * sizeof(float);
  2242. }
  2243. int32_t clip_image_size(const struct clip_ctx * ctx) {
  2244. return ctx->vision_model.hparams.image_size;
  2245. }
  2246. int32_t clip_patch_size(const struct clip_ctx * ctx) {
  2247. return ctx->vision_model.hparams.patch_size;
  2248. }
  2249. int32_t clip_hidden_size(const struct clip_ctx * ctx) {
  2250. return ctx->vision_model.hparams.hidden_size;
  2251. }
  2252. const char * clip_patch_merge_type(const struct clip_ctx * ctx) {
  2253. return ctx->vision_model.hparams.mm_patch_merge_type;
  2254. }
  2255. const int32_t * clip_image_grid(const struct clip_ctx * ctx) {
  2256. if (ctx->vision_model.hparams.image_grid_pinpoints.size()) {
  2257. return &ctx->vision_model.hparams.image_grid_pinpoints.front();
  2258. }
  2259. return nullptr;
  2260. }
  2261. size_t get_clip_image_grid_size(const struct clip_ctx * ctx) {
  2262. return ctx->vision_model.hparams.image_grid_pinpoints.size();
  2263. }
  2264. int clip_n_patches(const struct clip_ctx * ctx) {
  2265. clip_image_f32 img;
  2266. img.nx = ctx->vision_model.hparams.image_size;
  2267. img.ny = ctx->vision_model.hparams.image_size;
  2268. return clip_n_patches_by_img(ctx, &img);
  2269. }
  2270. int clip_n_patches_by_img(const struct clip_ctx * ctx, struct clip_image_f32 * img) {
  2271. const auto & params = ctx->vision_model.hparams;
  2272. int n_patches = (params.image_size / params.patch_size) * (params.image_size / params.patch_size);
  2273. if (ctx->proj_type == PROJECTOR_TYPE_LDP || ctx->proj_type == PROJECTOR_TYPE_LDPV2 || ctx->proj_type == PROJECTOR_TYPE_GLM_EDGE) {
  2274. n_patches /= 4;
  2275. } else if (ctx->proj_type == PROJECTOR_TYPE_RESAMPLER) {
  2276. if (ctx->minicpmv_version == 2) {
  2277. n_patches = 96;
  2278. }
  2279. else if (ctx->minicpmv_version == 3) {
  2280. n_patches = 64;
  2281. }
  2282. else if (ctx->minicpmv_version == 4) {
  2283. n_patches = 64;
  2284. }
  2285. } else if (ctx->proj_type == PROJECTOR_TYPE_MERGER) {
  2286. int patch_size = params.patch_size * 2;
  2287. int x_patch = img->nx / patch_size + (int)(img->nx % patch_size > 0);
  2288. int y_patch = img->ny / patch_size + (int)(img->ny % patch_size > 0);
  2289. n_patches = x_patch * y_patch;
  2290. }
  2291. return n_patches;
  2292. }
  2293. static std::vector<std::vector<std::vector<float>>> get_1d_sincos_pos_embed_from_grid_new(int embed_dim, const std::vector<std::vector<float>> & pos) {
  2294. assert(embed_dim % 2 == 0);
  2295. int H = pos.size();
  2296. int W = pos[0].size();
  2297. std::vector<float> omega(embed_dim / 2);
  2298. for (int i = 0; i < embed_dim / 2; ++i) {
  2299. omega[i] = 1.0 / pow(10000.0, static_cast<float>(i) / (embed_dim / 2));
  2300. }
  2301. std::vector<std::vector<std::vector<float>>> emb(H, std::vector<std::vector<float>>(W, std::vector<float>(embed_dim)));
  2302. for (int h = 0; h < H; ++h) {
  2303. for (int w = 0; w < W; ++w) {
  2304. for (int d = 0; d < embed_dim / 2; ++d) {
  2305. float out_value = pos[h][w] * omega[d];
  2306. emb[h][w][d] = sin(out_value);
  2307. emb[h][w][d + embed_dim / 2] = cos(out_value);
  2308. }
  2309. }
  2310. }
  2311. return emb;
  2312. }
  2313. static std::vector<std::vector<std::vector<float>>> get_2d_sincos_pos_embed_from_grid(int embed_dim, const std::vector<std::vector<std::vector<float>>> & grid) {
  2314. assert(embed_dim % 2 == 0);
  2315. std::vector<std::vector<std::vector<float>>> emb_h = get_1d_sincos_pos_embed_from_grid_new(embed_dim / 2, grid[0]); // (H, W, D/2)
  2316. std::vector<std::vector<std::vector<float>>> emb_w = get_1d_sincos_pos_embed_from_grid_new(embed_dim / 2, grid[1]); // (H, W, D/2)
  2317. int H = emb_h.size();
  2318. int W = emb_h[0].size();
  2319. std::vector<std::vector<std::vector<float>>> emb(H, std::vector<std::vector<float>>(W, std::vector<float>(embed_dim)));
  2320. for (int h = 0; h < H; ++h) {
  2321. for (int w = 0; w < W; ++w) {
  2322. for (int d = 0; d < embed_dim / 2; ++d) {
  2323. emb[h][w][d] = emb_h[h][w][d];
  2324. emb[h][w][d + embed_dim / 2] = emb_w[h][w][d];
  2325. }
  2326. }
  2327. }
  2328. return emb;
  2329. }
  2330. static std::vector<std::vector<float>> get_2d_sincos_pos_embed(int embed_dim, const std::pair<int, int> image_size) {
  2331. int grid_h_size = image_size.first;
  2332. int grid_w_size = image_size.second;
  2333. std::vector<float> grid_h(grid_h_size);
  2334. std::vector<float> grid_w(grid_w_size);
  2335. for (int i = 0; i < grid_h_size; ++i) {
  2336. grid_h[i] = static_cast<float>(i);
  2337. }
  2338. for (int i = 0; i < grid_w_size; ++i) {
  2339. grid_w[i] = static_cast<float>(i);
  2340. }
  2341. std::vector<std::vector<float>> grid(grid_h_size, std::vector<float>(grid_w_size));
  2342. for (int h = 0; h < grid_h_size; ++h) {
  2343. for (int w = 0; w < grid_w_size; ++w) {
  2344. grid[h][w] = grid_w[w];
  2345. }
  2346. }
  2347. std::vector<std::vector<std::vector<float>>> grid_2d = {grid, grid};
  2348. for (int h = 0; h < grid_h_size; ++h) {
  2349. for (int w = 0; w < grid_w_size; ++w) {
  2350. grid_2d[0][h][w] = grid_h[h];
  2351. grid_2d[1][h][w] = grid_w[w];
  2352. }
  2353. }
  2354. std::vector<std::vector<std::vector<float>>> pos_embed_3d = get_2d_sincos_pos_embed_from_grid(embed_dim, grid_2d);
  2355. int H = image_size.first;
  2356. int W = image_size.second;
  2357. std::vector<std::vector<float>> pos_embed_2d(H * W, std::vector<float>(embed_dim));
  2358. for (int h = 0; h < H; ++h) {
  2359. for (int w = 0; w < W; ++w) {
  2360. pos_embed_2d[w * H + h] = pos_embed_3d[h][w];
  2361. }
  2362. }
  2363. return pos_embed_2d;
  2364. }
  2365. bool clip_image_encode(struct clip_ctx * ctx, const int n_threads, clip_image_f32 * img, float * vec) {
  2366. if (!ctx->has_vision_encoder) {
  2367. LOG_ERR("This gguf file seems to have no vision encoder\n");
  2368. return false;
  2369. }
  2370. clip_image_f32_batch imgs{};
  2371. imgs.size = 1;
  2372. imgs.data = img;
  2373. return clip_image_batch_encode(ctx, n_threads, &imgs, vec);
  2374. }
  2375. bool clip_image_batch_encode(clip_ctx * ctx, const int n_threads, const clip_image_f32_batch * imgs, float * vec) {
  2376. if (!ctx->has_vision_encoder) {
  2377. LOG_ERR("This gguf file seems to have no vision encoder\n");
  2378. return false;
  2379. }
  2380. int batch_size = imgs->size;
  2381. if (ctx->has_llava_projector) {
  2382. GGML_ASSERT(batch_size == 1); // TODO: support multiple images
  2383. }
  2384. if (ctx->has_minicpmv_projector) {
  2385. GGML_ASSERT(batch_size == 1);
  2386. }
  2387. if (ctx->has_glm_projector) {
  2388. GGML_ASSERT(batch_size == 1);
  2389. ggml_tensor * boi = ctx->vision_model.boi_w;
  2390. ggml_backend_tensor_get(boi,vec,0,ggml_nbytes(boi));
  2391. vec = (float*)(vec+ggml_nelements(boi)); //offset for boi
  2392. }
  2393. // build the inference graph
  2394. ggml_backend_sched_reset(ctx->sched.get());
  2395. ggml_cgraph * gf = clip_image_build_graph(ctx, imgs, ctx->load_image_size, true);
  2396. ggml_backend_sched_alloc_graph(ctx->sched.get(), gf);
  2397. // set inputs
  2398. const auto & model = ctx->vision_model;
  2399. const auto & hparams = model.hparams;
  2400. const int image_size = hparams.image_size;
  2401. int image_size_width = image_size;
  2402. int image_size_height = image_size;
  2403. if (ctx->has_minicpmv_projector | ctx->has_qwen2vl_merger) {
  2404. image_size_width = imgs->data[0].nx;
  2405. image_size_height = imgs->data[0].ny;
  2406. }
  2407. const int patch_size = hparams.patch_size;
  2408. const int num_patches = ((image_size_width / patch_size) * (image_size_height / patch_size));
  2409. const int num_positions = num_patches + (ctx->has_class_embedding ? 1 : 0);
  2410. if(ctx->load_image_size==nullptr){
  2411. ctx->load_image_size= clip_image_size_init();
  2412. }
  2413. const int pos_w = ctx->load_image_size->width/patch_size;
  2414. const int pos_h = ctx->load_image_size->height/patch_size;
  2415. {
  2416. struct ggml_tensor * inp_raw = ggml_graph_get_tensor(gf, "inp_raw");
  2417. float * data = (float *)malloc(ggml_nbytes(inp_raw));
  2418. for (size_t i = 0; i < imgs->size; i++) {
  2419. const int nx = imgs->data[i].nx;
  2420. const int ny = imgs->data[i].ny;
  2421. if (!(ctx->has_minicpmv_projector | ctx->has_qwen2vl_merger)) {
  2422. GGML_ASSERT(nx == image_size && ny == image_size);
  2423. }
  2424. const int n = nx * ny;
  2425. for (int b = 0; b < batch_size; b++) {
  2426. for (int k = 0; k < 3; k++) {
  2427. for (int y = 0; y < ny; y++) {
  2428. for (int x = 0; x < nx; x++) {
  2429. data[(b * 3 * n) + k * n + y * nx + x] = imgs->data[b].buf[3 * (y * nx + x) + k];
  2430. }
  2431. }
  2432. }
  2433. }
  2434. }
  2435. ggml_backend_tensor_set(inp_raw, data, 0, ggml_nbytes(inp_raw));
  2436. free(data);
  2437. }
  2438. if (ctx->has_minicpmv_projector) {
  2439. {
  2440. // inspired from siglip:
  2441. // -> https://huggingface.co/HuggingFaceM4/siglip-so400m-14-980-flash-attn2-navit
  2442. // -> https://huggingface.co/HuggingFaceM4/siglip-so400m-14-980-flash-attn2-navit/blob/d66538faeba44480d0bfaa42145eef26f9423199/modeling_siglip.py#L316
  2443. struct ggml_tensor * positions = ggml_graph_get_tensor(gf, "positions");
  2444. int* positions_data = (int*)malloc(ggml_nbytes(positions));
  2445. int bucket_coords_h[1024];
  2446. int bucket_coords_w[1024];
  2447. for (int i = 0; i < pos_h; i++){
  2448. bucket_coords_h[i] = std::floor(70.0*i/pos_h);
  2449. }
  2450. for (int i = 0; i < pos_w; i++){
  2451. bucket_coords_w[i] = std::floor(70.0*i/pos_w);
  2452. }
  2453. for (int i = 0, id = 0; i < pos_h; i++){
  2454. for (int j = 0; j < pos_w; j++){
  2455. positions_data[id++] = bucket_coords_h[i]*70 + bucket_coords_w[j];
  2456. }
  2457. }
  2458. ggml_backend_tensor_set(positions, positions_data, 0, ggml_nbytes(positions));
  2459. free(positions_data);
  2460. }
  2461. {
  2462. // inspired from resampler of Qwen-VL:
  2463. // -> https://huggingface.co/Qwen/Qwen-VL/tree/main
  2464. // -> https://huggingface.co/Qwen/Qwen-VL/blob/0547ed36a86561e2e42fecec8fd0c4f6953e33c4/visual.py#L23
  2465. struct ggml_tensor * pos_embed = ggml_graph_get_tensor(gf, "pos_embed");
  2466. int embed_dim = 4096;
  2467. if (ctx->minicpmv_version == 2) {
  2468. embed_dim = 4096;
  2469. }
  2470. else if (ctx->minicpmv_version == 3) {
  2471. embed_dim = 3584;
  2472. }
  2473. else if (ctx->minicpmv_version == 4) {
  2474. embed_dim = 3584;
  2475. }
  2476. auto pos_embed_t = get_2d_sincos_pos_embed(embed_dim, std::make_pair(pos_w, pos_h));
  2477. float * pos_embed_data = (float *)malloc(ggml_nbytes(pos_embed));
  2478. for(int i=0;i < pos_w * pos_h; ++i){
  2479. for(int j=0; j < embed_dim; ++j){
  2480. pos_embed_data[i * embed_dim + j] = pos_embed_t[i][j];
  2481. }
  2482. }
  2483. ggml_backend_tensor_set(pos_embed, pos_embed_data, 0, ggml_nbytes(pos_embed));
  2484. free(pos_embed_data);
  2485. }
  2486. }
  2487. else{
  2488. {
  2489. if (ctx->has_class_embedding) {
  2490. struct ggml_tensor * embeddings = ggml_graph_get_tensor(gf, "embeddings");
  2491. void* zero_mem = malloc(ggml_nbytes(embeddings));
  2492. memset(zero_mem, 0, ggml_nbytes(embeddings));
  2493. ggml_backend_tensor_set(embeddings, zero_mem, 0, ggml_nbytes(embeddings));
  2494. free(zero_mem);
  2495. }
  2496. }
  2497. if (ctx->has_qwen2vl_merger) {
  2498. struct ggml_tensor * positions = ggml_graph_get_tensor(gf, "positions");
  2499. const int pw = image_size_width / patch_size;
  2500. const int ph = image_size_height / patch_size;
  2501. int* positions_data = (int*)malloc(ggml_nbytes(positions));
  2502. int ptr = 0;
  2503. for (int y = 0; y < ph; y+=2)
  2504. {
  2505. for (int x = 0; x < pw; x+=2)
  2506. {
  2507. for (int dy = 0; dy < 2; dy++) {
  2508. for (int dx = 0; dx < 2; dx++) {
  2509. positions_data[ptr] = y + dy;
  2510. positions_data[num_patches + ptr] = x + dx;
  2511. positions_data[num_patches * 2 + ptr] = y + dy;
  2512. positions_data[num_patches * 3 + ptr] = x + dx;
  2513. ptr++;
  2514. }
  2515. }
  2516. }
  2517. }
  2518. ggml_backend_tensor_set(positions, positions_data, 0, ggml_nbytes(positions));
  2519. free(positions_data);
  2520. }
  2521. else if (ctx->proj_type == PROJECTOR_TYPE_GEMMA3) {
  2522. // do nothing
  2523. }
  2524. else {
  2525. struct ggml_tensor * positions = ggml_graph_get_tensor(gf, "positions");
  2526. int* positions_data = (int*)malloc(ggml_nbytes(positions));
  2527. for (int i = 0; i < num_positions; i++) {
  2528. positions_data[i] = i;
  2529. }
  2530. ggml_backend_tensor_set(positions, positions_data, 0, ggml_nbytes(positions));
  2531. free(positions_data);
  2532. if (!ctx->has_glm_projector) {
  2533. struct ggml_tensor * patches = ggml_graph_get_tensor(gf, "patches");
  2534. // The patches vector is used to get rows to index into the embeds with;
  2535. // we should skip dim 0 only if we have CLS to avoid going out of bounds
  2536. // when retrieving the rows.
  2537. int patch_offset = ctx->has_class_embedding ? 1 : 0;
  2538. int* patches_data = (int*)malloc(ggml_nbytes(patches));
  2539. for (int i = 0; i < num_patches; i++) {
  2540. patches_data[i] = i + patch_offset;
  2541. }
  2542. ggml_backend_tensor_set(patches, patches_data, 0, ggml_nbytes(patches));
  2543. free(patches_data);
  2544. }
  2545. }
  2546. }
  2547. ggml_backend_cpu_set_n_threads(ctx->backend_cpu, n_threads);
  2548. auto status = ggml_backend_sched_graph_compute(ctx->sched.get(), gf);
  2549. if (status != GGML_STATUS_SUCCESS) {
  2550. LOG_ERR("%s: ggml_backend_sched_graph_compute failed with error %d\n", __func__, status);
  2551. return false;
  2552. }
  2553. // the last node is the embedding tensor
  2554. struct ggml_tensor * embeddings = ggml_graph_node(gf, -1);
  2555. // copy the embeddings to the location passed by the user
  2556. ggml_backend_tensor_get(embeddings, vec, 0, ggml_nbytes(embeddings));
  2557. if (ctx->has_glm_projector) {
  2558. //eoi
  2559. ggml_tensor * eoi = ctx->vision_model.eoi_w;
  2560. int offset = ggml_nelements(embeddings);
  2561. ggml_backend_tensor_get(eoi, vec+offset, 0, ggml_nbytes(eoi));
  2562. }
  2563. return true;
  2564. }
  2565. bool clip_model_quantize(const char * fname_inp, const char * fname_out, const int itype) {
  2566. assert(itype < GGML_TYPE_COUNT);
  2567. ggml_type type = static_cast<ggml_type>(itype);
  2568. auto * ctx_clip = clip_model_load(fname_inp, 2);
  2569. const auto & ctx_src = ctx_clip->ctx_gguf;
  2570. const auto & ctx_data = ctx_clip->ctx_data;
  2571. auto * ctx_out = gguf_init_empty();
  2572. gguf_set_kv(ctx_out, ctx_src);
  2573. gguf_set_val_u32(ctx_out, "general.quantization_version", GGML_QNT_VERSION);
  2574. gguf_set_val_u32(ctx_out, "general.file_type", itype);
  2575. auto fout = std::ofstream(fname_out, std::ios::binary);
  2576. const int n_tensors = gguf_get_n_tensors(ctx_src);
  2577. for (int i = 0; i < n_tensors; ++i) {
  2578. const char * name = gguf_get_tensor_name(ctx_src, i);
  2579. struct ggml_tensor * cur = ggml_get_tensor(ctx_data, name);
  2580. gguf_add_tensor(ctx_out, cur);
  2581. }
  2582. const size_t meta_size = gguf_get_meta_size(ctx_out);
  2583. for (size_t i = 0; i < meta_size; ++i) {
  2584. fout.put(0);
  2585. }
  2586. // regexes of tensor names to be quantized
  2587. const std::vector<std::string> k_names = {
  2588. ".*weight",
  2589. };
  2590. std::vector<uint8_t> work(512);
  2591. std::vector<float> conv_buf(512);
  2592. size_t total_size_org = 0;
  2593. size_t total_size_new = 0;
  2594. for (int i = 0; i < n_tensors; ++i) {
  2595. const std::string name = gguf_get_tensor_name(ctx_src, i);
  2596. struct ggml_tensor * cur = ggml_get_tensor(ctx_data, name.c_str());
  2597. enum ggml_type new_type;
  2598. void * new_data;
  2599. size_t new_size;
  2600. bool quantize = false;
  2601. for (const auto & s : k_names) {
  2602. if (std::regex_match(name, std::regex(s))) {
  2603. quantize = true;
  2604. break;
  2605. }
  2606. }
  2607. // quantize only 2D tensors and bigger than block size
  2608. quantize &= (ggml_n_dims(cur) == 2) && cur->ne[0] > ggml_blck_size(type);
  2609. if (quantize) {
  2610. new_type = type;
  2611. if (new_type >= GGML_TYPE_Q2_K && name.find("embd") != std::string::npos) {
  2612. new_type = GGML_TYPE_Q8_0; // ggml_get_rows needs non K type
  2613. // LOG_ERR("%s: quantizing %s to %s\n", __func__, name.c_str(), ggml_type_name(new_type));
  2614. }
  2615. const size_t n_elms = ggml_nelements(cur);
  2616. float * f32_data;
  2617. switch (cur->type) {
  2618. case GGML_TYPE_F32:
  2619. f32_data = (float *)cur->data;
  2620. break;
  2621. case GGML_TYPE_F16:
  2622. if (conv_buf.size() < n_elms) {
  2623. conv_buf.resize(n_elms);
  2624. }
  2625. for (size_t j = 0; j < n_elms; ++j) {
  2626. conv_buf[j] = ggml_fp16_to_fp32(((ggml_fp16_t *)cur->data)[j]);
  2627. }
  2628. f32_data = (float *)conv_buf.data();
  2629. break;
  2630. default:
  2631. LOG_ERR("Please use an input file in f32 or f16\n");
  2632. gguf_free(ctx_out);
  2633. return false;
  2634. }
  2635. if (work.size() < n_elms * 4) {
  2636. work.resize(n_elms * 4);
  2637. }
  2638. new_data = work.data();
  2639. new_size = ggml_quantize_chunk(new_type, f32_data, new_data, 0, n_elms/cur->ne[0], cur->ne[0], nullptr);
  2640. } else {
  2641. new_type = cur->type;
  2642. new_data = cur->data;
  2643. new_size = ggml_nbytes(cur);
  2644. }
  2645. const size_t orig_size = ggml_nbytes(cur);
  2646. total_size_org += orig_size;
  2647. total_size_new += new_size;
  2648. gguf_set_tensor_type(ctx_out, name.c_str(), new_type);
  2649. GGML_ASSERT(gguf_get_tensor_size(ctx_out, gguf_find_tensor(ctx_out, name.c_str())) == new_size);
  2650. gguf_set_tensor_data(ctx_out, name.c_str(), new_data);
  2651. fout.write((const char *)new_data, new_size);
  2652. size_t pad = GGML_PAD(new_size, gguf_get_alignment(ctx_out)) - new_size;
  2653. for (size_t j = 0; j < pad; ++j) {
  2654. fout.put(0);
  2655. }
  2656. LOG_INF("%s: n_dims = %d | quantize=%d | size = %f MB -> %f MB\n", name.c_str(), ggml_n_dims(cur), quantize,
  2657. orig_size / 1024.0 / 1024.0, new_size / 1024.0 / 1024.0);
  2658. }
  2659. // go back to beginning of file and write the updated metadata
  2660. fout.seekp(0, std::ios::beg);
  2661. std::vector<uint8_t> meta(meta_size);
  2662. gguf_get_meta_data(ctx_out, meta.data());
  2663. fout.write((const char *)meta.data(), meta_size);
  2664. fout.close();
  2665. clip_free(ctx_clip);
  2666. gguf_free(ctx_out);
  2667. {
  2668. LOG_INF("%s: original size = %8.2f MB\n", __func__, total_size_org / 1024.0 / 1024.0);
  2669. LOG_INF("%s: quantized size = %8.2f MB\n", __func__, total_size_new / 1024.0 / 1024.0);
  2670. }
  2671. return true;
  2672. }
  2673. int clip_n_mmproj_embd(const struct clip_ctx * ctx) {
  2674. if (ctx->proj_type == PROJECTOR_TYPE_LDP) {
  2675. return ctx->vision_model.mm_model_block_1_block_2_1_b->ne[0];
  2676. }
  2677. if (ctx->proj_type == PROJECTOR_TYPE_LDPV2) {
  2678. return ctx->vision_model.mm_model_peg_0_b->ne[0];
  2679. }
  2680. if (ctx->proj_type == PROJECTOR_TYPE_MLP) {
  2681. return ctx->vision_model.mm_2_b->ne[0];
  2682. }
  2683. if (ctx->proj_type == PROJECTOR_TYPE_MLP_NORM) {
  2684. return ctx->vision_model.mm_3_b->ne[0];
  2685. }
  2686. if (ctx->proj_type == PROJECTOR_TYPE_RESAMPLER) {
  2687. if (ctx->minicpmv_version == 2) {
  2688. return 4096;
  2689. }
  2690. else if (ctx->minicpmv_version == 3) {
  2691. return 3584;
  2692. }
  2693. else if (ctx->minicpmv_version == 4) {
  2694. return 3584;
  2695. }
  2696. }
  2697. if (ctx->proj_type == PROJECTOR_TYPE_GLM_EDGE){
  2698. return ctx->vision_model.mm_model_mlp_3_w->ne[1];
  2699. }
  2700. if (ctx->proj_type == PROJECTOR_TYPE_MERGER) {
  2701. return ctx->vision_model.mm_1_b->ne[0];
  2702. }
  2703. if (ctx->proj_type == PROJECTOR_TYPE_GEMMA3) {
  2704. return ctx->vision_model.mm_input_proj_w->ne[0];
  2705. }
  2706. std::string proj_type = PROJECTOR_TYPE_NAMES[ctx->proj_type];
  2707. throw std::runtime_error(format("%s: don't support projector with: %s currently\n", __func__, proj_type.c_str()));
  2708. }
  2709. int clip_is_minicpmv(const struct clip_ctx * ctx) {
  2710. if (ctx->has_minicpmv_projector) {
  2711. return ctx->minicpmv_version;
  2712. }
  2713. return 0;
  2714. }
  2715. bool clip_is_glm(const struct clip_ctx * ctx) {
  2716. return ctx->has_glm_projector;
  2717. }
  2718. bool clip_is_qwen2vl(const struct clip_ctx * ctx) {
  2719. return ctx->has_qwen2vl_merger;
  2720. }
  2721. // Determine the number of encoder layers to iterate over
  2722. int get_deepest_feature_layer(const struct clip_ctx * ctx) {
  2723. // Get the index of the second to last layer; this is the
  2724. // default for models that have a llava projector
  2725. const auto & hparams = ctx->vision_model.hparams;
  2726. int n_layer = hparams.n_layer - 1;
  2727. int deepest_feature_layer = -1;
  2728. // Handle other projectors; incrementing here indicates that we
  2729. // should use the last encoder layer for the vision features.
  2730. if (ctx->has_minicpmv_projector || ctx->has_glm_projector || ctx->has_qwen2vl_merger) {
  2731. n_layer += 1;
  2732. }
  2733. // If we set explicit vision feature layers, only go up to the deepest one
  2734. for (const auto & feature_layer : hparams.vision_feature_layer) {
  2735. if (feature_layer > deepest_feature_layer) {
  2736. deepest_feature_layer = feature_layer;
  2737. }
  2738. }
  2739. return deepest_feature_layer < 0 ? n_layer : deepest_feature_layer;
  2740. }
  2741. bool clip_encode_float_image (struct clip_ctx * ctx, int n_threads, float * img, int h, int w, float * vec) {
  2742. clip_image_f32 clip_img;
  2743. clip_img.buf.resize(h * w * 3);
  2744. for (int i = 0; i < h*w*3; i++)
  2745. {
  2746. clip_img.buf[i] = img[i];
  2747. }
  2748. clip_img.nx = w;
  2749. clip_img.ny = h;
  2750. clip_image_encode(ctx, n_threads, &clip_img, vec);
  2751. return true;
  2752. }