Skip to content

Commit 7adfc50

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Set RESULT_CUTOFF based on Phase I-b
1 parent 528a1ec commit 7adfc50

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def objective(trial: optuna.Trial) -> float:
8787

8888
GENERATION_PROMPT_LEN = 25
8989
MAX_NEW_TOKENS = MAX_SEQ_LENGTH - GENERATION_PROMPT_LEN
90-
RESULT_CUTOFF = 20 # 100 # <---<< In production 100 # Only print out verbose text samples when perplexity is < RESULT_CUTOFF
90+
RESULT_CUTOFF = 50 # 20 # 100 # <---<< In production 100 # Only print out verbose text samples when perplexity is < RESULT_CUTOFF
9191

9292
if GENERATION_PROMPT_LEN + MAX_NEW_TOKENS > MAX_SEQ_LENGTH:
9393
raise ValueError("Sequence length overflow: Generated text length (GENERATION_PROMPT_LEN + MAX_NEW_TOKENS) "

0 commit comments

Comments
 (0)