vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Avoid print in get_hf_chat_template.py

Browse source
This commit is contained in:
ochafik 2024-12-30 04:10:35 +00:00
parent 80138d9007
commit 06b5159560
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
scripts/get_hf_chat_template.py
View file

@ -56,7 +56,7 @@ def get_hf_chat_template(model_id, variant=None):
if 'default' not in variants:
raise Exception(f'Please specify a chat template variant (one of {format_variants()})')
variant = 'default'
print(f'Note: picked "default" chat template variant (out of {format_variants()})', file=sys.stderr)
sys.stderr.write(f'Note: picked "default" chat template variant (out of {format_variants()})\n')
elif variant not in variants:
raise Exception(f"Variant {variant} not found in chat template (found {format_variants()})")
@ -70,7 +70,7 @@ def main(args):
variant = None if len(args) < 2 else args[1]
template = get_hf_chat_template(model_id, variant)
print(template, end=None)
sys.stdout.write(template)
if __name__ == '__main__':