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)
|