Skip to content

Commit

Permalink
fix: fixed set seed to large int from torch.seed
Browse files Browse the repository at this point in the history
  • Loading branch information
Tan Nian Wei committed Sep 27, 2021
1 parent 3b8c85d commit 32de14f
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions app.py
Original file line number Diff line number Diff line change
Expand Up @@ -266,9 +266,16 @@ def generate_image(

seed_widget = st.sidebar.empty()
if set_seed is True:
seed = seed_widget.number_input(
"Seed", value=defaults["seed"], help="Random seed to use"
# Use text_input as number_input relies on JS
# which can't natively handle large numbers
# torch.seed() generates int w/ 19 or 20 chars!
seed_str = seed_widget.text_input(
"Seed", value=str(defaults["seed"]), help="Random seed to use"
)
try:
seed = int(seed_str)
except ValueError as e:
st.error("seed input needs to be int")
else:
seed = None

Expand Down

0 comments on commit 32de14f

Please sign in to comment.