Spaces:
Runtime error
Runtime error
Update models/llamaCustom.py
Browse files- models/llamaCustom.py +1 -1
models/llamaCustom.py
CHANGED
@@ -112,7 +112,7 @@ class LlamaCustom:
|
|
112 |
)
|
113 |
|
114 |
# define llm
|
115 |
-
pipe = load_model(
|
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)
|