SaniaE commited on
Commit
0a7dd0a
·
verified ·
1 Parent(s): 6b4af39

fixed seed logic

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -88,7 +88,9 @@ def generate_random():
88
  if gen_model is None: raise HTTPException(status_code=503)
89
 
90
  with torch.inference_mode():
91
- torch.seed()
 
 
92
  noise = torch.randn(1, Z_DIM, device=DEVICE)
93
  print("NOISE:", noise[0, :5])
94
  fake_img = gen_model(noise)
 
88
  if gen_model is None: raise HTTPException(status_code=503)
89
 
90
  with torch.inference_mode():
91
+ seed = torch.seed()
92
+ torch.manual_seed(seed)
93
+
94
  noise = torch.randn(1, Z_DIM, device=DEVICE)
95
  print("NOISE:", noise[0, :5])
96
  fake_img = gen_model(noise)