informers 1.0.2 → 1.1.0

data/lib/informers/utils/tensor.rb

@@ -22,5 +22,51 @@ module Informers
         row.map { |v| v / norm }
       end
     end
+
+    def self.stack(tensors, dim = 0)
+      tensors
+    end
+
+    def self.ones_like(tensor)
+      if tensor[0].is_a?(Array)
+        return tensor.map { |v| ones_like(v) }
+      end
+      tensor.map { |_| 1 }
+    end
+
+    def self.dims(tensor)
+      dims = []
+      while tensor.is_a?(Array)
+        dims << tensor.size
+        tensor = tensor[0]
+      end
+      dims
+    end
+
+    def self.interpolate(input, shape, mode = "bilinear", align_corners = false)
+      out_height, out_width = shape
+
+      # Input image dimensions
+      in_channels = dims(input)[-3] || 1
+      in_height = dims(input)[-2]
+      in_width = dims(input)[-1]
+
+      output = interpolate_data(
+        input.flatten,
+        [in_channels, in_height, in_width],
+        [out_height, out_width],
+        mode,
+        align_corners
+      )
+      reshape(output, [in_channels, out_height, out_width])
+    end
+
+    def self.reshape(arr, dims)
+      arr = arr.flatten
+      dims[1..-1].reverse.each do |dim|
+        arr = arr.each_slice(dim)
+      end
+      arr.to_a
+    end
   end
 end

data/lib/informers/version.rb

@@ -1,3 +1,3 @@
 module Informers
-  VERSION = "1.0.2"
+  VERSION = "1.1.0"
 end

data/lib/informers.rb

@@ -11,13 +11,16 @@ require "uri"
 
 # modules
 require_relative "informers/utils/core"
+require_relative "informers/utils/generation"
 require_relative "informers/utils/hub"
+require_relative "informers/utils/image"
 require_relative "informers/utils/math"
 require_relative "informers/utils/tensor"
 require_relative "informers/configs"
 require_relative "informers/env"
 require_relative "informers/model"
 require_relative "informers/models"
+require_relative "informers/processors"
 require_relative "informers/tokenizers"
 require_relative "informers/version"
 require_relative "informers/pipelines"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: informers
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.1.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-08-28 00:00:00.000000000 Z
+date: 2024-09-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: onnxruntime
@@ -30,14 +30,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.5.2
+        version: 0.5.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.5.2
+        version: 0.5.3
 description:
 email: andrew@ankane.org
 executables: []
@@ -53,9 +53,12 @@ files:
 - lib/informers/model.rb
 - lib/informers/models.rb
 - lib/informers/pipelines.rb
+- lib/informers/processors.rb
 - lib/informers/tokenizers.rb
 - lib/informers/utils/core.rb
+- lib/informers/utils/generation.rb
 - lib/informers/utils/hub.rb
+- lib/informers/utils/image.rb
 - lib/informers/utils/math.rb
 - lib/informers/utils/tensor.rb
 - lib/informers/version.rb
@@ -78,7 +81,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.11
+rubygems_version: 3.5.16
 signing_key:
 specification_version: 4
 summary: Fast transformer inference for Ruby