Spaces:
Sleeping
Sleeping
Commit
·
b236df4
1
Parent(s):
2be296c
testing global preprocessor
Browse files
app.py
CHANGED
|
@@ -110,9 +110,11 @@ pipe.to("cuda")
|
|
| 110 |
|
| 111 |
print("loading preprocessor")
|
| 112 |
preprocessor = Preprocessor()
|
| 113 |
-
|
| 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
|