Spaces:
Runtime error
Runtime error
Amir Zait
commited on
Commit
•
6c6d0a0
1
Parent(s):
7c6a43f
fixed prng
Browse files- app.py +2 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import soundfile as sf
|
| 2 |
import gradio as gr
|
|
|
|
| 3 |
import numpy as np
|
| 4 |
import os
|
| 5 |
from PIL import Image
|
|
@@ -44,7 +45,7 @@ def generate_image(text):
|
|
| 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,
|
|
|
|
| 1 |
import soundfile as sf
|
| 2 |
import gradio as gr
|
| 3 |
+
import jax
|
| 4 |
import numpy as np
|
| 5 |
import os
|
| 6 |
from PIL import Image
|
|
|
|
| 45 |
|
| 46 |
encoded_images = model.generate(
|
| 47 |
**tokenized_prompt,
|
| 48 |
+
prng_key=jax.random.PRNGKey(random.randint(0, 1e7)),
|
| 49 |
params=model.params,
|
| 50 |
top_k=gen_top_k,
|
| 51 |
top_p=gen_top_p,
|
requirements.txt
CHANGED
|
@@ -4,6 +4,7 @@ torch
|
|
| 4 |
transformers
|
| 5 |
sentencepiece
|
| 6 |
sox
|
|
|
|
| 7 |
dalle-mini
|
| 8 |
Pillow
|
| 9 |
numpy
|
|
|
|
| 4 |
transformers
|
| 5 |
sentencepiece
|
| 6 |
sox
|
| 7 |
+
jax
|
| 8 |
dalle-mini
|
| 9 |
Pillow
|
| 10 |
numpy
|