Adapt to the changes of gcpp::ImageTokens

ufownl · Nov 5, 2024 · ae2fd2f · ae2fd2f
1 parent 4ca1ef6
commit ae2fd2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/session.cpp b/src/session.cpp
@@ -319,7 +319,7 @@ session::session(instance* inst, int argc, char* argv[])
   }
   const auto& cfg = inst->model().GetModelConfig();
   if (cfg.vit_seq_len > 0) {
-    img_ = gcpp::ImageTokens(cfg.vit_seq_len, cfg.model_dim);
+    img_ = gcpp::ImageTokens(gcpp::Extents2D(cfg.vit_seq_len, cfg.model_dim));
   }
   kv_cache_ = gcpp::KVCache::Create(cfg, args_.prefill_tbatch_size);
 }