pass jiuge.py

Signed-off-by: Ceng23333 <441651826@qq.com>

Author: Ceng23333

csrc/config/config_factory.cpp

@@ -16,7 +16,7 @@ std::shared_ptr<infinilm::config::ModelConfig> ConfigFactory::createConfig(const
     if (it != config_map.end()) {
         it->second(model_config);
     } else {
-        std::vector<std::string> classic_models = {"llama", "qwen2", "minicpm", "fm9g", "fm9g7b"};
+        std::vector<std::string> classic_models = {"llama", "qwen2", "minicpm", "minicpm_sala", "fm9g", "fm9g7b"};
         const std::string &model_type = model_config->get<std::string>("model_type");
         if (std::find(classic_models.begin(), classic_models.end(), model_type) == classic_models.end()) {
             throw std::invalid_argument("infinilm::config::ConfigFactory::createConfig: Unsupported model config type: " + model_type);

csrc/engine/rank_worker.cpp

@@ -262,7 +262,7 @@ void RankWorker::thread_loop() {
                     rank_info_.device,
                     pending_cache_config_ != nullptr ? pending_cache_config_.get() : nullptr);
             } else {
-                std::vector<std::string> classic_models = {"llama", "qwen2", "minicpm", "fm9g", "fm9g7b"};
+                std::vector<std::string> classic_models = {"llama", "qwen2", "minicpm", "minicpm_sala", "fm9g", "fm9g7b"};
                 if ((std::find(classic_models.begin(), classic_models.end(), model_type) != classic_models.end())) {
                     model_ = InfinilmModelFactory::createModel(
                         model_config_,

csrc/models/minicpm_sala/minicpm_sala_decoderLayer.cpp

@@ -60,4 +60,13 @@ std::shared_ptr<InfinilmModel> InfinilmModelFactory::createModel(
 
     return model;
 }
+
+std::shared_ptr<InfinilmModel> InfinilmModelFactory::createModel(
+    std::shared_ptr<infinilm::config::ModelConfig> model_config,
+    const infinicore::Device &device,
+    const cache::CacheConfig *cache) {
+    engine::distributed::RankInfo rank_info;
+    rank_info.device = device;
+    return createModel(model_config, rank_info, cache, backends::AttentionBackend::Default);
+}
 } // namespace infinilm