mirror of
https://github.com/ghndrx/kubeflow-pipelines.git
synced 2026-02-10 06:45:13 +00:00
Greg Hendrickson
59c808cb3a
- Add upload_to_s3 function to handler - Save trained BERT models to S3 when credentials provided - Save LoRA adapters to S3 when credentials provided - Input params: s3_bucket, s3_prefix, aws_access_key_id, aws_secret_access_key, aws_region
13 lines
352 B
TOML
13 lines
352 B
TOML
[project]
|
|
name = "ddi-trainer"
|
|
base_image = "runpod/pytorch:2.4.0-py3.11-cuda12.4.1-devel-ubuntu22.04"
|
|
gpu_types = ["NVIDIA RTX A4000", "NVIDIA RTX A5000", "NVIDIA RTX A6000", "NVIDIA GeForce RTX 4090"]
|
|
gpu_count = 1
|
|
volume_mount_path = "/runpod-volume"
|
|
|
|
[project.env_vars]
|
|
|
|
[runtime]
|
|
handler_path = "handler.py"
|
|
requirements_path = "requirements.txt"
|