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) 


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

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:



533
534
535
 
# File 'sig/types.rbs', line 533

def acceleration
  @acceleration
end

#batch_size ⇒ Integer?

Returns the value of attribute batch_size.

Returns:

  • (Integer, nil) 


530
531
532
 
# File 'sig/types.rbs', line 530

def batch_size
  @batch_size
end

#cache_dir ⇒ String?

Returns the value of attribute cache_dir.

Returns:

  • (String, nil) 


532
533
534
 
# File 'sig/types.rbs', line 532

def cache_dir
  @cache_dir
end

#max_rerank_duration_secs ⇒ Integer?

Returns the value of attribute max_rerank_duration_secs.

Returns:

  • (Integer, nil) 


534
535
536
 
# File 'sig/types.rbs', line 534

def max_rerank_duration_secs
  @max_rerank_duration_secs
end

#model ⇒ RerankerModelType?

Returns the value of attribute model.

Returns:



528
529
530
 
# File 'sig/types.rbs', line 528

def model
  @model
end

#show_download_progress ⇒ Boolean?

Returns the value of attribute show_download_progress.

Returns:

  • (Boolean, nil) 


531
532
533
 
# File 'sig/types.rbs', line 531

def show_download_progress
  @show_download_progress
end

#top_k ⇒ Integer?

Returns the value of attribute top_k.

Returns:

  • (Integer, nil) 


529
530
531
 
# File 'sig/types.rbs', line 529

def top_k
  @top_k
end

Class Method Details

.default ⇒ RerankerConfig

Returns:



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

def self.default: () -> RerankerConfig