AbstractPhil commited on
Commit
dfcfa0d
Β·
1 Parent(s): 1ee003a
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -132,8 +132,8 @@ def encode_sdxl_prompt(prompt, negative_prompt=""):
132
  # ─── Inference ────────────────────────────────────────────
133
 
134
 
135
- @spaces.GPU
136
  @torch.no_grad()
 
137
  def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
138
  use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
139
 
@@ -335,14 +335,12 @@ with gr.Blocks(title="SDXL Dual Shunt Adapter", theme=gr.themes.Soft()) as demo:
335
  stats_g = gr.Textbox(label="CLIP-G Stats", interactive=False)
336
 
337
  # Event handlers
338
- @spaces.GPU
339
  def process_adapters(adapter_l_val, adapter_g_val):
340
  # Convert "None" back to None for processing
341
  adapter_l_processed = None if adapter_l_val == "None" else adapter_l_val
342
  adapter_g_processed = None if adapter_g_val == "None" else adapter_g_val
343
  return adapter_l_processed, adapter_g_processed
344
 
345
- @spaces.GPU
346
  def run_inference(*args):
347
  # Process adapter selections
348
  adapter_l_processed, adapter_g_processed = process_adapters(args[2], args[3])
 
132
  # ─── Inference ────────────────────────────────────────────
133
 
134
 
 
135
  @torch.no_grad()
136
+ @spaces.GPU
137
  def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
138
  use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
139
 
 
335
  stats_g = gr.Textbox(label="CLIP-G Stats", interactive=False)
336
 
337
  # Event handlers
 
338
  def process_adapters(adapter_l_val, adapter_g_val):
339
  # Convert "None" back to None for processing
340
  adapter_l_processed = None if adapter_l_val == "None" else adapter_l_val
341
  adapter_g_processed = None if adapter_g_val == "None" else adapter_g_val
342
  return adapter_l_processed, adapter_g_processed
343
 
 
344
  def run_inference(*args):
345
  # Process adapter selections
346
  adapter_l_processed, adapter_g_processed = process_adapters(args[2], args[3])