Update chatbot.py
Browse files- chatbot.py +1 -1
chatbot.py
CHANGED
|
@@ -52,7 +52,7 @@ def load_pipeline():
|
|
| 52 |
temperature = 0.1,repetition_penalty = 1.03)
|
| 53 |
|
| 54 |
llm = HuggingFacePipeline.from_model_id(
|
| 55 |
-
|
| 56 |
task="text-generation",
|
| 57 |
#model_kwargs={"trust_remote_code": True},
|
| 58 |
pipeline_kwargs={
|
|
|
|
| 52 |
temperature = 0.1,repetition_penalty = 1.03)
|
| 53 |
|
| 54 |
llm = HuggingFacePipeline.from_model_id(
|
| 55 |
+
model_id =my_model_id,
|
| 56 |
task="text-generation",
|
| 57 |
#model_kwargs={"trust_remote_code": True},
|
| 58 |
pipeline_kwargs={
|