Spaces:

omer11a
/

bounded-attention

Runtime error

omer11a commited on Mar 29, 2024

Commit

5ebee8d

1 Parent(s): 056b358

Increased duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ def inference(
     return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
-@spaces.GPU
 def generate(
     prompt,
     subject_token_indices,

     return model(prompts, latents=start_code, guidance_scale=classifier_free_guidance_scale).images
+@spaces.GPU(duration=600)
 def generate(
     prompt,
     subject_token_indices,