Spaces:
Running
Running
ChenyuRabbitLove
commited on
Commit
•
882ffef
1
Parent(s):
685a650
feat: rename ntu img to ruby
Browse files
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 =
|
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=
|
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":
|
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,
|