developerskyebrowse commited on
Commit
b236df4
·
1 Parent(s): 2be296c

testing global preprocessor

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -110,9 +110,11 @@ pipe.to("cuda")
110
 
111
  print("loading preprocessor")
112
  preprocessor = Preprocessor()
113
- # preprocessor.load("NormalBae")
114
 
115
  print("---------------Loaded controlnet pipeline---------------")
 
 
116
 
117
  # @spaces.GPU(duration=12)
118
  # def init(pipe):
@@ -401,6 +403,8 @@ def process_image(
401
  torch.cuda.synchronize()
402
  preprocess_start = time.time()
403
  print("processing image")
 
 
404
  preprocessor.load("NormalBae")
405
 
406
  global compiled
 
110
 
111
  print("loading preprocessor")
112
  preprocessor = Preprocessor()
113
+ preprocessor.load("NormalBae")
114
 
115
  print("---------------Loaded controlnet pipeline---------------")
116
+ print(f"CUDA memory allocated: {torch.cuda.max_memory_allocated(device='cuda') / 1e9:.2f} GB")
117
+
118
 
119
  # @spaces.GPU(duration=12)
120
  # def init(pipe):
 
403
  torch.cuda.synchronize()
404
  preprocess_start = time.time()
405
  print("processing image")
406
+
407
+ global preprocessor
408
  preprocessor.load("NormalBae")
409
 
410
  global compiled