github/home-llm
1
1
Fork 0
mirror of https://github.com/acon96/home-llm.git synced 2026-01-08 05:14:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: address dataset generation review feedback

Browse Source 
Co-authored-by: acon96 <35843486+acon96@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2025-12-21 04:28:33 +00:00
parent eee2a6ed11
commit 6e667b17cc
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
data/generate_data.py
View File

@@ -650,7 +650,6 @@ def generate_refusal_example(refusal_case: dict, persona: str, language: str, ma
"available_tools": available_tools,
"question": question,
"answers": [response_text],
"answer_starting": "",
"tool_calls": []
}
@@ -853,7 +852,10 @@ def generate_sft_file(
missing_responses.add(str(ex))
for refusal_case in tqdm(piles.pile_of_refusals):
run_factor_times(generate_refusal_example, generated_examples, refusal_case, person, refusal_factor, language)
try:
run_factor_times(generate_refusal_example, generated_examples, refusal_case, person, refusal_factor, language)
except NoResponseAvailableException as ex:
missing_responses.add(str(ex))
for status_request in tqdm(piles.pile_of_status_requests):
run_factor_times(generate_status_request, generated_examples, status_request, "assistant", status_request_factor, language)