Spaces:
Enderchef
/
Running on Zero

Enderchef commited on
Commit
46f6bfb
·
verified ·
1 Parent(s): e39d67a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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"))