Балаганский Никита Николаевич
commited on
Commit
•
333ca76
1
Parent(s):
87f68f3
fix
Browse files
app.py
CHANGED
|
@@ -88,6 +88,7 @@ def inference(
|
|
| 88 |
caif_sampler = load_sampler(cls_model_name=cls_model_name, lm_tokenizer=lm_tokenizer)
|
| 89 |
else:
|
| 90 |
caif_sampler = None
|
|
|
|
| 91 |
if entropy_threshold < 0.05:
|
| 92 |
entropy_threshold = None
|
| 93 |
generator.set_caif_sampler(caif_sampler)
|
|
|
|
| 88 |
caif_sampler = load_sampler(cls_model_name=cls_model_name, lm_tokenizer=lm_tokenizer)
|
| 89 |
else:
|
| 90 |
caif_sampler = None
|
| 91 |
+
entropy_threshold = None
|
| 92 |
if entropy_threshold < 0.05:
|
| 93 |
entropy_threshold = None
|
| 94 |
generator.set_caif_sampler(caif_sampler)
|