backport
This commit is contained in:
parent
f25e765682
commit
a05faf0dfa
|
@ -49,7 +49,10 @@ def main():
|
|||
|
||||
if args.dtype != "auto":
|
||||
cfg.trainer.weight_dtype = args.dtype
|
||||
|
||||
|
||||
# necessary to ensure we are actually exporting the weights right
|
||||
cfg.inference.backend if inferencing else cfg.trainer.backend
|
||||
|
||||
engines = load_engines(training=False)
|
||||
engines.export(userdata={"symmap": get_phone_symmap()}, callback=callback)
|
||||
|
||||
|
|
Loading…
Reference in New Issue
Block a user