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},
 
 )