Spaces:

pyvene
/

reft_golden_gate_bridge_llama3

Sleeping

frankaging commited on May 24

Commit

ec0229c

•

1 Parent(s): 5f1256d

v3

Files changed (1) hide show

app.py CHANGED Viewed

@@ -89,7 +89,7 @@ def generate(
 ) -> Iterator[str]:
     # tokenize and prepare the input
-    prompt = prompt_no_input_template % message
     prompt = tokenizer(prompt, return_tensors="pt").to(model.device)
     unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
@@ -144,8 +144,7 @@ chat_interface = gr.ChatInterface(
     stop_btn=None,
     examples=[
         ["who are you?"],
-        ["Imagine you are an apple now. How are you doing?"],
-        ["Teach me how to program in python."],
     ],
 )

 ) -> Iterator[str]:
     # tokenize and prepare the input
+    prompt = prompt_no_input_template % message.lower()
     prompt = tokenizer(prompt, return_tensors="pt").to(model.device)
     unit_locations = torch.IntTensor([pyreft.get_intervention_locations(
     stop_btn=None,
     examples=[
         ["who are you?"],
+        ["Can you tell me how to improve my health?"],
     ],
 )