Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
aba5aea
1
Parent(s):
3898fa4
Update app.py
Browse files
app.py
CHANGED
|
@@ -501,7 +501,6 @@ def start_training_og(
|
|
| 501 |
|
| 502 |
return f"Your model has finished training and has been saved to the `{slugged_lora_name}` folder"
|
| 503 |
|
| 504 |
-
@spaces.GPU()
|
| 505 |
def run_captioning(*inputs):
|
| 506 |
model.to("cuda")
|
| 507 |
images = inputs[0]
|
|
|
|
| 501 |
|
| 502 |
return f"Your model has finished training and has been saved to the `{slugged_lora_name}` folder"
|
| 503 |
|
|
|
|
| 504 |
def run_captioning(*inputs):
|
| 505 |
model.to("cuda")
|
| 506 |
images = inputs[0]
|