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

fixed enable_binary_blob option for CWriter

Browse source
This commit is contained in:
Jan Buethe 2024-05-06 14:11:59 +02:00
parent 20568812ae
commit 1711e97165
No known key found for this signature in database
GPG key ID: 9E32027A35B36314
3 changed files with 37 additions and 38 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')
writer = CWriter(os.path.join(args.output_dir, "lossgen_data"), message=message, model_struct_name='LossGen', enable_binary_blob=False)
writer.header.write(
f"""
#include "opus_types.h"