dvilasuero HF staff commited on
Commit
0511a77
1 Parent(s): 29b9e30

Update src/distilabel_dataset_generator/pipelines/sft.py

Browse files
src/distilabel_dataset_generator/pipelines/sft.py CHANGED
@@ -116,19 +116,12 @@ User dataset description:
116
 
117
  MODEL = "meta-llama/Meta-Llama-3.1-70B-Instruct"
118
  DEFAULT_DATASET_DESCRIPTIONS = (
119
- "assistant that solves complex math problems using python, it only responds with python code",
120
  "highly proficient assistant for PyTorch and CUDA expert developers to resolve complex issues",
121
  "skilled high school math assistant who helps students solve problems",
122
  "attentive and well-educated customer service assistant for a clothes e-commerce platform",
123
  )
124
- DEFAULT_SYSTEM_PROMPT = """You are an AI assistant specialized in solving complex mathematical problems using Python.
125
-
126
- Your purpose is to provide users with executable and well-documented Python code that accurately solves mathematical problems, ranging from simple algebra to advanced calculus, and including linear algebra, differential equations, and more.
127
-
128
- Users will ask questions in natural language
129
-
130
- Respond with concise Python code that is easy to understand and includes necessary libraries, comments, and variable definitions, focusing solely on providing the solution in code without explaining the math behind it.
131
- """
132
  DEFAULT_DATASET = pd.DataFrame(
133
  {
134
  "prompt": [
 
116
 
117
  MODEL = "meta-llama/Meta-Llama-3.1-70B-Instruct"
118
  DEFAULT_DATASET_DESCRIPTIONS = (
119
+ "assistant that solves complex math problems using python. The assistant always answers in Python to problems described in natural language",
120
  "highly proficient assistant for PyTorch and CUDA expert developers to resolve complex issues",
121
  "skilled high school math assistant who helps students solve problems",
122
  "attentive and well-educated customer service assistant for a clothes e-commerce platform",
123
  )
124
+ DEFAULT_SYSTEM_PROMPT = """You are an AI assistant specialized in solving complex math problems using Python. Your purpose is to help users overcome mathematical challenges by providing Python code that accurately addresses the problem. Always answer in Python, using descriptive variable names and clear comments to explain your thought process. When necessary, provide additional context or explanations to help users understand the solution."""
 
 
 
 
 
 
 
125
  DEFAULT_DATASET = pd.DataFrame(
126
  {
127
  "prompt": [