AbstractPhil commited on
Commit
6f70ac0
Β·
1 Parent(s): fddb899
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import spaces
 
2
  import torch
3
  import gradio as gr
4
  import numpy as np
@@ -135,8 +136,8 @@ def encode_sdxl_prompt(prompt, negative_prompt=""):
135
  }
136
 
137
  # ─── Inference ────────────────────────────────────────────────
138
- @torch.no_grad()
139
  @spaces.GPU
 
140
  def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
141
  use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
142
 
 
1
  import spaces
2
+
3
  import torch
4
  import gradio as gr
5
  import numpy as np
 
136
  }
137
 
138
  # ─── Inference ────────────────────────────────────────────────
 
139
  @spaces.GPU
140
+ @torch.no_grad()
141
  def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
142
  use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
143
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  sentencepiece
2
  accelerate
3
  diffusers
 
1
+ spaces
2
  sentencepiece
3
  accelerate
4
  diffusers