Spaces:
Sleeping
Sleeping
Bobby
commited on
Commit
·
8e103f0
1
Parent(s):
24438e9
removed spaces gpu initialization
Browse files
app.py
CHANGED
|
@@ -98,13 +98,13 @@ if gr.NO_RELOAD:
|
|
| 98 |
# gc.collect()
|
| 99 |
print("---------------Loaded controlnet pipeline---------------")
|
| 100 |
|
| 101 |
-
@spaces.GPU(duration=12)
|
| 102 |
-
def init(pipe):
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
init(pipe)
|
| 108 |
|
| 109 |
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
| 110 |
if randomize_seed:
|
|
|
|
| 98 |
# gc.collect()
|
| 99 |
print("---------------Loaded controlnet pipeline---------------")
|
| 100 |
|
| 101 |
+
# @spaces.GPU(duration=12)
|
| 102 |
+
# def init(pipe):
|
| 103 |
+
# pipe.enable_xformers_memory_efficient_attention()
|
| 104 |
+
# pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
| 105 |
+
# pipe.unet.set_attn_processor(AttnProcessor2_0())
|
| 106 |
+
# print("Model Compiled!")
|
| 107 |
+
# init(pipe)
|
| 108 |
|
| 109 |
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
| 110 |
if randomize_seed:
|