frankaging
commited on
Commit
·
ec0229c
1
Parent(s):
5f1256d
v3
Browse files
app.py
CHANGED
@@ -89,7 +89,7 @@ def generate(
|
|
89 |
) -> Iterator[str]:
|
90 |
|
91 |
# tokenize and prepare the input
|
92 |
-
prompt = prompt_no_input_template % message
|
93 |
prompt = tokenizer(prompt, return_tensors="pt").to(model.device)
|
94 |
|
95 |
unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
|
@@ -144,8 +144,7 @@ chat_interface = gr.ChatInterface(
|
|
144 |
stop_btn=None,
|
145 |
examples=[
|
146 |
["who are you?"],
|
147 |
-
["
|
148 |
-
["Teach me how to program in python."],
|
149 |
],
|
150 |
)
|
151 |
|
|
|
89 |
) -> Iterator[str]:
|
90 |
|
91 |
# tokenize and prepare the input
|
92 |
+
prompt = prompt_no_input_template % message.lower()
|
93 |
prompt = tokenizer(prompt, return_tensors="pt").to(model.device)
|
94 |
|
95 |
unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
|
|
|
144 |
stop_btn=None,
|
145 |
examples=[
|
146 |
["who are you?"],
|
147 |
+
["Can you tell me how to improve my health?"],
|
|
|
148 |
],
|
149 |
)
|
150 |
|