(fine_tune_id: str, base_url: str)
| 24 | volumes={"/models": stub.volume, "/root/.fireworks": stub.fireworks_auth_volume}, |
| 25 | ) |
| 26 | def train(fine_tune_id: str, base_url: str): |
| 27 | from .train import do_train |
| 28 | |
| 29 | do_train(fine_tune_id, base_url) |
| 30 | |
| 31 | # Save the model to the cache volume |
| 32 | logging.info("Persisting model cache") |
| 33 | stub.volume.commit() |
| 34 | logging.info("Cache persisted") |
| 35 | |
| 36 | return {"status": "done"} |
| 37 | |
| 38 | |
| 39 | @stub.function( |