Class: Informers::Model

Inherits:
Object
  • Object
show all
Defined in:
lib/informers/model.rb

Overview

TODO remove in 2.0

Instance Method Summary collapse

Constructor Details

#initialize(model_id, quantized: false) ⇒ Model

Returns a new instance of Model.



4
5
6
7
 
# File 'lib/informers/model.rb', line 4

def initialize(model_id, quantized: false)
  @model = Informers.pipeline("embedding", model_id, quantized: quantized)
  @options = model_id == "mixedbread-ai/mxbai-embed-large-v1" ? {pooling: "cls", normalize: false} : {}
end

Instance Method Details

#embed(texts) ⇒ Object 



9
10
11
 
# File 'lib/informers/model.rb', line 9

def embed(texts)
  @model.(texts, **@options)
end