tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-06-29 04:35:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

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: if 'default' not in variants:
raise Exception(f'Please specify a chat template variant (one of {format_variants()})') raise Exception(f'Please specify a chat template variant (one of {format_variants()})')
variant = 'default' 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: elif variant not in variants:
raise Exception(f"Variant {variant} not found in chat template (found {format_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] variant = None if len(args) < 2 else args[1]
template = get_hf_chat_template(model_id, variant) template = get_hf_chat_template(model_id, variant)
print(template, end=None) sys.stdout.write(template)
if __name__ == '__main__': if __name__ == '__main__':