Skip to content

Commit

Permalink
Fixed compilation
Browse files Browse the repository at this point in the history
  • Loading branch information
ilya-lavrenov committed Dec 27, 2024
1 parent 37beb35 commit 69b603e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/cpp/src/generation_config.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ GenerationConfig::GenerationConfig(const std::filesystem::path& json_path) {
read_json_param(data, "eos_token_id", ordered_stop_token_ids);

if (!ordered_stop_token_ids.empty()) {
std::copy(ordered_stop_token_ids.begin(), ordered_stop_token_ids.end(), std::back_inserter(stop_token_ids));
for (int64_t stop_token_id : ordered_stop_token_ids)
stop_token_ids.insert(stop_token_id);

if (eos_token_id == -1) {
eos_token_id = ordered_stop_token_ids[0];
Expand Down

0 comments on commit 69b603e

Please sign in to comment.