Remove "first token must be BOS" restriction

This commit is contained in:
oobabooga 2023-07-08 23:50:19 -03:00 committed by GitHub
parent 64639555ff
commit 3d9871715a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1291,12 +1291,6 @@ static bool llama_eval_internal(
LLAMA_ASSERT((!tokens && embd) || (tokens && !embd)); LLAMA_ASSERT((!tokens && embd) || (tokens && !embd));
// enforce that the first token is BOS
if (tokens && n_past == 0 && tokens[0] != llama_token_bos()) {
fprintf(stderr, "%s: first token must be BOS\n", __func__);
return false;
}
const int64_t t_start_us = ggml_time_us(); const int64_t t_start_us = ggml_time_us();
const int N = n_tokens; const int N = n_tokens;