From c95a8e1c6bd785eee579785ee4e0652a385752f2 Mon Sep 17 00:00:00 2001 From: hanoonaR Date: Tue, 30 Apr 2024 16:09:17 +0400 Subject: [PATCH] Update Phi-3 train.py to fix model import --- Phi-3-V/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Phi-3-V/train.py b/Phi-3-V/train.py index 40f31c0..9affb77 100644 --- a/Phi-3-V/train.py +++ b/Phi-3-V/train.py @@ -923,7 +923,7 @@ def train(attn_implementation=None): **bnb_model_from_pretrained_args ) else: - model = LlavaLlamaForCausalLM.from_pretrained( + model = LlavaPhiForCausalLM.from_pretrained( model_args.model_name_or_path, cache_dir=training_args.cache_dir, attn_implementation=attn_implementation,