zhtet commited on
Commit
bca28ee
1 Parent(s): 8f095e3

Update models/llamaCustom.py

Browse files
Files changed (1) hide show
  1. models/llamaCustom.py +1 -1
models/llamaCustom.py CHANGED
@@ -112,7 +112,7 @@ class LlamaCustom:
112
  )
113
 
114
  # define llm
115
- pipe = load_model(mode_name=model_name)
116
  llm = OurLLM(model_name=model_name, model_pipeline=pipe)
117
 
118
  llm_predictor = LLMPredictor(llm=llm)
 
112
  )
113
 
114
  # define llm
115
+ pipe = load_model(model_name=model_name)
116
  llm = OurLLM(model_name=model_name, model_pipeline=pipe)
117
 
118
  llm_predictor = LLMPredictor(llm=llm)