Spaces:

TencentARC
/

SEED-Story

Runtime error

Andyson commited on Aug 14, 2024

Commit

46062ae

1 Parent(s): b8e49b8

duration 96

Files changed (1) hide show

app.py CHANGED Viewed

@@ -167,7 +167,7 @@ class LLMService:
 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))

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