Spaces:
Running
on
Zero
Running
on
Zero
mikonvergence
commited on
Commit
·
d787e0c
1
Parent(s):
4d22630
misplaced decorater
Browse files- 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
|