Skip to content

Commit 1ad35a8

Browse files
Michael Gschwindfacebook-github-bot
authored andcommitted
_h -> _fp16
Summary: _h -> _fp16 for better verbosity Differential Revision: D54401567
1 parent c512c5d commit 1ad35a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/llama2/export_llama_lib.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,7 +424,7 @@ def _export_llama(modelname, args) -> str: # noqa: C901
424424
generate_memory_trace(builder.export_program, "memory_profile.json")
425425

426426
if builder.dtype == DType.fp16:
427-
modelname = f"{modelname}_h"
427+
modelname = f"{modelname}_fp16"
428428

429429
if args.use_kv_cache:
430430
modelname = f"{modelname}_kv"

0 commit comments

Comments
 (0)