Method ListLocalModels

apps/mnncli/src/local_model_utils.cpp:126–156

}

int LocalModelUtils::ListLocalModels() {
  // Iterate through all known providers
  static const std::vector<std::string> kProviders = {
    "HuggingFace", "ModelScope", "Modelers"
  };
  
  std::vector<std::string> all_models;
  auto &config_mgr = ConfigManager::GetInstance();
  auto config = config_mgr.LoadConfig();

  mnn::downloader::Config downloader_config;
  downloader_config.cache_dir = config.cache_dir;
  downloader_config.download_provider = config.download_provider;

  for (const auto& provider : kProviders) {
    auto models = ListLocalModelsInner(provider, config_mgr.GetBaseCacheDir());
    all_models.insert(all_models.end(), models.begin(), models.end());
  }
  if (!all_models.empty()) {
    std::cout << "Local models:\n";
    for (const auto &name : all_models) {
      std::cout << mnn::downloader::ModelNameUtils::GetDisplayModelName(name, downloader_config)
                << "\n";
    }
  } else {
    std::cout << "No local models found.\n";
    std::cout << "Use 'mnncli search <keyword>' to search remote models\n";
    std::cout << "Use 'mnncli download <name>' to download models\n";
  }
  return 0;
}

} // namespace mnncli