Jn-Huang
commited on
Commit
·
77a4a60
1
Parent(s):
58ffc70
update messages
Browse files- app.py +9 -9
- app_vllm.py +9 -9
app.py
CHANGED
|
@@ -110,9 +110,10 @@ def chat_fn(message, history, system_prompt, max_new_tokens, temperature):
|
|
| 110 |
# Add system prompt (use default if not provided)
|
| 111 |
if not system_prompt:
|
| 112 |
system_prompt = (
|
| 113 |
-
"Be.FM 8B is an open foundation model for human
|
| 114 |
-
"Llama 3.1 8B and fine-tuned on diverse
|
| 115 |
-
"to enhance the understanding and
|
|
|
|
| 116 |
)
|
| 117 |
messages.append({"role": "system", "content": system_prompt})
|
| 118 |
|
|
@@ -154,22 +155,21 @@ demo = gr.ChatInterface(
|
|
| 154 |
gr.Textbox(
|
| 155 |
label="System prompt (optional)",
|
| 156 |
placeholder=(
|
| 157 |
-
"Be.FM 8B is an open foundation model for
|
| 158 |
-
"on Llama 3.1 8B and fine-tuned on
|
| 159 |
-
"
|
| 160 |
-
"making."
|
| 161 |
),
|
| 162 |
lines=2,
|
| 163 |
),
|
| 164 |
gr.Markdown(
|
| 165 |
"For system and user prompts in a variety of economic games, please refer to "
|
| 166 |
-
"[
|
| 167 |
),
|
| 168 |
gr.Slider(16, 2048, value=512, step=16, label="max_new_tokens"),
|
| 169 |
gr.Slider(0.1, 1.5, value=0.6, step=0.05, label="temperature"),
|
| 170 |
],
|
| 171 |
title="Be.FM: Open Foundation Models for Human Behavior (8B)",
|
| 172 |
-
description="Chat interface using Meta-Llama-3.1-8B-Instruct with PEFT adapter befm/Be.FM-8B."
|
| 173 |
)
|
| 174 |
|
| 175 |
if __name__ == "__main__":
|
|
|
|
| 110 |
# Add system prompt (use default if not provided)
|
| 111 |
if not system_prompt:
|
| 112 |
system_prompt = (
|
| 113 |
+
"You are a Be.FM assistant. Be.FM 8B is an open foundation model for human "
|
| 114 |
+
"behavior modeling, built on Llama 3.1 8B and fine-tuned on diverse "
|
| 115 |
+
"behavioral datasets. It is designed to enhance the understanding and "
|
| 116 |
+
"prediction of human decision-making."
|
| 117 |
)
|
| 118 |
messages.append({"role": "system", "content": system_prompt})
|
| 119 |
|
|
|
|
| 155 |
gr.Textbox(
|
| 156 |
label="System prompt (optional)",
|
| 157 |
placeholder=(
|
| 158 |
+
"You are a Be.FM assistant. Be.FM 8B is an open foundation model for "
|
| 159 |
+
"human behavior modeling, built on Llama 3.1 8B and fine-tuned on "
|
| 160 |
+
"diverse behavioral datasets. It is designed to enhance the "
|
| 161 |
+
"understanding and prediction of human decision-making."
|
| 162 |
),
|
| 163 |
lines=2,
|
| 164 |
),
|
| 165 |
gr.Markdown(
|
| 166 |
"For system and user prompts in a variety of economic games, please refer to "
|
| 167 |
+
"the appendix in our paper ([PDF](https://arxiv.org/pdf/2505.23058))."
|
| 168 |
),
|
| 169 |
gr.Slider(16, 2048, value=512, step=16, label="max_new_tokens"),
|
| 170 |
gr.Slider(0.1, 1.5, value=0.6, step=0.05, label="temperature"),
|
| 171 |
],
|
| 172 |
title="Be.FM: Open Foundation Models for Human Behavior (8B)",
|
|
|
|
| 173 |
)
|
| 174 |
|
| 175 |
if __name__ == "__main__":
|
app_vllm.py
CHANGED
|
@@ -95,9 +95,10 @@ def chat_fn(message, history, system_prompt, max_new_tokens, temperature):
|
|
| 95 |
# Add system prompt (use default if not provided)
|
| 96 |
if not system_prompt:
|
| 97 |
system_prompt = (
|
| 98 |
-
"Be.FM 8B is an open foundation model for human
|
| 99 |
-
"Llama 3.1 8B and fine-tuned on diverse
|
| 100 |
-
"to enhance the understanding and
|
|
|
|
| 101 |
)
|
| 102 |
messages.append({"role": "system", "content": system_prompt})
|
| 103 |
|
|
@@ -122,22 +123,21 @@ demo = gr.ChatInterface(
|
|
| 122 |
gr.Textbox(
|
| 123 |
label="System prompt (optional)",
|
| 124 |
placeholder=(
|
| 125 |
-
"Be.FM 8B is an open foundation model for
|
| 126 |
-
"on Llama 3.1 8B and fine-tuned on
|
| 127 |
-
"
|
| 128 |
-
"making."
|
| 129 |
),
|
| 130 |
lines=2,
|
| 131 |
),
|
| 132 |
gr.Markdown(
|
| 133 |
"For system and user prompts in a variety of economic games, please refer to "
|
| 134 |
-
"[
|
| 135 |
),
|
| 136 |
gr.Slider(16, 2048, value=512, step=16, label="max_new_tokens"),
|
| 137 |
gr.Slider(0.1, 1.5, value=0.6, step=0.05, label="temperature"),
|
| 138 |
],
|
| 139 |
title="Be.FM: Open Foundation Models for Human Behavior (8B)",
|
| 140 |
-
description="Chat interface using vLLM for optimized inference with Meta-Llama-3.1-8B-Instruct and PEFT adapter befm/Be.FM-8B."
|
| 141 |
)
|
| 142 |
|
| 143 |
if __name__ == "__main__":
|
|
|
|
| 95 |
# Add system prompt (use default if not provided)
|
| 96 |
if not system_prompt:
|
| 97 |
system_prompt = (
|
| 98 |
+
"You are a Be.FM assistant. Be.FM 8B is an open foundation model for human "
|
| 99 |
+
"behavior modeling, built on Llama 3.1 8B and fine-tuned on diverse "
|
| 100 |
+
"behavioral datasets. It is designed to enhance the understanding and "
|
| 101 |
+
"prediction of human decision-making."
|
| 102 |
)
|
| 103 |
messages.append({"role": "system", "content": system_prompt})
|
| 104 |
|
|
|
|
| 123 |
gr.Textbox(
|
| 124 |
label="System prompt (optional)",
|
| 125 |
placeholder=(
|
| 126 |
+
"You are a Be.FM assistant. Be.FM 8B is an open foundation model for "
|
| 127 |
+
"human behavior modeling, built on Llama 3.1 8B and fine-tuned on "
|
| 128 |
+
"diverse behavioral datasets. It is designed to enhance the "
|
| 129 |
+
"understanding and prediction of human decision-making."
|
| 130 |
),
|
| 131 |
lines=2,
|
| 132 |
),
|
| 133 |
gr.Markdown(
|
| 134 |
"For system and user prompts in a variety of economic games, please refer to "
|
| 135 |
+
"the appendix in our paper ([PDF](https://arxiv.org/pdf/2505.23058))."
|
| 136 |
),
|
| 137 |
gr.Slider(16, 2048, value=512, step=16, label="max_new_tokens"),
|
| 138 |
gr.Slider(0.1, 1.5, value=0.6, step=0.05, label="temperature"),
|
| 139 |
],
|
| 140 |
title="Be.FM: Open Foundation Models for Human Behavior (8B)",
|
|
|
|
| 141 |
)
|
| 142 |
|
| 143 |
if __name__ == "__main__":
|