Amir Zait commited on
Commit
7c6a43f
1 Parent(s): 23dd537
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -44,12 +44,12 @@ def generate_image(text):
44
 
45
  encoded_images = model.generate(
46
  **tokenized_prompt,
47
- random.randint(0, 1e7),
48
- model.params,
49
- gen_top_k,
50
- gen_top_p,
51
- temperature,
52
- cond_scale,
53
  )
54
  encoded_images = encoded_images.sequences[..., 1:]
55
  decoded_images = model.decode(encoded_images, vqgan.params)
 
44
 
45
  encoded_images = model.generate(
46
  **tokenized_prompt,
47
+ prng_key=random.randint(0, 1e7),
48
+ params=model.params,
49
+ top_k=gen_top_k,
50
+ top_p=gen_top_p,
51
+ temperature=temperature,
52
+ condition_scale=cond_scale,
53
  )
54
  encoded_images = encoded_images.sequences[..., 1:]
55
  decoded_images = model.decode(encoded_images, vqgan.params)