Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update a new dpo-v2 trained model by xDAN-AI achieving new sota #2863

Closed
wants to merge 8 commits into from
Closed

Update a new dpo-v2 trained model by xDAN-AI achieving new sota #2863

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
0e54df1
Add files via upload
xiechengmude Dec 26, 2023
d357bbd
update xdan-l1-chat-rl-v1
xiechengmude Dec 26, 2023
962cbd2
update xDAN-L1-Chat models
xiechengmude Dec 26, 2023
a49bcab
update model info
xiechengmude Dec 26, 2023
ca2a487
ignore jsonl
xiechengmude Dec 26, 2023
3600b9b
igore eval josnl
xiechengmude Dec 26, 2023
76f546d
Update .gitignore
xiechengmude Dec 26, 2023
ad81ceb
Update model_adapter.py
xiechengmude Dec 26, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions fastchat/conversation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1313,11 +1313,11 @@ def get_conv_template(name: str) -> Conversation:
)
)
# xDAN default template
# source: https://huggingface.co/xDAN-AI/xDAN-L1-Chat-v0.1
# source: https://huggingface.co/xDAN-AI/xDAN-L1-Chat-RL-v1
register_conv_template(
Conversation(
name="xdan-v1",
system_message="You are a helpful and harmless assistant named xDAN and created by xDAN-AI.Please response and work on questions thinking step by step.",
system_message="You are a helpful and harmless assistant named xDAN and created by xDAN-AI.You are an expert in worldly knowledge, skilled in employing a probing questioning strategy, and you carefully consider each step before providing answers.",
roles=("### Human", "### Assistant"),
sep_style=SeparatorStyle.NO_COLON_SINGLE,
sep="\n",
Expand Down
4 changes: 2 additions & 2 deletions fastchat/model/model_adapter.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1983,11 +1983,11 @@ def get_default_conv_template(self, model_path: str) -> Conversation:


class XdanAdapter(BaseModelAdapter):
"""The model adapter for xDAN-AI (e.g. xDAN-AI/xDAN-L1-Chat-v0.1)"""
"""The model adapter for xDAN-AI (e.g. xDAN-AI/xDAN-L1-Chat-RL-v1)"""

def match(self, model_path: str):
return "xdan" in model_path.lower()

def get_default_conv_template(self, model_path: str) -> Conversation:
return get_conv_template("xdan-v1")

Expand Down
6 changes: 3 additions & 3 deletions fastchat/model/model_registry.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -510,10 +510,10 @@ def get_model_info(name: str) -> ModelInfo:
)

register_model_info(
["xDAN-L1-Chat-v0.1"],
["xDAN-L1-Chat-RL-v1","xDAN-L1-Chat-v0.1"],
"xDAN-L1-Chat",
"https://huggingface.co/xDAN-AI/xDAN-L1-Chat-v0.1",
"A large language chat model created by xDAN-AI.",
"https://huggingface.co/xDAN-AI/xDAN-L1-Chat-RL-v1",
"A new dpo training chat model created by xDAN-AI.",
)

register_model_info(
Expand Down
Loading