KvrParaskevi
commited on
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={
|