Spaces:
Running on Zero
Running on Zero
change hyperparamater temperature
Browse files
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 =
|
| 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"])
|