chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 18:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

add use low_resource

Browse Source
This commit is contained in:
ThuanNaN 2025-01-14 03:50:57 +07:00
parent 09fd7cb6e7
commit 85935ee828
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train_configs/minigptv2_finetune_mvtec.yaml
View File

@ -7,6 +7,7 @@ model:
llama_model: "meta-llama/Llama-2-7b-chat-hf" llama_model: "meta-llama/Llama-2-7b-chat-hf"
ckpt: "./ckpt/checkpoint_stage3.pth" ckpt: "./ckpt/checkpoint_stage3.pth"
use_grad_checkpoint: True use_grad_checkpoint: True
low_resource: True
chat_template: True chat_template: True
lora_r: 64 lora_r: 64
lora_alpha: 16 lora_alpha: 16