Ruurd commited on
Commit
0904e3f
·
verified ·
1 Parent(s): 153c673

change hyperparamater temperature

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ BIGRAM_REMASKING_CONFIDENCE = "Diffusion probability"
31
  PMI_MIN_COUNT = 1
32
  PMI_CLIP = 2.0
33
  # Preserve the candidate pool and balance used in the matched Viterbi tests.
34
- DEFAULT_TEMPERATURE = 1.0 if BIGRAM_ENABLED else 0.7
35
  DEFAULT_TOP_K = 16 if BIGRAM_ENABLED else 3
36
 
37
  MODEL_REPO_ID = os.getenv("MODEL_REPO_ID", SPACE["model_repo_id"])
 
31
  PMI_MIN_COUNT = 1
32
  PMI_CLIP = 2.0
33
  # Preserve the candidate pool and balance used in the matched Viterbi tests.
34
+ DEFAULT_TEMPERATURE = 2.0 if BIGRAM_ENABLED else 0.7
35
  DEFAULT_TOP_K = 16 if BIGRAM_ENABLED else 3
36
 
37
  MODEL_REPO_ID = os.getenv("MODEL_REPO_ID", SPACE["model_repo_id"])