Skip to content

Commit 7f3f790

Browse files
authored
Merge pull request #87 from jenan-anthropic/anthropic-params
Use correct params for Anthropic models.
2 parents 6b2a551 + 563fab1 commit 7f3f790

File tree

2 files changed

+4
-51
lines changed

2 files changed

+4
-51
lines changed

server/lib/inference/__init__.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -606,10 +606,9 @@ def __anthropic_text_generation__(self, provider_details: ProviderDetails, infer
606606

607607
response = c.completion_stream(
608608
prompt=f"{anthropic.HUMAN_PROMPT} {inference_request.prompt}{anthropic.AI_PROMPT}",
609-
stopSequences=[anthropic.HUMAN_PROMPT] + inference_request.model_parameters['stopSequences'],
609+
stop_sequences=[anthropic.HUMAN_PROMPT] + inference_request.model_parameters['stopSequences'],
610610
temperature=float(inference_request.model_parameters['temperature']),
611-
topP=float(inference_request.model_parameters['topP']),
612-
topK=int(inference_request.model_parameters['topK']),
611+
top_p=float(inference_request.model_parameters['topP']),
613612
max_tokens_to_sample=inference_request.model_parameters['maximumLength'],
614613
model=inference_request.model_name,
615614
stream=True,

server/models.json

+2-48
Original file line numberDiff line numberDiff line change
@@ -504,9 +504,7 @@
504504
"claude-instant-v1": {
505505
"enabled": false,
506506
"status": "ready",
507-
"capabilities": [
508-
"logprobs"
509-
],
507+
"capabilities": [],
510508
"parameters": {
511509
"temperature": {
512510
"value": 1,
@@ -529,27 +527,6 @@
529527
1
530528
]
531529
},
532-
"topK": {
533-
"value": 1,
534-
"range": [
535-
1,
536-
500
537-
]
538-
},
539-
"presencePenalty": {
540-
"value": 1,
541-
"range": [
542-
0,
543-
1
544-
]
545-
},
546-
"frequencyPenalty": {
547-
"value": 1,
548-
"range": [
549-
0,
550-
1
551-
]
552-
},
553530
"stopSequences": {
554531
"value": [],
555532
"range": []
@@ -559,9 +536,7 @@
559536
"claude-v1": {
560537
"enabled": false,
561538
"status": "ready",
562-
"capabilities": [
563-
"logprobs"
564-
],
539+
"capabilities": [],
565540
"parameters": {
566541
"temperature": {
567542
"value": 1,
@@ -584,27 +559,6 @@
584559
1
585560
]
586561
},
587-
"topK": {
588-
"value": 1,
589-
"range": [
590-
1,
591-
500
592-
]
593-
},
594-
"presencePenalty": {
595-
"value": 1,
596-
"range": [
597-
0,
598-
1
599-
]
600-
},
601-
"frequencyPenalty": {
602-
"value": 1,
603-
"range": [
604-
0,
605-
1
606-
]
607-
},
608562
"stopSequences": {
609563
"value": [],
610564
"range": []

0 commit comments

Comments
 (0)