ford442 commited on
Commit
3781ec5
·
verified ·
1 Parent(s): 22cec4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -243,6 +243,10 @@ def generate_30(
243
  if sage==False:
244
  F.scaled_dot_product_attention = F.scaled_dot_product_attention
245
  seed = random.randint(0, MAX_SEED)
 
 
 
 
246
  generator = torch.Generator(device='cuda').manual_seed(seed)
247
  options = {
248
  "prompt": [prompt],
@@ -297,6 +301,10 @@ def generate_60(
297
  if sage==False:
298
  F.scaled_dot_product_attention = F.scaled_dot_product_attention
299
  seed = random.randint(0, MAX_SEED)
 
 
 
 
300
  generator = torch.Generator(device='cuda').manual_seed(seed)
301
  options = {
302
  "prompt": [prompt],
@@ -351,6 +359,10 @@ def generate_90(
351
  if sage==False:
352
  F.scaled_dot_product_attention = F.scaled_dot_product_attention
353
  seed = random.randint(0, MAX_SEED)
 
 
 
 
354
  generator = torch.Generator(device='cuda').manual_seed(seed)
355
  options = {
356
  "prompt": [prompt],
 
243
  if sage==False:
244
  F.scaled_dot_product_attention = F.scaled_dot_product_attention
245
  seed = random.randint(0, MAX_SEED)
246
+ random.seed(seed)
247
+ np.random.seed(seed)
248
+ torch.manual_seed(seed)
249
+ torch.cuda.manual_seed_all(seed)
250
  generator = torch.Generator(device='cuda').manual_seed(seed)
251
  options = {
252
  "prompt": [prompt],
 
301
  if sage==False:
302
  F.scaled_dot_product_attention = F.scaled_dot_product_attention
303
  seed = random.randint(0, MAX_SEED)
304
+ random.seed(seed)
305
+ np.random.seed(seed)
306
+ torch.manual_seed(seed)
307
+ torch.cuda.manual_seed_all(seed)
308
  generator = torch.Generator(device='cuda').manual_seed(seed)
309
  options = {
310
  "prompt": [prompt],
 
359
  if sage==False:
360
  F.scaled_dot_product_attention = F.scaled_dot_product_attention
361
  seed = random.randint(0, MAX_SEED)
362
+ random.seed(seed)
363
+ np.random.seed(seed)
364
+ torch.manual_seed(seed)
365
+ torch.cuda.manual_seed_all(seed)
366
  generator = torch.Generator(device='cuda').manual_seed(seed)
367
  options = {
368
  "prompt": [prompt],