JMalott commited on
Commit
4928505
1 Parent(s): 153e928

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -127,7 +127,7 @@ def generate2(prompt,crazy,k):
127
  grid_size=1,
128
  is_seamless=False,
129
  temperature=crazy,
130
- top_k=1,#2128,
131
  supercondition_factor=16,
132
  is_verbose=False
133
  )
 
127
  grid_size=1,
128
  is_seamless=False,
129
  temperature=crazy,
130
+ top_k=k,#2128,
131
  supercondition_factor=16,
132
  is_verbose=False
133
  )