adjust prompt for sample generation

flairNLP · Apr 18, 2024 · 572351c · 572351c
1 parent cdf6c7c
commit 572351c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/small_model_training/iterative_dataset_synthesis_movies.py b/src/small_model_training/iterative_dataset_synthesis_movies.py
@@ -5,15 +5,15 @@
 label_options = ["positive", "negative"]
 
 prompt = BasePrompt(
- task_description="You are writing a movie review. Write only one {} movie review similar to the following examples.",
+ task_description="Generate a {} movie review similar to this:",
  label_options=label_options,
  target_formatting_template="",
 )
 
 prompt_node = PromptNode(
  model_name_or_path="mistralai/Mistral-7B-Instruct-v0.1",
  max_length=500,
- model_kwargs={"temperature": 0.9},
+ model_kwargs={"temperature": 2},
 
 )