Hugging Face's logo

Spaces:
multimodalart
/
AuraFlow
Running on Zero

App Files Community
4
multimodalart HF staff commited on
Commit
19dc0a3
·
verified ·
1 Parent(s): bbf4c64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ pipe = AuraFlowPipeline.from_pretrained(
35
  MAX_SEED = np.iinfo(np.int32).max
36
  MAX_IMAGE_SIZE = 1024
37
 
38
- @spaces.GPU
39
  def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=28, model_version="0.2", comparison_mode=False, progress=gr.Progress(track_tqdm=True)):
40
 
41
  if randomize_seed:
 
35
  MAX_SEED = np.iinfo(np.int32).max
36
  MAX_IMAGE_SIZE = 1024
37
 
38
+ @spaces.GPU(duration=160)
39
  def infer(prompt, negative_prompt="", seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=5.0, num_inference_steps=28, model_version="0.2", comparison_mode=False, progress=gr.Progress(track_tqdm=True)):
40
 
41
  if randomize_seed: