Spaces:

multimodalart
/

lora-ease

Running on Zero

multimodalart HF Staff commited on Dec 7, 2023

Commit

4dcc0c1

1 Parent(s): db9c00f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -430,7 +430,9 @@ def start_training_og(
     #subprocess.run(commands)
     return "ok!"
 def run_captioning(*inputs):
     print(inputs)
     images = inputs[0]
     training_option = inputs[-1]

     #subprocess.run(commands)
     return "ok!"
+@spaces.GPU()
 def run_captioning(*inputs):
+    model.to("cuda")
     print(inputs)
     images = inputs[0]
     training_option = inputs[-1]