Class: Xberg::RerankerConfig

Inherits:
Object
  • Object
show all
Defined in:
sig/types.rbs

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ RerankerConfig

Returns a new instance of RerankerConfig.

Parameters:

  • model: (RerankerModelType)
  • top_k: (Integer)
  • batch_size: (Integer)
  • show_download_progress: (Boolean)
  • cache_dir: (String)
  • acceleration: (AccelerationConfig)
  • max_rerank_duration_secs: (Integer) 


508
 
# File 'sig/types.rbs', line 508

def initialize: (?model: RerankerModelType, ?top_k: Integer, ?batch_size: Integer, ?show_download_progress: bool, ?cache_dir: String, ?acceleration: AccelerationConfig, ?max_rerank_duration_secs: Integer) -> void

Instance Attribute Details

#acceleration ⇒ AccelerationConfig?

Returns the value of attribute acceleration.

Returns:



505
506
507
 
# File 'sig/types.rbs', line 505

def acceleration
  @acceleration
end

#batch_size ⇒ Integer?

Returns the value of attribute batch_size.

Returns:

  • (Integer, nil) 


502
503
504
 
# File 'sig/types.rbs', line 502

def batch_size
  @batch_size
end

#cache_dir ⇒ String?

Returns the value of attribute cache_dir.

Returns:

  • (String, nil) 


504
505
506
 
# File 'sig/types.rbs', line 504

def cache_dir
  @cache_dir
end

#max_rerank_duration_secs ⇒ Integer?

Returns the value of attribute max_rerank_duration_secs.

Returns:

  • (Integer, nil) 


506
507
508
 
# File 'sig/types.rbs', line 506

def max_rerank_duration_secs
  @max_rerank_duration_secs
end

#model ⇒ RerankerModelType?

Returns the value of attribute model.

Returns:



500
501
502
 
# File 'sig/types.rbs', line 500

def model
  @model
end

#show_download_progress ⇒ Boolean?

Returns the value of attribute show_download_progress.

Returns:

  • (Boolean, nil) 


503
504
505
 
# File 'sig/types.rbs', line 503

def show_download_progress
  @show_download_progress
end

#top_k ⇒ Integer?

Returns the value of attribute top_k.

Returns:

  • (Integer, nil) 


501
502
503
 
# File 'sig/types.rbs', line 501

def top_k
  @top_k
end

Class Method Details

.default ⇒ RerankerConfig

Returns:



509
 
# File 'sig/types.rbs', line 509

def self.default: () -> RerankerConfig