eienmojiki commited on
Commit
e920d8b
·
verified ·
1 Parent(s): f03e875

A small fix :D

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
52
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
53
  if randomize_seed:
54
  seed = random.randint(0, MAX_SEED)
55
- return seed
56
 
57
  def seed_everything(seed: int) -> torch.Generator:
58
  torch.manual_seed(seed)
 
52
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
53
  if randomize_seed:
54
  seed = random.randint(0, MAX_SEED)
55
+ return seed
56
 
57
  def seed_everything(seed: int) -> torch.Generator:
58
  torch.manual_seed(seed)