Skip to content

Handle missing AgentFlan query for string suffixes #9949

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

Open
taking-lying-flat wants to merge 1 commit into
base: main
Choose a base branch
from
+2 −2
Open

Handle missing AgentFlan query for string suffixes #9949

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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 swift/loss_scale/utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from swift.template import split_str_parts_by


def calculate_loss_scale(query: str,
def calculate_loss_scale(query: Optional[str],
response: str,
response_loss_scale_map: Dict[str, list],
query_loss_scale_map: Optional[Dict[str, list]] = None) -> Tuple[List[str], List[float]]:
Expand All @@ -28,7 +28,7 @@ def calculate_loss_scale(query: str,
A tuple of agent response parts and their weights.
"""
# query loss scale map
if query_loss_scale_map is not None:
if query_loss_scale_map is not None and query is not None:
for key in query_loss_scale_map.keys():
if key in query:
if isinstance(query_loss_scale_map[key], (float, int)):
Expand Down
Loading