|
|
|
@@ -430,7 +430,7 @@ class SageMakerLargeLanguageModel(LargeLanguageModel): |
|
|
|
type=ParameterType.INT, |
|
|
|
use_template="max_tokens", |
|
|
|
min=1, |
|
|
|
max=credentials.get("context_length", 2048), |
|
|
|
max=int(credentials.get("context_length", 2048)), |
|
|
|
default=512, |
|
|
|
label=I18nObject(zh_Hans="最大生成长度", en_US="Max Tokens"), |
|
|
|
), |
|
|
|
@@ -448,7 +448,7 @@ class SageMakerLargeLanguageModel(LargeLanguageModel): |
|
|
|
if support_vision: |
|
|
|
features.append(ModelFeature.VISION) |
|
|
|
|
|
|
|
context_length = credentials.get("context_length", 2048) |
|
|
|
context_length = int(credentials.get("context_length", 2048)) |
|
|
|
|
|
|
|
entity = AIModelEntity( |
|
|
|
model=model, |