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) 


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

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:



489
490
491
 
# File 'sig/types.rbs', line 489

def acceleration
  @acceleration
end

#batch_size ⇒ Integer?

Returns the value of attribute batch_size.

Returns:

  • (Integer, nil) 


486
487
488
 
# File 'sig/types.rbs', line 486

def batch_size
  @batch_size
end

#cache_dir ⇒ String?

Returns the value of attribute cache_dir.

Returns:

  • (String, nil) 


488
489
490
 
# File 'sig/types.rbs', line 488

def cache_dir
  @cache_dir
end

#max_rerank_duration_secs ⇒ Integer?

Returns the value of attribute max_rerank_duration_secs.

Returns:

  • (Integer, nil) 


490
491
492
 
# File 'sig/types.rbs', line 490

def max_rerank_duration_secs
  @max_rerank_duration_secs
end

#model ⇒ RerankerModelType?

Returns the value of attribute model.

Returns:



484
485
486
 
# File 'sig/types.rbs', line 484

def model
  @model
end

#show_download_progress ⇒ Boolean?

Returns the value of attribute show_download_progress.

Returns:

  • (Boolean, nil) 


487
488
489
 
# File 'sig/types.rbs', line 487

def show_download_progress
  @show_download_progress
end

#top_k ⇒ Integer?

Returns the value of attribute top_k.

Returns:

  • (Integer, nil) 


485
486
487
 
# File 'sig/types.rbs', line 485

def top_k
  @top_k
end

Class Method Details

.default ⇒ RerankerConfig

Returns:



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

def self.default: () -> RerankerConfig