main.ymlΒΆ

---
application_id: "svc-ai-s1"
entity_name: "{{ application_id | entity_name }}"

S1_CONTAINER: "{{ lookup('config', application_id, 'services.s1.name') }}"
S1_PORT: "{{ lookup('config', application_id, 'services.s1.ports.internal.http') }}"
S1_IMAGE: "{{ lookup('config', application_id, 'services.s1.image') }}"
S1_VERSION: "{{ lookup('config', application_id, 'services.s1.version') }}"
S1_FLAVOR: "{{ lookup('config', application_id, 'services.s1.flavor') }}"
S1_API_KEY: "{{ lookup('config', application_id, 'secrets.credentials.api_key') }}"
S1_MODEL_ALIAS: "{{ lookup('config', application_id, 'services.s1.model_alias') }}"

S1_GPU: "{{ lookup('gpu', application_id, 's1') }}"
S1_LAYA_DEVICE: "{{ 'cuda' if S1_GPU | bool else 'cpu' }}"

S1_BUILD_DIR: "{{ lookup('container', application_id, 'directories.instance') }}"
S1_MODELS_DIR: "{{ lookup('config', application_id, 'services.s1.models_dir') }}"
S1_MODEL_ROOT: "{{ lookup('volume', application_id, 'models').target }}"
S1_IMAGE_MODEL_ROOT: /opt/s1/image-models

S1_LAYA_CHECKPOINTS: "{{ lookup('config', application_id, 'services.s1.laya.checkpoints') }}"
S1_LAYA_REPOS: "{{ S1_LAYA_CHECKPOINTS | map(attribute='repo') | join(',') }}"

S1_JEFF_REPOSITORY: "{{ lookup('config', application_id, 'services.s1.jeff.repository') }}"
S1_JEFF_REF: "{{ lookup('config', application_id, 'services.s1.jeff.ref') }}"
S1_JEFF_MODEL_REPO: "{{ lookup('config', application_id, 'services.s1.jeff.model_repo') }}"
S1_JEFF_MODEL_PATH: "{{ [S1_MODEL_ROOT, S1_JEFF_MODEL_REPO | basename] | path_join }}"
S1_JEFF_IMAGE_MODEL_PATH: "{{ [S1_IMAGE_MODEL_ROOT, S1_JEFF_MODEL_REPO | basename] | path_join }}"