Skip to content
Permalink

Comparing changes

Choose two branches to see what’s changed or to start a new pull request. If you need to, you can also or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: LiveCodeBench/LiveCodeBench
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: main
Choose a base ref
...
head repository: abacusai/LiveCodeBench
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: main
Choose a head ref
Able to merge. These branches can be automatically merged.
  • 1 commit
  • 1 file changed
  • 1 contributor

Commits on Aug 23, 2024

  1. fix for prompt in dracarys-llama

    @sreemanti-abacusai
    sreemanti-abacusai committed Aug 23, 2024
    Copy the full SHA
    a7d2f7c View commit details
Showing with 21 additions and 1 deletion.
  1. +21 −1 lcb_runner/prompts/code_execution.py
22 changes: 21 additions & 1 deletion lcb_runner/prompts/code_execution.py
View file
Original file line number Diff line number Diff line change
@@ -154,7 +154,27 @@ def format_prompt_execution_base(
]
return chat_messages
elif LanguageModelStyle == LMStyle.DracarysLlama:
return prompt
chat_messages = [
{
"role": "system",
"content": system_message,
},
]
chat_messages += [
{"role": "user", "content": prompt},
]
from transformers import AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained(
"abacusai/Dracarys-Llama-3.1-70B-Instruct", padding_side="right", use_fast=False
)
return tokenizer.apply_chat_template(
chat_messages,
tokenize=False,
add_generation_prompt=True,
truncation=False,
padding=False,
)
elif LanguageModelStyle == LMStyle.DracarysQwen:
return prompt
else: