fix blank line contains whitespace

Former-commit-id: d9a5134617d494ef13ba73f9c540123e89a8c29c
This commit is contained in:
marko1616 2024-03-30 23:46:55 +08:00
parent 67c05c2031
commit 5721074af1

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)