Bobby commited on
Commit
8e103f0
·
1 Parent(s): 24438e9

removed spaces gpu initialization

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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
- 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:
 
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: