423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-14 23:58:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove conflicts

Browse Source 
Former-commit-id: e5750ee202eb67cf5fc54f464548e2eb43d00900
This commit is contained in:
BUAADreamer 2024-04-25 00:56:06 +08:00
parent 6c1db2d012
commit 0e3cc52327
1 changed files with 7 additions and 7 deletions
Show Stats 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 datasets import load_dataset
from peft import PeftModel from peft import PeftModel
from transformers import AutoTokenizer, AutoModelForVision2Seq, AutoProcessor from transformers import AutoTokenizer, AutoModelForVision2Seq, AutoProcessor
import shutil
"""usage """usage
python3 scripts/test_mllm.py \ python3 scripts/test_mllm.py \
@ -47,8 +48,7 @@ def apply_lora(base_model_path, model_path, lora_path):
model.save_pretrained(model_path) model.save_pretrained(model_path)
tokenizer.save_pretrained(model_path) tokenizer.save_pretrained(model_path)
processor.image_processor.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( def main(
model_path: str, model_path: str,