AbstractPhil commited on
Commit
23d07c9
Β·
1 Parent(s): 25bf19b
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  import gradio as gr
3
  import numpy as np
@@ -133,6 +134,7 @@ def encode_sdxl_prompt(prompt, negative_prompt=""):
133
 
134
  # ─── Inference ────────────────────────────────────────────────
135
  @torch.no_grad()
 
136
  def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
137
  use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
138
 
 
1
+ import spaces
2
  import torch
3
  import gradio as gr
4
  import numpy as np
 
134
 
135
  # ─── Inference ────────────────────────────────────────────────
136
  @torch.no_grad()
137
+ @spaces.GPU
138
  def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
139
  use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
140