Class: Informers::Model

Inherits:

Object

Object
Informers::Model

show all

Defined in:: lib/informers/model.rb

Instance Method Summary collapse

#embed(texts) ⇒ Object
#initialize(model_id, quantized: false) ⇒ Model constructor

A new instance of Model.

Constructor Details

#initialize(model_id, quantized: false) ⇒ `Model`

Returns a new instance of Model.

# File 'lib/informers/model.rb', line 3

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`



8
9
10

# File 'lib/informers/model.rb', line 8

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