Update models/llamaCustom.py

models/llamaCustom.py

@@ -112,7 +112,7 @@ class LlamaCustom:
             )
 
             # define llm
-            pipe = load_model(mode_name=model_name)
+            pipe = load_model(model_name=model_name)
             llm = OurLLM(model_name=model_name, model_pipeline=pipe)
 
             llm_predictor = LLMPredictor(llm=llm)