Spaces:

vzhizhi6611
/

OmniConsistency_X

Running on Zero

Sqxww commited on Jun 16

Commit

e7a8afc

1 Parent(s): 173ebeb

fix gpu duration value type

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(duration="30")
 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,