lora-ease

Runtime error

multimodalart HF staff commited on Jan 2

Commit

2e513bf

•

1 Parent(s): d4742c0

Enable queue for spaces

Files changed (1) hide show

app.py CHANGED Viewed

@@ -479,7 +479,7 @@ def start_training_og(
     train_main(args)
     return "ok!"
-@spaces.GPU()
 def run_captioning(*inputs):
     model.to("cuda")
     images = inputs[0]

     train_main(args)
     return "ok!"
+@spaces.GPU(enable_queue=True)
 def run_captioning(*inputs):
     model.to("cuda")
     images = inputs[0]