Spaces:
Runtime error
Runtime error
Amir Zait
commited on
Commit
·
7c6a43f
1
Parent(s):
23dd537
fixed
Browse files
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)
|