Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,7 @@ DESCRIPTION = "# ICONN Lite Chat"
|
|
15 |
if not torch.cuda.is_available():
|
16 |
DESCRIPTION += "\n<p class='warning'>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
17 |
|
|
|
18 |
MAX_MAX_NEW_TOKENS = 100000000
|
19 |
DEFAULT_MAX_NEW_TOKENS = 10240
|
20 |
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|
|
|
15 |
if not torch.cuda.is_available():
|
16 |
DESCRIPTION += "\n<p class='warning'>Running on CPU 🥶 This demo does not work on CPU.</p>"
|
17 |
|
18 |
+
top_k: int = 50
|
19 |
MAX_MAX_NEW_TOKENS = 100000000
|
20 |
DEFAULT_MAX_NEW_TOKENS = 10240
|
21 |
MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
|