valhalla commited on
Commit
125b475
1 Parent(s): c55fdff

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +1 -1
server.py CHANGED
@@ -51,7 +51,7 @@ def generate(prompt):
51
  def sample(prompt):
52
  # Sampling
53
  images = (
54
- model.sampling(prompt=prompt, top_k=256, top_p=None, softmax_temperature=1.0, num_candidates=9, device=device)
55
  .cpu()
56
  .numpy()
57
  )
 
51
  def sample(prompt):
52
  # Sampling
53
  images = (
54
+ model.sampling(prompt=prompt, top_k=96, top_p=None, softmax_temperature=1.0, num_candidates=9, device=device)
55
  .cpu()
56
  .numpy()
57
  )