remove print statement

huu4ontocord · Sep 12, 2023 · 74dadef · 74dadef
1 parent bed3ec1
commit 74dadef
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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