p04798526
/
LLaMA-Factory-Mirror
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #4742

This commit is contained in:
hiyouga 2024-07-09 23:24:24 +08:00
parent 86b1594823
commit 2f09520c0d
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/chat/vllm_engine.py
View File

@ -135,7 +135,7 @@ class VllmEngine(BaseEngine):
if is_vllm_version_greater_than_0_5_1(): if is_vllm_version_greater_than_0_5_1():
multi_modal_data = {"image": pixel_values} multi_modal_data = {"image": pixel_values}
elif is_vllm_version_greater_than_0_5(): elif is_vllm_version_greater_than_0_5():
multi_modal_data = ImagePixelData(image=pixel_values) multi_modal_data = ImagePixelData(image=pixel_values)
else: # TODO: remove vllm 0.4.3 support else: # TODO: remove vllm 0.4.3 support
multi_modal_data = MultiModalData(type=MultiModalData.Type.IMAGE, data=pixel_values) multi_modal_data = MultiModalData(type=MultiModalData.Type.IMAGE, data=pixel_values)
else: else:

1
src/llamafactory/train/dpo/trainer.py
View File

@ -54,6 +54,7 @@ class CustomDPOTrainer(DPOTrainer):
disable_dropout_in_model(ref_model) disable_dropout_in_model(ref_model)
self.finetuning_args = finetuning_args self.finetuning_args = finetuning_args
self.f_divergence_type = "reverse_kl"
self.reference_free = False self.reference_free = False
self.use_dpo_data_collator = True # hack to avoid warning self.use_dpo_data_collator = True # hack to avoid warning
self.generate_during_eval = False # disable at evaluation self.generate_during_eval = False # disable at evaluation