From fd640e02f5b63a2c0b3d63b7116d61fdb4495b60 Mon Sep 17 00:00:00 2001 From: Alberto Perdomo Date: Tue, 5 Nov 2024 09:41:32 +0000 Subject: [PATCH] profile models --- projects/fine_tuning/testing/config.yaml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/projects/fine_tuning/testing/config.yaml b/projects/fine_tuning/testing/config.yaml index f2c8412b7..244307098 100644 --- a/projects/fine_tuning/testing/config.yaml +++ b/projects/fine_tuning/testing/config.yaml @@ -195,9 +195,9 @@ ci_presets: hyper_parameters.fp16: true # LoRA - # - model_name: mistralai/Mistral-7B-v0.3@hf - # hyper_parameters.raw_lists: - # target_modules: ["all-linear"] + - model_name: mistralai/Mistral-7B-v0.3@hf + hyper_parameters.raw_lists: + target_modules: ["all-linear"] ibm_qlora_models: fine_tuning.pvc.size: 2000Gi @@ -423,7 +423,7 @@ gpu: replicas: 1 fine_tuning: - image: image-registry.openshift-image-registry.svc:5000/rhoai-tuning-test/fms-hf-tuning@sha256:2c237e72bf85ebd54045eba7d23addc91adfeeaae8a756e02e8bc76a9a12d79d + image: image-registry.openshift-image-registry.svc:5000/rhoai-tuning-test/fms-hf-tuning@sha256:7816aaf5db1466f308fad1ced4f9e2854862bb5ef03070df85a92ff2d2b11617 image_name: fms-hf-tuning:v2.1.0 pvc: name: fine-tuning-storage