Class: Aws::SageMakerTrainingSessionRuntime::Types::ComputeLogprobsConfig

Inherits:
Struct
  • Object
show all
Defined in:
gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb

Overview

The configuration for computing log-probabilities of prompt tokens without generating new text.

See Also:

Constant Summary collapse

SENSITIVE = 
[:prompt]

Instance Attribute Summary collapse

Instance Attribute Details

#model_package_arn ⇒ String

The ARN of the model package whose checkpoint this operation should run against. Must be a valid SageMaker model package ARN in the format arn:aws:sagemaker:REGION:ACCOUNT:model-package/PACKAGE_NAME/VERSION. When absent, the operation runs against the session's latest checkpoint.

Returns:

  • (String) 


78
79
80
81
82
83
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 78

class ComputeLogprobsConfig < Struct.new(
  :prompt,
  :model_package_arn)
  SENSITIVE = [:prompt]
  include Aws::Structure
end

#prompt ⇒ Types::ModelInput

The input prompt whose token log-probabilities will be computed.

Returns:



78
79
80
81
82
83
 
# File 'gems/aws-sdk-sagemakertrainingsessionruntime/lib/aws-sdk-sagemakertrainingsessionruntime/types.rb', line 78

class ComputeLogprobsConfig < Struct.new(
  :prompt,
  :model_package_arn)
  SENSITIVE = [:prompt]
  include Aws::Structure
end