mikonvergence commited on
Commit
d787e0c
·
1 Parent(s): 4d22630

misplaced decorater

Browse files
Files changed (1) hide show
  1. src/backend.py +1 -1
src/backend.py CHANGED
@@ -203,7 +203,6 @@ def run_inference(config, nnet, autoencoder, img_tensors):
203
 
204
  return generated_samples
205
 
206
- @spaces.GPU
207
  def custom_inference(images, generate_modalities, condition_modalities, num_inference_steps, seed=None):
208
  """
209
  Run custom inference with user-specified parameters
@@ -228,6 +227,7 @@ def custom_inference(images, generate_modalities, condition_modalities, num_infe
228
 
229
  return results
230
 
 
231
  def generate_output(s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_inference_steps_slider, seed_number, ignore_seed):
232
 
233
  seed = seed_number if not ignore_seed else None
 
203
 
204
  return generated_samples
205
 
 
206
  def custom_inference(images, generate_modalities, condition_modalities, num_inference_steps, seed=None):
207
  """
208
  Run custom inference with user-specified parameters
 
227
 
228
  return results
229
 
230
+ @spaces.GPU
231
  def generate_output(s2l1c_input, s2l2a_input, s1rtc_input, dem_input, num_inference_steps_slider, seed_number, ignore_seed):
232
 
233
  seed = seed_number if not ignore_seed else None