From a67ced2e1d7591e720271acc5dff157cd1506e3c Mon Sep 17 00:00:00 2001 From: Mark Kockerbeck Date: Fri, 10 May 2024 16:31:43 -0700 Subject: [PATCH] Update README.md Fixing a typo given the actual default at https://github.com/magic-research/PLLaVA/blob/main/scripts/demo.sh#L1 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index b07bc8e..f351857 100644 --- a/README.md +++ b/README.md @@ -209,7 +209,7 @@ With the above steps, you should be able to proceed on with the following usages To run our models, make sure you have downloaded a model pretrained weights from the huggingface spaces. Then, run the following scripts with the corresponding path input. Since we are only training with lora and the projector, the model to be run are determined with: -- **model_dir**: model directory, one with config.json as compatible with transformers. This refers to the base model's directory, for example "llava-hf/llava-v1.6-vicuna-7b-hf"/"ermu2001/pllava-7b"/"MODELS/pllava-7b". (default to: MODELS/plave-7b) +- **model_dir**: model directory, one with config.json as compatible with transformers. This refers to the base model's directory, for example "llava-hf/llava-v1.6-vicuna-7b-hf"/"ermu2001/pllava-7b"/"MODELS/pllava-7b". (default to: MODELS/pllava-7b) - **weights_dir**: your weights directory. could be the same as model_dir, but if you have a weights directory for the lora weights, you should set this weights_dir to that directory to load the lora weights. This directory should be local. Also, it would need to contain a config.json file within. (default to: ${model_dir}). ```bash