ChenyuRabbitLove commited on
Commit
882ffef
1 Parent(s): 685a650

feat: rename ntu img to ruby

Browse files
Files changed (1) hide show
  1. utils/completion_reward.py +3 -3
utils/completion_reward.py CHANGED
@@ -217,7 +217,7 @@ class OpenAIAgent:
217
  self.temperature = 0.8
218
  self.frequency_penalty = 0
219
  self.presence_penalty = 0
220
- self.max_tokens = 1024
221
 
222
  def get_story(self, user_log):
223
  system_prompt = """
@@ -329,7 +329,7 @@ class AWSAgent:
329
  start_time = time.time()
330
  completion = client.completions.create(
331
  model="anthropic.claude-v2",
332
- max_tokens_to_sample=1024,
333
  prompt=f"{anthropic_bedrock.HUMAN_PROMPT}{system_prompt},以下是我的故事紀錄```{user_log}``` {anthropic_bedrock.AI_PROMPT}",
334
  )
335
  chinese_converter = OpenCC("s2tw")
@@ -523,7 +523,7 @@ class NTUAgent:
523
  "temperature": 0.7,
524
  "top_p": 1,
525
  "n": 1,
526
- "max_tokens": 1024,
527
  "stop": ["string"],
528
  "stream": False,
529
  "presence_penalty": 0,
 
217
  self.temperature = 0.8
218
  self.frequency_penalty = 0
219
  self.presence_penalty = 0
220
+ self.max_tokens = 2048
221
 
222
  def get_story(self, user_log):
223
  system_prompt = """
 
329
  start_time = time.time()
330
  completion = client.completions.create(
331
  model="anthropic.claude-v2",
332
+ max_tokens_to_sample=2048,
333
  prompt=f"{anthropic_bedrock.HUMAN_PROMPT}{system_prompt},以下是我的故事紀錄```{user_log}``` {anthropic_bedrock.AI_PROMPT}",
334
  )
335
  chinese_converter = OpenCC("s2tw")
 
523
  "temperature": 0.7,
524
  "top_p": 1,
525
  "n": 1,
526
+ "max_tokens": 2048,
527
  "stop": ["string"],
528
  "stream": False,
529
  "presence_penalty": 0,