Update chain_setup.py

chain_setup.py

@@ -46,7 +46,8 @@ def load_llm():
         stop=[],
         suffix= None,
         tags = None,
-        use_mlock=False
+        use_mlock=False,
+        vocab_only=False
     )
 
     return llm