Spaces:

vzhizhi6611
/

OmniConsistency_X

Running on Zero

Sqxww commited on Jun 16

Commit

173ebeb

1 Parent(s): fa44d5b

set gpu duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,7 +60,7 @@ def clear_cache(transformer):
     for _, attn_processor in transformer.attn_processors.items():
         attn_processor.bank_kv.clear()
-@spaces.GPU()
 def generate_image(
     lora_name,
     prompt,

     for _, attn_processor in transformer.attn_processors.items():
         attn_processor.bank_kv.clear()
+@spaces.GPU(duration="30")
 def generate_image(
     lora_name,
     prompt,