developerskyebrowse commited on
Commit
5120359
·
1 Parent(s): 9e112ae

no gr.noreload and preprocessor in the top

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -27,6 +27,9 @@ API_KEY = os.environ.get("API_KEY", None)
27
  print("CUDA version:", torch.version.cuda)
28
  print("loading pipe")
29
  compiled = False
 
 
 
30
  # api = HfApi()
31
 
32
  import spaces
@@ -99,9 +102,6 @@ def init(pipe):
99
  pipe.enable_xformers_memory_efficient_attention()
100
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
101
  pipe.unet.set_attn_processor(AttnProcessor2_0())
102
- from preprocess import Preprocessor
103
- preprocessor = Preprocessor()
104
- preprocessor.load("NormalBae")
105
  print("Model Compiled!")
106
  init(pipe)
107
 
 
27
  print("CUDA version:", torch.version.cuda)
28
  print("loading pipe")
29
  compiled = False
30
+ from preprocess import Preprocessor
31
+ preprocessor = Preprocessor()
32
+ preprocessor.load("NormalBae")
33
  # api = HfApi()
34
 
35
  import spaces
 
102
  pipe.enable_xformers_memory_efficient_attention()
103
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
104
  pipe.unet.set_attn_processor(AttnProcessor2_0())
 
 
 
105
  print("Model Compiled!")
106
  init(pipe)
107