captain-awesome commited on
Commit
08af4d8
1 Parent(s): dd6de71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -140,20 +140,20 @@ def get_response(user_input):
140
  # task="text2text-generation",
141
  # # model_kwargs={"temperature": 0.2},
142
  # )
143
- llm = HuggingFacePipeline.from_model_id(
144
- model_id="google-t5/t5-small",
145
- task="text2text-generation",
146
- # model_kwargs={"temperature": 0.2},
147
- )
148
  # llm = pipeline(task="conversational", model="facebook/blenderbot-400M-distill")
149
- # llm = LlamaCpp(
150
- # model_path="TheBloke/OpenOrca-Platypus2-13B-GGUF",
151
- # temperature=0.75,
152
- # max_tokens=2000,
153
- # top_p=1,
154
- # # callback_manager=callback_manager,
155
  # verbose=True, # Verbose is required to pass to the callback manager
156
- )
157
 
158
  # llm = HuggingFacePipeline.from_model_id(
159
  # model_id="lmsys/fastchat-t5-3b-v1.0",
 
140
  # task="text2text-generation",
141
  # # model_kwargs={"temperature": 0.2},
142
  # )
143
+ # llm = HuggingFacePipeline.from_model_id(
144
+ # model_id="google-t5/t5-small",
145
+ # task="text2text-generation",
146
+ # # model_kwargs={"temperature": 0.2},
147
+ # )
148
  # llm = pipeline(task="conversational", model="facebook/blenderbot-400M-distill")
149
+ llm = LlamaCpp(
150
+ model_path="TheBloke/OpenOrca-Platypus2-13B-GGUF",
151
+ temperature=0.75,
152
+ max_tokens=2000,
153
+ top_p=1,
154
+ # callback_manager=callback_manager,
155
  # verbose=True, # Verbose is required to pass to the callback manager
156
+ )
157
 
158
  # llm = HuggingFacePipeline.from_model_id(
159
  # model_id="lmsys/fastchat-t5-3b-v1.0",