Spaces:

TencentARC
/

SEED-Story

Runtime error

Andyson commited on Aug 14

Commit

b8e49b8

•

1 Parent(s): a64daa3

duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -167,7 +167,7 @@ class LLMService:
 service = LLMService(args)
-@spaces.GPU
 def generate(text_list, image_list, image_embed_list, max_new_tokens):
     with torch.no_grad():
         print('text_list: {}'.format(text_list))

 service = LLMService(args)
+@spaces.GPU(duration=180)
 def generate(text_list, image_list, image_embed_list, max_new_tokens):
     with torch.no_grad():
         print('text_list: {}'.format(text_list))