This commit is contained in:
mrq 2024-07-22 20:56:45 -05:00
parent f25e765682
commit a05faf0dfa

View File

@ -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)