Corianas commited on
Commit
630a257
1 Parent(s): 0398507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def add_caseifer(text):
31
 
32
  max_new_tokens = 88 # number of tokens generated in each sample
33
  temperature = 0.8 # 1.0 = no change, < 1.0 = less random, > 1.0 = more random, in predictions
34
- top_k = 48 # retain only the top_k most likely tokens, clamp others to have 0 probability
35
  device = 'cpu' # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
36
  dtype = 'bfloat16' # 'float32' or 'bfloat16' or 'float16'
37
  out_dir = 'Eml' # ignored if init_from is not 'resume'
 
31
 
32
  max_new_tokens = 88 # number of tokens generated in each sample
33
  temperature = 0.8 # 1.0 = no change, < 1.0 = less random, > 1.0 = more random, in predictions
34
+ top_k = None # retain only the top_k most likely tokens, clamp others to have 0 probability
35
  device = 'cpu' # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
36
  dtype = 'bfloat16' # 'float32' or 'bfloat16' or 'float16'
37
  out_dir = 'Eml' # ignored if init_from is not 'resume'