Class: Xberg::SparseEmbeddingModelTypeCustom

Inherits:
Data
  • Object
show all
Extended by:
T::Sig
Includes:
SparseEmbeddingModelType
Defined in:
lib/xberg/native.rb

Overview

Use a custom SPLADE (BertForMaskedLM) ONNX model from HuggingFace.

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#additional_files ⇒ Object (readonly)

rubocop:disable Lint/UselessMethodDefinition



1026
1027
1028
 
# File 'lib/xberg/native.rb', line 1026

def additional_files
  @additional_files
end

#max_length ⇒ Object (readonly)

rubocop:disable Lint/UselessMethodDefinition



1026
1027
1028
 
# File 'lib/xberg/native.rb', line 1026

def max_length
  @max_length
end

#model_file ⇒ Object (readonly)

rubocop:disable Lint/UselessMethodDefinition



1026
1027
1028
 
# File 'lib/xberg/native.rb', line 1026

def model_file
  @model_file
end

#model_id ⇒ Object (readonly)

rubocop:disable Lint/UselessMethodDefinition



1026
1027
1028
 
# File 'lib/xberg/native.rb', line 1026

def model_id
  @model_id
end

Class Method Details

.from_hash(hash) ⇒ Object 



1056
1057
1058
1059
1060
1061
1062
1063
 
# File 'lib/xberg/native.rb', line 1056

def self.from_hash(hash)
  new(
    model_id: hash[:model_id] || hash["model_id"],
    model_file: hash[:model_file] || hash["model_file"],
    additional_files: hash[:additional_files] || hash["additional_files"],
    max_length: hash[:max_length] || hash["max_length"]
  )
end

Instance Method Details

#custom? ⇒ Boolean

Returns:

  • (Boolean) 


1050
 
# File 'lib/xberg/native.rb', line 1050

def custom? = true

#plugin? ⇒ Boolean

Returns:

  • (Boolean) 


1052
1053
1054
 
# File 'lib/xberg/native.rb', line 1052

def plugin? = false
# @param hash [Hash] deserialized from the native extension
# @return [self]

#preset? ⇒ Boolean

Returns:

  • (Boolean) 


1048
 
# File 'lib/xberg/native.rb', line 1048

def preset? = false