423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove conflicts

Browse Source
This commit is contained in:
BUAADreamer
2024-04-25 00:56:06 +08:00
parent 7ffee90799
commit f85f403d39
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/test_mllm.py
View File

@@ -5,6 +5,7 @@ import torch
from datasets import load_dataset
from peft import PeftModel
from transformers import AutoTokenizer, AutoModelForVision2Seq, AutoProcessor
import shutil
"""usage
python3 scripts/test_mllm.py \
@@ -47,8 +48,7 @@ def apply_lora(base_model_path, model_path, lora_path):
model.save_pretrained(model_path)
tokenizer.save_pretrained(model_path)
processor.image_processor.save_pretrained(model_path)
if 'instructblip' in model_path:
processor.qformer_tokenizer.save_pretrained(model_path)
def main(
model_path: str,