Unverified Commit 05c879cc authored by Lintang Sutawika's avatar Lintang Sutawika Committed by GitHub
Browse files

Merge pull request #700 from ZZR0/greedy_until--patch-1

Early stop bug of greedy_until (primary_until should be a list of str)
parents 8e8212fc 984f8793
......@@ -723,7 +723,7 @@ class HFLM(LM):
else:
max_gen_toks = self.max_gen_toks
# first stop sequence is used to halt generation upon encountering
(primary_until) = until[0]
primary_until = [until[0]]
# set the max length in tokens of inputs ("context_enc")
if self.AUTO_MODEL_CLASS == transformers.AutoModelForCausalLM:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment