informers 1.1.1 → 1.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a61f01755798e81a975641d60e5bfe09484ced7ce6a3453020c9978dc35b1942
-  data.tar.gz: 811f9c1dc4499ae7de8ebf8e02c0c4e98a0c0bc0af6aaca51025e42ba8165540
+  metadata.gz: dfeaf81a8a597d5c25d340e38d0f0be665d3737a1cc271ece883e5b83472da31
+  data.tar.gz: 36943e4c472c864e5951a02ea336c47192e77bcc51161113e022675cce552b82
 SHA512:
-  metadata.gz: 97b27363fab1e43895e368dbddc819fd4db23d42ce517359e5971347cd902b654f0c66700f07b36cd5f476bd3ea205a91e4f7e7ee0e7d8d455f0dce377bedb2b
-  data.tar.gz: dd1a7f795609423419ce213b00a5aca409f6b4a5bffb111250b4deffcbc6a8113fadf8d603c59fa78fa0f310904a0a3299e3bcdc48101f574171a024d13567e6
+  metadata.gz: f6f9a35abcdcb57ddd24c3d81498fa517df554a48b722d7fcee0776e950762ef6412f90916688ca273b95bac79b2441a92a26bcecdc27e2e3c504eafc2b39793
+  data.tar.gz: b85b9a12439001996c4a1580bfdca650ae920791867a3245531b8b13de0b082ebca5ee76be70e8581947e38cf7ebd8c6474d537234ccef4cddb04cc2c6b3dfa5

data/CHANGELOG.md

@@ -1,3 +1,14 @@
+## 1.2.1 (2025-02-01)
+
+- Fixed error when terminal width is zero
+
+## 1.2.0 (2024-11-14)
+
+- Added support for models with external data
+- Added `device` option
+- Added `dtype` option
+- Added `session_options` option
+
 ## 1.1.1 (2024-10-14)
 
 - Added `audio-classification` pipeline

data/README.md

@@ -1,6 +1,6 @@
 # Informers
 
-:fire: Fast [transformer](https://github.com/xenova/transformers.js) inference for Ruby
+:fire: Fast [transformer](https://github.com/huggingface/transformers.js) inference for Ruby
 
 For non-ONNX models, check out [Transformers.rb](https://github.com/ankane/transformers-ruby) :slightly_smiling_face:
 
@@ -24,7 +24,9 @@ gem "informers"
 Embedding
 
 - [sentence-transformers/all-MiniLM-L6-v2](#sentence-transformersall-MiniLM-L6-v2)
-- [Xenova/multi-qa-MiniLM-L6-cos-v1](#xenovamulti-qa-MiniLM-L6-cos-v1)
+- [sentence-transformers/multi-qa-MiniLM-L6-cos-v1](#sentence-transformersmulti-qa-MiniLM-L6-cos-v1)
+- [sentence-transformers/all-mpnet-base-v2](#sentence-transformersall-mpnet-base-v2)
+- [sentence-transformers/paraphrase-MiniLM-L6-v2](#sentence-transformersparaphrase-minilm-l6-v2)
 - [mixedbread-ai/mxbai-embed-large-v1](#mixedbread-aimxbai-embed-large-v1)
 - [Supabase/gte-small](#supabasegte-small)
 - [intfloat/e5-base-v2](#intfloate5-base-v2)
@@ -32,13 +34,13 @@ Embedding
 - [BAAI/bge-base-en-v1.5](#baaibge-base-en-v15)
 - [jinaai/jina-embeddings-v2-base-en](#jinaaijina-embeddings-v2-base-en)
 - [Snowflake/snowflake-arctic-embed-m-v1.5](#snowflakesnowflake-arctic-embed-m-v15)
-- [Xenova/all-mpnet-base-v2](#xenovaall-mpnet-base-v2)
 
 Reranking
 
 - [mixedbread-ai/mxbai-rerank-base-v1](#mixedbread-aimxbai-rerank-base-v1)
 - [jinaai/jina-reranker-v1-turbo-en](#jinaaijina-reranker-v1-turbo-en)
 - [BAAI/bge-reranker-base](#baaibge-reranker-base)
+- [Xenova/ms-marco-MiniLM-L-6-v2](#xenovams-marco-minilm-l-6-v2)
 
 ### sentence-transformers/all-MiniLM-L6-v2
 
@@ -51,13 +53,7 @@ model = Informers.pipeline("embedding", "sentence-transformers/all-MiniLM-L6-v2"
 embeddings = model.(sentences)
 ```
 
-For a quantized version, use:
-
-```ruby
-model = Informers.pipeline("embedding", "Xenova/all-MiniLM-L6-v2", quantized: true)
-```
-
-### Xenova/multi-qa-MiniLM-L6-cos-v1
+### sentence-transformers/multi-qa-MiniLM-L6-cos-v1
 
 [Docs](https://huggingface.co/Xenova/multi-qa-MiniLM-L6-cos-v1)
 
@@ -65,13 +61,35 @@ model = Informers.pipeline("embedding", "Xenova/all-MiniLM-L6-v2", quantized: tr
 query = "How many people live in London?"
 docs = ["Around 9 Million people live in London", "London is known for its financial district"]
 
-model = Informers.pipeline("embedding", "Xenova/multi-qa-MiniLM-L6-cos-v1")
+model = Informers.pipeline("embedding", "sentence-transformers/multi-qa-MiniLM-L6-cos-v1")
 query_embedding = model.(query)
 doc_embeddings = model.(docs)
 scores = doc_embeddings.map { |e| e.zip(query_embedding).sum { |d, q| d * q } }
 doc_score_pairs = docs.zip(scores).sort_by { |d, s| -s }
 ```
 
+### sentence-transformers/all-mpnet-base-v2
+
+[Docs](https://huggingface.co/sentence-transformers/all-mpnet-base-v2)
+
+```ruby
+sentences = ["This is an example sentence", "Each sentence is converted"]
+
+model = Informers.pipeline("embedding", "sentence-transformers/all-mpnet-base-v2")
+embeddings = model.(sentences)
+```
+
+### sentence-transformers/paraphrase-MiniLM-L6-v2
+
+[Docs](https://huggingface.co/sentence-transformers/paraphrase-MiniLM-L6-v2)
+
+```ruby
+sentences = ["This is an example sentence", "Each sentence is converted"]
+
+model = Informers.pipeline("embedding", "sentence-transformers/paraphrase-MiniLM-L6-v2")
+embeddings = model.(sentences, normalize: false)
+```
+
 ### mixedbread-ai/mxbai-embed-large-v1
 
 [Docs](https://huggingface.co/mixedbread-ai/mxbai-embed-large-v1)
@@ -180,17 +198,6 @@ model = Informers.pipeline("embedding", "Snowflake/snowflake-arctic-embed-m-v1.5
 embeddings = model.(input, model_output: "sentence_embedding", pooling: "none")
 ```
 
-### Xenova/all-mpnet-base-v2
-
-[Docs](https://huggingface.co/Xenova/all-mpnet-base-v2)
-
-```ruby
-sentences = ["This is an example sentence", "Each sentence is converted"]
-
-model = Informers.pipeline("embedding", "Xenova/all-mpnet-base-v2")
-embeddings = model.(sentences)
-```
-
 ### mixedbread-ai/mxbai-rerank-base-v1
 
 [Docs](https://huggingface.co/mixedbread-ai/mxbai-rerank-base-v1)
@@ -227,9 +234,21 @@ model = Informers.pipeline("reranking", "BAAI/bge-reranker-base")
 result = model.(query, docs)
 ```
 
+### Xenova/ms-marco-MiniLM-L-6-v2
+
+[Docs](https://huggingface.co/Xenova/ms-marco-MiniLM-L-6-v2)
+
+```ruby
+query = "How many people live in London?"
+docs = ["Around 9 Million people live in London", "London is known for its financial district"]
+
+model = Informers.pipeline("reranking", "Xenova/ms-marco-MiniLM-L-6-v2")
+result = model.(query, docs)
+```
+
 ### Other
 
-The model must include a `.onnx` file ([example](https://huggingface.co/Xenova/all-MiniLM-L6-v2/tree/main/onnx)). If the file is not at `onnx/model.onnx` or `onnx/model_quantized.onnx`, use the `model_file_name` option to specify the location.
+The model must include a `.onnx` file ([example](https://huggingface.co/Xenova/all-MiniLM-L6-v2/tree/main/onnx)). If the file is not at `onnx/model.onnx`, use the `model_file_name` option to specify the location.
 
 ## Pipelines
 
@@ -411,9 +430,31 @@ qa = Informers.pipeline("document-question-answering")
 qa.("image.jpg", "What is the invoice number?")
 ```
 
+## Reference
+
+Specify a variant of the model if available (`fp32`, `fp16`, `int8`, `uint8`, `q8`, `q4`, `q4f16`, or `bnb4`)
+
+```ruby
+Informers.pipeline("embedding", "Xenova/all-MiniLM-L6-v2", dtype: "fp16")
+```
+
+Specify a device (`cpu`, `cuda`, or `coreml`)
+
+```ruby
+Informers.pipeline("embedding", device: "cuda")
+```
+
+Note: Follow [these instructions](https://github.com/ankane/onnxruntime-ruby?tab=readme-ov-file#gpu-support) for `cuda`
+
+Specify ONNX Runtime [session options](https://github.com/ankane/onnxruntime-ruby?tab=readme-ov-file#session-options)
+
+```ruby
+Informers.pipeline("embedding", session_options: {log_severity_level: 2})
+```
+
 ## Credits
 
-This library was ported from [Transformers.js](https://github.com/xenova/transformers.js) and is available under the same license.
+This library was ported from [Transformers.js](https://github.com/huggingface/transformers.js) and is available under the same license.
 
 ## Upgrading
 

data/lib/informers/backends/onnx.rb

@@ -0,0 +1,19 @@
+module Informers
+  module Backends
+    module Onnx
+      def self.device_to_execution_providers(device)
+        case device&.to_s
+        when "cpu", nil
+          []
+        when "cuda"
+          ["CUDAExecutionProvider"]
+        when "coreml"
+          ["CoreMLExecutionProvider"]
+        else
+          supported_devices = ["cpu", "cuda", "coreml"]
+          raise ArgumentError, "Unsupported device: #{device}. Should be one of: #{supported_devices.join(", ")}"
+        end
+      end
+    end
+  end
+end

data/lib/informers/model.rb

@@ -1,4 +1,5 @@
 module Informers
+  # TODO remove in 2.0
   class Model
     def initialize(model_id, quantized: false)
       @model = Informers.pipeline("embedding", model_id, quantized: quantized)

data/lib/informers/models.rb

@@ -22,7 +22,10 @@ module Informers
       cache_dir: nil,
       local_files_only: false,
       revision: "main",
-      model_file_name: nil
+      device: nil,
+      dtype: nil,
+      model_file_name: nil,
+      session_options: {}
     )
       options = {
         quantized:,
@@ -31,7 +34,10 @@ module Informers
         cache_dir:,
         local_files_only:,
         revision:,
-        model_file_name:
+        device:,
+        dtype:,
+        model_file_name:,
+        session_options:
       }
       config = AutoConfig.from_pretrained(pretrained_model_name_or_path, **options)
       if options[:config].nil?
@@ -109,7 +115,10 @@ module Informers
       cache_dir: nil,
       local_files_only: false,
       revision: "main",
-      model_file_name: nil
+      device: nil,
+      dtype: nil,
+      model_file_name: nil,
+      session_options: {}
     )
       options = {
         quantized:,
@@ -118,22 +127,25 @@ module Informers
         cache_dir:,
         local_files_only:,
         revision:,
-        model_file_name:
+        device:,
+        dtype:,
+        model_file_name:,
+        session_options:
       }
 
       model_name = MODEL_CLASS_TO_NAME_MAPPING[self]
       model_type = MODEL_TYPE_MAPPING[model_name]
 
+      config ||= AutoConfig.from_pretrained(pretrained_model_name_or_path, **options)
+
       if model_type == MODEL_TYPES[:DecoderOnly]
         info = [
-          AutoConfig.from_pretrained(pretrained_model_name_or_path, **options),
           construct_session(pretrained_model_name_or_path, options[:model_file_name] || "decoder_model_merged", **options),
           Utils::Hub.get_model_json(pretrained_model_name_or_path, "generation_config.json", false, **options)
         ]
 
       elsif model_type == MODEL_TYPES[:Seq2Seq] || model_type == MODEL_TYPES[:Vision2Seq]
         info = [
-          AutoConfig.from_pretrained(pretrained_model_name_or_path, **options),
           construct_session(pretrained_model_name_or_path, "encoder_model", **options),
           construct_session(pretrained_model_name_or_path, "decoder_model_merged", **options),
           Utils::Hub.get_model_json(pretrained_model_name_or_path, "generation_config.json", false, **options)
@@ -141,14 +153,12 @@ module Informers
 
       elsif model_type == MODEL_TYPES[:MaskGeneration]
         info = [
-          AutoConfig.from_pretrained(pretrained_model_name_or_path, **options),
           construct_session(pretrained_model_name_or_path, "vision_encoder", **options),
           construct_session(pretrained_model_name_or_path, "prompt_encoder_mask_decoder", **options)
         ]
 
       elsif model_type == MODEL_TYPES[:EncoderDecoder]
         info = [
-          AutoConfig.from_pretrained(pretrained_model_name_or_path, **options),
           construct_session(pretrained_model_name_or_path, "encoder_model", **options),
           construct_session(pretrained_model_name_or_path, "decoder_model_merged", **options)
         ]
@@ -158,12 +168,11 @@ module Informers
           warn "Model type for '#{model_name || config[:model_type]}' not found, assuming encoder-only architecture. Please report this."
         end
         info = [
-          AutoConfig.from_pretrained(pretrained_model_name_or_path, **options),
           construct_session(pretrained_model_name_or_path, options[:model_file_name] || "model", **options)
         ]
       end
 
-      new(*info)
+      new(config, *info)
     end
 
     def self.construct_session(pretrained_model_name_or_path, file_name, **options)
@@ -175,10 +184,27 @@ module Informers
         prefix = ""
         file_name = file_name[1..]
       end
-      model_file_name = "#{prefix}#{file_name}#{options[:quantized] ? "_quantized" : ""}.onnx"
+      dtype = options[:dtype] || (options[:quantized] ? "q8" : "fp32")
+      suffix = Utils::DEFAULT_DTYPE_SUFFIX_MAPPING[dtype.to_sym]
+      if !suffix
+        raise ArgumentError, "Invalid dtype: #{dtype}. Should be one of: #{Utils::DEFAULT_DTYPE_SUFFIX_MAPPING.keys.join(", ")}"
+      end
+      model_file_name = "#{prefix}#{file_name}#{suffix}.onnx"
       path = Utils::Hub.get_model_file(pretrained_model_name_or_path, model_file_name, true, **options)
 
-      OnnxRuntime::InferenceSession.new(path)
+      session_options = {
+        providers: Backends::Onnx.device_to_execution_providers(options[:device]),
+        log_severity_level: 4
+      }.merge(options[:session_options] || {})
+
+      begin
+        OnnxRuntime::InferenceSession.new(path, **session_options)
+      rescue OnnxRuntime::Error => e
+        raise e unless e.message.include?("No such file or directory") && e.message.include?(".onnx_data")
+
+        Utils::Hub.get_model_file(pretrained_model_name_or_path, "#{model_file_name}_data", true, **options)
+        OnnxRuntime::InferenceSession.new(path, **session_options)
+      end
     end
 
     def call(model_inputs, **kwargs)
@@ -783,12 +809,49 @@ module Informers
     end
   end
 
+  class ModernBertPreTrainedModel < PreTrainedModel
+  end
+
+  class ModernBertModel < ModernBertPreTrainedModel
+  end
+
+  class ModernBertForMaskedLM < ModernBertPreTrainedModel
+    def call(model_inputs)
+      MaskedLMOutput.new(*super(model_inputs))
+    end
+  end
+
+  class ModernBertForSequenceClassification < ModernBertPreTrainedModel
+    def call(model_inputs)
+      SequenceClassifierOutput.new(*super(model_inputs))
+    end
+  end
+
+  class ModernBertForTokenClassification < ModernBertPreTrainedModel
+    def call(model_inputs)
+      TokenClassifierOutput.new(*super(model_inputs))
+    end
+  end
+
   class NomicBertPreTrainedModel < PreTrainedModel
   end
 
   class NomicBertModel < NomicBertPreTrainedModel
   end
 
+  class ConvBertPreTrainedModel < PreTrainedModel
+  end
+
+  class ConvBertModel < ConvBertPreTrainedModel
+  end
+
+  class ElectraPreTrainedModel < PreTrainedModel
+  end
+
+  # TODO add ElectraForPreTraining
+  class ElectraModel < ElectraPreTrainedModel
+  end
+
   class DebertaV2PreTrainedModel < PreTrainedModel
   end
 
@@ -939,6 +1002,18 @@ module Informers
     end
   end
 
+  class RobertaForTokenClassification <  RobertaPreTrainedModel
+    def call(model_inputs)
+      TokenClassifierOutput.new(*super(model_inputs))
+    end
+  end
+
+  class RobertaForSequenceClassification < RobertaPreTrainedModel
+    def call(model_inputs)
+      SequenceClassifierOutput.new(*super(model_inputs))
+    end
+  end
+
   class XLMRobertaPreTrainedModel < PreTrainedModel
   end
 
@@ -1147,7 +1222,10 @@ module Informers
 
   MODEL_MAPPING_NAMES_ENCODER_ONLY = {
     "bert" => ["BertModel", BertModel],
+    "modernbert" => ["ModernBertModel", ModernBertModel],
     "nomic_bert" => ["NomicBertModel", NomicBertModel],
+    "electra" => ["ElectraModel", ElectraModel],
+    "convbert" => ["ConvBertModel", ConvBertModel],
     "deberta-v2" => ["DebertaV2Model", DebertaV2Model],
     "mpnet" => ["MPNetModel", MPNetModel],
     "distilbert" => ["DistilBertModel", DistilBertModel],
@@ -1182,13 +1260,17 @@ module Informers
 
   MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING_NAMES = {
     "bert" => ["BertForSequenceClassification", BertForSequenceClassification],
+    "modernbert" => ["ModernBertForSequenceClassification", ModernBertForSequenceClassification],
     "distilbert" => ["DistilBertForSequenceClassification", DistilBertForSequenceClassification],
+    "roberta" => ["RobertaForSequenceClassification", RobertaForSequenceClassification],
     "xlm-roberta" => ["XLMRobertaForSequenceClassification", XLMRobertaForSequenceClassification],
     "bart" => ["BartForSequenceClassification", BartForSequenceClassification]
   }
 
   MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMES = {
-    "bert" => ["BertForTokenClassification", BertForTokenClassification]
+    "bert" => ["BertForTokenClassification", BertForTokenClassification],
+    "modernbert" => ["ModernBertForTokenClassification", ModernBertForTokenClassification],
+    "roberta" => ["RobertaForTokenClassification", RobertaForTokenClassification]
   }
 
   MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES = {
@@ -1204,6 +1286,7 @@ module Informers
 
   MODEL_FOR_MASKED_LM_MAPPING_NAMES = {
     "bert" => ["BertForMaskedLM", BertForMaskedLM],
+    "modernbert" => ["ModernBertForMaskedLM", ModernBertForMaskedLM],
     "roberta" => ["RobertaForMaskedLM", RobertaForMaskedLM]
   }
 

data/lib/informers/pipelines.rb

@@ -1336,6 +1336,7 @@ module Informers
     stream = $stderr
     tty = stream.tty?
     width = tty ? stream.winsize[1] : 80
+    width = 80 if width == 0
 
     if msg[:status] == "progress" && tty
       stream.print "\r#{Utils::Hub.display_progress(msg[:file], width, msg[:size], msg[:total_size])}"
@@ -1360,13 +1361,16 @@ module Informers
       cache_dir: nil,
       local_files_only: false,
       revision: "main",
-      model_file_name: nil
+      device: nil,
+      dtype: nil,
+      model_file_name: nil,
+      session_options: {}
     )
       # Apply aliases
       task = TASK_ALIASES[task] || task
 
       if quantized == NO_DEFAULT
-        # TODO move default to task class
+        # TODO no quantization by default in 2.0
         quantized = ["text-classification", "token-classification", "question-answering", "feature-extraction"].include?(task)
       end
 
@@ -1389,7 +1393,10 @@ module Informers
         cache_dir:,
         local_files_only:,
         revision:,
-        model_file_name:
+        device:,
+        dtype:,
+        model_file_name:,
+        session_options:
       }
 
       classes = {

data/lib/informers/tokenizers.rb

@@ -261,7 +261,8 @@ module Informers
       "GPT2Tokenizer" => GPT2Tokenizer,
       "NllbTokenizer" => NllbTokenizer,
       "M2M100Tokenizer" => M2M100Tokenizer,
-      "SpeechT5Tokenizer" => SpeechT5Tokenizer
+      "SpeechT5Tokenizer" => SpeechT5Tokenizer,
+      "PreTrainedTokenizer" => PreTrainedTokenizer
     }
 
     def self.from_pretrained(

data/lib/informers/utils/dtypes.rb

@@ -0,0 +1,14 @@
+module Informers
+  module Utils
+    DEFAULT_DTYPE_SUFFIX_MAPPING = {
+      fp32: "",
+      fp16: "_fp16",
+      int8: "_int8",
+      uint8: "_uint8",
+      q8: "_quantized",
+      q4: "_q4",
+      q4f16: "_q4f16",
+      bnb4: "_bnb4"
+    }
+  end
+end

data/lib/informers/utils/hub.rb

@@ -81,12 +81,18 @@ module Informers
           file if file.exists
         end
 
-        def put(request, buffer)
+        def put(request, response)
           output_path = resolve_path(request)
 
           begin
+            tmp_path = "#{output_path}.incomplete"
             FileUtils.mkdir_p(File.dirname(output_path))
-            File.binwrite(output_path, buffer)
+            File.open(tmp_path, "wb") do |f|
+              while !response.eof?
+                f.write(response.read(1024 * 1024))
+              end
+            end
+            FileUtils.move(tmp_path, output_path)
           rescue => e
             warn "An error occurred while writing the file to cache: #{e}"
           end
@@ -189,10 +195,8 @@ module Informers
           to_cache_response = cache && !response.is_a?(FileResponse) && response.status[0] == "200"
         end
 
-        buffer = response.read
-
         if to_cache_response && cache_key && cache.match(cache_key).nil?
-          cache.put(cache_key, buffer)
+          cache.put(cache_key, response)
         end
 
         Utils.dispatch_callback(options[:progress_callback], {
@@ -229,8 +233,8 @@ module Informers
       end
 
       def self.display_progress(filename, width, size, expected_size)
-        bar_width = width - (filename.length + 3)
-        progress = size / expected_size.to_f
+        bar_width = [width - (filename.length + 3), 1].max
+        progress = expected_size && expected_size > 0 ? size / expected_size.to_f : 0
         done = (progress * bar_width).round
         not_done = bar_width - done
         "#{filename} |#{"█" * done}#{" " * not_done}|"

data/lib/informers/version.rb

@@ -1,3 +1,3 @@
 module Informers
-  VERSION = "1.1.1"
+  VERSION = "1.2.1"
 end

data/lib/informers.rb

@@ -11,8 +11,10 @@ require "stringio"
 require "uri"
 
 # modules
+require_relative "informers/backends/onnx"
 require_relative "informers/utils/audio"
 require_relative "informers/utils/core"
+require_relative "informers/utils/dtypes"
 require_relative "informers/utils/generation"
 require_relative "informers/utils/ffmpeg"
 require_relative "informers/utils/hub"

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: informers
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.2.1
 platform: ruby
 authors:
 - Andrew Kane
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-15 00:00:00.000000000 Z
+date: 2025-02-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: onnxruntime
@@ -38,7 +37,6 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.5.3
-description:
 email: andrew@ankane.org
 executables: []
 extensions: []
@@ -48,6 +46,7 @@ files:
 - LICENSE.txt
 - README.md
 - lib/informers.rb
+- lib/informers/backends/onnx.rb
 - lib/informers/configs.rb
 - lib/informers/env.rb
 - lib/informers/model.rb
@@ -57,6 +56,7 @@ files:
 - lib/informers/tokenizers.rb
 - lib/informers/utils/audio.rb
 - lib/informers/utils/core.rb
+- lib/informers/utils/dtypes.rb
 - lib/informers/utils/ffmpeg.rb
 - lib/informers/utils/generation.rb
 - lib/informers/utils/hub.rb
@@ -68,7 +68,6 @@ homepage: https://github.com/ankane/informers
 licenses:
 - Apache-2.0
 metadata: {}
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -83,8 +82,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.16
-signing_key:
+rubygems_version: 3.6.2
 specification_version: 4
 summary: Fast transformer inference for Ruby
 test_files: []