valhalla commited on
Commit
11b197a
1 Parent(s): 405665c

Update server.py

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