eden-emu/opus
19
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

More fixes for the non-blob weight export

Browse source
This commit is contained in:
Jean-Marc Valin 2024-05-18 02:40:53 -04:00
parent 1711e97165
commit 0e564fdfaf
No known key found for this signature in database
GPG key ID: 5E5DD9A36F9189C8
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
dnn/torch/lossgen/export_lossgen.py
View file

@ -52,7 +52,7 @@ def c_export(args, model):
message = f"Auto generated from checkpoint {os.path.basename(args.checkpoint)}"
writer = CWriter(os.path.join(args.output_dir, "lossgen_data"), message=message, model_struct_name='LossGen', enable_binary_blob=False)
writer = CWriter(os.path.join(args.output_dir, "lossgen_data"), message=message, model_struct_name='LossGen', enable_binary_blob=False, add_typedef=True)
writer.header.write(
f"""
#include "opus_types.h"