Балаганский Никита Николаевич
commited on
Commit
•
116ed83
1
Parent(s):
d2c745e
add logging
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ def inference(lm_model_name: str, cls_model_name: str, prompt: str, fp16: bool =
|
|
72 |
max_length=20,
|
73 |
caif_period=1,
|
74 |
caif_tokens_num=100,
|
75 |
-
entropy=
|
76 |
**kwargs
|
77 |
)
|
78 |
print(f"Output for prompt: {sequences}")
|
|
|
72 |
max_length=20,
|
73 |
caif_period=1,
|
74 |
caif_tokens_num=100,
|
75 |
+
entropy=None,
|
76 |
**kwargs
|
77 |
)
|
78 |
print(f"Output for prompt: {sequences}")
|