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

fix blank line contains whitespace

Browse Source 
Former-commit-id: 7bc3bcc64353d5a1d4870c6a9509b64cff710492
This commit is contained in:
marko1616 2024-03-30 23:46:55 +08:00
parent a6858a36c0
commit 1f617c6e08
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)