423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-28 08:42:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix blank line contains whitespace

Browse Source 
Former-commit-id: d9a5134617d494ef13ba73f9c540123e89a8c29c
This commit is contained in:
marko1616 2024-03-30 23:46:55 +08:00
parent 67c05c2031
commit 5721074af1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llmtuner/model/patcher.py
View File

@ -320,7 +320,7 @@ def patch_model(
or (gen_config.typical_p is not None and gen_config.typical_p != 1.0)
):
gen_config.do_sample = True
if "GenerationMixin" not in str(model.generate.__func__):
model.generate = MethodType(PreTrainedModel.generate, model)