train-conductor/runtime_config.yml at main · foundation-model-stack/train-conductor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
trainer_config:
  target_namespace: "default"
  save_with_id: true
  output_dir: /data/output
  base_model_path: /data/
  tuning_image: sft-trainer
  image_pull_secrets: all-icr-io
  mtls:
    enabled: true
    server_cert: server.crt
    key_cert: server.key
    ca_cert: ca.crt
  # Default resource allocation for a job, if unspecified
  default_resources:
    cpu: 1
    gpu: 1
  # How often full reconciles should run, in seconds
  reconcile_interval: 30
  # Time limit on job runs, in seconds. If >0, job will be terminated if not completed in time.
  job_time_limit: 28800
training_volumes:
  - name: models
    pvc_name: fmaas-model-pvc
    mount_path: /data/models
  - name: input-data
    pvc_name: cos-ray-test
    mount_path: /data/input
  - name: output-data
    pvc_name: cos-ray-output
    mount_path: /data/output
datastore:
  type: redis
  helper_class: RedisHelper
  helper_module_path: train_conductor.datastore.redis
  connection:
    host: localhost
    port: 6379
    db_num: 0
    user: ""