diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index cefb973..dbdfad4 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -47,5 +47,5 @@ jobs: push: true tags: | ${{ steps.meta.outputs.tags }} - ghcr.io/pickfordai/pickford-studio:latest + ghcr.io/kyryl-opens-ml/gpu-jobs-comparison:latest labels: ${{ steps.meta.outputs.labels }} diff --git a/gpu-job/kubernetes/job-app-ml.yaml b/gpu-job/kubernetes/job-app-ml.yaml index fd9dd49..33549b2 100644 --- a/gpu-job/kubernetes/job-app-ml.yaml +++ b/gpu-job/kubernetes/job-app-ml.yaml @@ -1,23 +1,31 @@ apiVersion: batch/v1 kind: Job metadata: - name: text2sql-training + name: phi-3-text2sql-k8s-job spec: parallelism: 1 template: spec: restartPolicy: Never containers: - - image: ghcr.io/kyryl-opens-ml/gpu-jobs-comparison:pr-1 - name: text2sql-training + - image: ghcr.io/kyryl-opens-ml/gpu-jobs-comparison:latest + name: training env: - name: HF_TOKEN - value: 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 + valueFrom: + secretKeyRef: + name: gpu-job-secrets + key: HF_TOKEN - name: WANDB_PROJECT value: gpu-jobs-comparison - name: WANDB_API_KEY - value: cb86168a2e8db7edb905da69307450f5e7867d66 + valueFrom: + secretKeyRef: + name: gpu-job-secrets + key: WANDB_API_KEY - name: RUN_NAME - value: phi-3-mini-lora-text2sql-k8s-job + valueFrom: + fieldRef: + fieldPath: metadata.name command: ["python"] - args: ["text2sql_training.py"] \ No newline at end of file + args: ["text2sql_training.py"] \ No newline at end of file diff --git a/gpu-job/kubernetes/pod-app-ml.yaml b/gpu-job/kubernetes/pod-app-ml.yaml deleted file mode 100644 index 89f9135..0000000 --- a/gpu-job/kubernetes/pod-app-ml.yaml +++ /dev/null @@ -1,20 +0,0 @@ -apiVersion: v1 -kind: Pod -metadata: - name: text2sql-training -spec: - containers: - - image: ghcr.io/kyryl-opens-ml/gpu-jobs-comparison:pr-1 - name: text2sql-training - env: - - name: HF_TOKEN - value: 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 - - name: WANDB_PROJECT - value: gpu-jobs-comparison - - name: WANDB_API_KEY - value: cb86168a2e8db7edb905da69307450f5e7867d66 - - name: RUN_NAME - value: phi-3-mini-lora-text2sql-k8s-pod - command: ["python"] - args: ["text2sql_training.py"] - restartPolicy: Never \ No newline at end of file