"lm_eval/tasks/catalan_bench/arc_ca_easy.yaml" did not exist on "c887796d245a489284a44ea57bf29e743aa56510"
universal_vocab.cc 342 Bytes
Newer Older
SWHL's avatar
SWHL committed
1
2
3
4
5
6
7
8
9
10
11
12
13
#include "universal_vocab.hh"

namespace lm {
namespace interpolate {

UniversalVocab::UniversalVocab(const std::vector<WordIndex>& model_vocab_sizes) {
  model_index_map_.resize(model_vocab_sizes.size());
  for (size_t i = 0; i < model_vocab_sizes.size(); ++i) {
    model_index_map_[i].resize(model_vocab_sizes[i]);
  }
}

}} // namespaces