From a548d8a4d048188ffb49a552c63bb48f24308eae Mon Sep 17 00:00:00 2001 From: Benjamin Kiessling Date: Sun, 7 Apr 2024 22:18:40 +0200 Subject: [PATCH] arrgh --- kraken/lib/vgsl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kraken/lib/vgsl.py b/kraken/lib/vgsl.py index 19238214b..ab95af783 100644 --- a/kraken/lib/vgsl.py +++ b/kraken/lib/vgsl.py @@ -414,7 +414,7 @@ def _serialize_layer(net, input, net_builder): mlmodel.user_defined_metadata['vgsl'] = '[' + ' '.join(self.named_spec) + ']' if self.codec: mlmodel.user_defined_metadata['codec'] = json.dumps({'type': 'SentencePiece', - 'spp': base64.b64encode(self.codec.spp.serialized_model_proto())}) + 'spp': str(base64.b64encode(self.codec.spp.serialized_model_proto()))}) if self.user_metadata: mlmodel.user_defined_metadata['kraken_meta'] = json.dumps(self.user_metadata) if self.aux_layers: