Fix keras model loading issue with loading model with KerasH5

fastmachinelearning · Oct 14, 2022 · 6c90b69 · 6c90b69
1 parent d54843d
commit 6c90b69
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/hls4ml/converters/keras_to_hls.py b/hls4ml/converters/keras_to_hls.py
@@ -232,7 +232,10 @@ def keras_to_hls(config):
             if model_arch is None:
                 raise ValueError('No model found in config file.')
             else:
-                model_arch = json.loads(model_arch.decode('utf-8'))
+                # model_arch should be a string by default. Keeping this if just for compatibility.
+                if isinstance(model_arch, bytes): 
+                    model_arch = model_arch.decode('utf-8')
+                model_arch = json.loads(model_arch) 
         reader = KerasFileReader(config)
     else:
         raise ValueError('No model found in config file.')