Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Cast hyperparameter types
Browse files
app.py
CHANGED
@@ -41,11 +41,11 @@ def proc_submission(
|
|
41 |
"""
|
42 |
|
43 |
settings = {
|
44 |
-
"length_penalty": length_penalty,
|
45 |
-
"repetition_penalty": repetition_penalty,
|
46 |
-
"no_repeat_ngram_size": no_repeat_ngram_size,
|
47 |
"encoder_no_repeat_ngram_size": 4,
|
48 |
-
"num_beams": num_beams,
|
49 |
"min_length": 4,
|
50 |
"max_length": int(token_batch_length // 4),
|
51 |
"early_stopping": True,
|
|
|
41 |
"""
|
42 |
|
43 |
settings = {
|
44 |
+
"length_penalty": float(length_penalty),
|
45 |
+
"repetition_penalty": float(repetition_penalty),
|
46 |
+
"no_repeat_ngram_size": int(no_repeat_ngram_size),
|
47 |
"encoder_no_repeat_ngram_size": 4,
|
48 |
+
"num_beams": int(num_beams),
|
49 |
"min_length": 4,
|
50 |
"max_length": int(token_batch_length // 4),
|
51 |
"early_stopping": True,
|