arrgh

mittagessen · Apr 7, 2024 · a548d8a · a548d8a
1 parent 9ca85d6
commit a548d8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: