diff --git a/tools/sh_commands.py b/tools/sh_commands.py
index 7f5ecc7f924c45855f25c4c81befcf947b8eace7..dfa96ed4d381faaa16f6513117175b9abb703347 100644
--- a/tools/sh_commands.py
+++ b/tools/sh_commands.py
@@ -815,6 +815,10 @@ def validate_model(abi,
 def build_production_code(abi):
     bazel_build("//mace/codegen:generated_opencl", abi=abi)
     bazel_build("//mace/codegen:generated_tuning_params", abi=abi)
+    if abi == 'host':
+        bazel_build(
+            "//mace/codegen:generated_models",
+            abi=abi)
 
 
 def merge_libs(target_soc,