linoyts HF Staff commited on
Commit
1bbae28
·
verified ·
1 Parent(s): bae80e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import spaces
4
  import torch
5
  import time
6
  import json
 
7
  from diffusers import BriaFiboPipeline
8
  from diffusers.modular_pipelines import ModularPipeline
9
 
@@ -18,7 +19,7 @@ from diffusers.modular_pipelines import ModularPipeline
18
  # "1280 800",
19
  # "1344 768",
20
  # ]
21
-
22
  dtype = torch.bfloat16
23
  device = "cuda" if torch.cuda.is_available() else "cpu"
24
 
 
4
  import torch
5
  import time
6
  import json
7
+ import numpy as np
8
  from diffusers import BriaFiboPipeline
9
  from diffusers.modular_pipelines import ModularPipeline
10
 
 
19
  # "1280 800",
20
  # "1344 768",
21
  # ]
22
+ MAX_SEED = np.iinfo(np.int32).max
23
  dtype = torch.bfloat16
24
  device = "cuda" if torch.cuda.is_available() else "cpu"
25