diff --git a/scripts/c-btmInference.py b/scripts/c-btmInference.py index c3f548e..a9f34e3 100644 --- a/scripts/c-btmInference.py +++ b/scripts/c-btmInference.py @@ -106,7 +106,6 @@ def generateSequence(embedder, prompt, end_token, models, tokenizers, maxLength, currToken, currSequence = None, prompt while len(currSequence) < maxLength:# or currToken != end_token: currToken, currTokenProb = findNextToken(embedder, models, tokenizers, currSequence, k, T, clusterCenters) - print("currToken: ", currToken, len(currSequence) < maxLength) currSequence = currSequence + currToken return generateSequence