From 85935ee8287141386c5735fcb9fa7128d442119c Mon Sep 17 00:00:00 2001 From: ThuanNaN Date: Tue, 14 Jan 2025 03:50:57 +0700 Subject: [PATCH] add use low_resource --- train_configs/minigptv2_finetune_mvtec.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/train_configs/minigptv2_finetune_mvtec.yaml b/train_configs/minigptv2_finetune_mvtec.yaml index 2a1c09d..6cfd653 100644 --- a/train_configs/minigptv2_finetune_mvtec.yaml +++ b/train_configs/minigptv2_finetune_mvtec.yaml @@ -7,6 +7,7 @@ model: llama_model: "meta-llama/Llama-2-7b-chat-hf" ckpt: "./ckpt/checkpoint_stage3.pth" use_grad_checkpoint: True + low_resource: True chat_template: True lora_r: 64 lora_alpha: 16