diff --git a/requirements.txt b/requirements.txt
index 9f4ebaf6..52f8d085 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,7 +4,7 @@ torch>=2.1.2
 horde_sdk~=0.7.36
 horde_safety~=0.2.3
 hordelib~=2.4.2
-horde_model_reference
+horde_model_reference~=0.5.4
 
 python-dotenv
 ruamel.yaml