transformers-rb 0.1.1 → 0.1.2

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +4 -0
  3. data/README.md +3 -0
  4. data/lib/transformers/pipelines/_init.rb +6 -5
  5. data/lib/transformers/version.rb +1 -1
  6. metadata +3 -3
checksums.yaml CHANGED
@@ -1,7 +1,7 @@
1
1
  ---
2
2
  SHA256:
3
- metadata.gz: 2b7441c0cc400d85d3d0775e6c53a631c83887e481588148a7cb3242e50c0f08
4
- data.tar.gz: df639eb5c3231d344ec8516ec861cb40239ac5ff0310faa5b1b102a79af97972
3
+ metadata.gz: 3f29055705824ba101cba238960d4f10825c75bc7867b9eb0b611cda6a547612
4
+ data.tar.gz: d0967f7742f7b2d6194376eb040a3be81e77a9ded94302aeb934de678959e434
5
5
  SHA512:
6
- metadata.gz: 566a03e8a65255d3eb7e87bb3e68c0ac4d228ba4bc55e62eaad98f6677b8a5f29ee9201efaefc3edb309d279c1a986a664dfffbf3cc292114394be1c10fa949a
7
- data.tar.gz: fdae84f4fa8aac90f1a627d738fb701ef91d384414173f4042bac56bfe5c78c6ceb98aaea8ef0f2158d324ec60d18cbd91f84bd30f42b49d63193392e0105e96
6
+ metadata.gz: 38b9ed4fd654ca593e3d6e7c7f20eb3c6b68ecfa5f86099fbc8d160f9093617cc79a571c331a0ec0c70a6770c8d9460194ba75e61d534f9e15931f22e5ae60c3
7
+ data.tar.gz: 00ce437ce8fe419fafddd59b7f9f61050d2ddf5817816b53beb1c43badbced9fe77ea28b26f2d607e87a9be855a19c4b254bc50df2881b4126f2def5c6875c3d
data/CHANGELOG.md CHANGED
@@ -1,3 +1,7 @@
1
+ ## 0.1.2 (2024-09-10)
2
+
3
+ - Fixed default revision for pipelines
4
+
1
5
  ## 0.1.1 (2024-08-29)
2
6
 
3
7
  - Added `embedding` pipeline
data/README.md CHANGED
@@ -32,6 +32,9 @@ Embedding
32
32
  - [intfloat/e5-base-v2](#intfloate5-base-v2)
33
33
  - [BAAI/bge-base-en-v1.5](#baaibge-base-en-v15)
34
34
  - [Snowflake/snowflake-arctic-embed-m-v1.5](#snowflakesnowflake-arctic-embed-m-v15)
35
+
36
+ Sparse embedding
37
+
35
38
  - [opensearch-project/opensearch-neural-sparse-encoding-v1](#opensearch-projectopensearch-neural-sparse-encoding-v1)
36
39
 
37
40
  ### sentence-transformers/all-MiniLM-L6-v2
data/lib/transformers/pipelines/_init.rb CHANGED
@@ -24,7 +24,7 @@ module Transformers
24
24
  "pt" => [AutoModel],
25
25
  "default" => {
26
26
  "model" => {
27
- "pt" => ["distilbert/distilbert-base-cased", "935ac13"]
27
+ "pt" => ["distilbert/distilbert-base-cased", "6ea8117"]
28
28
  }
29
29
  },
30
30
  "type" => "multimodal"
@@ -34,7 +34,7 @@ module Transformers
34
34
  "pt" => [AutoModelForSequenceClassification],
35
35
  "default" => {
36
36
  "model" => {
37
- "pt" => ["distilbert/distilbert-base-uncased-finetuned-sst-2-english", "af0f99b"]
37
+ "pt" => ["distilbert/distilbert-base-uncased-finetuned-sst-2-english", "714eb0f"]
38
38
  }
39
39
  },
40
40
  "type" => "text"
@@ -44,7 +44,7 @@ module Transformers
44
44
  "pt" => [AutoModelForTokenClassification],
45
45
  "default" => {
46
46
  "model" => {
47
- "pt" => ["dbmdz/bert-large-cased-finetuned-conll03-english", "f2482bf"]
47
+ "pt" => ["dbmdz/bert-large-cased-finetuned-conll03-english", "4c53496"]
48
48
  }
49
49
  },
50
50
  "type" => "text"
@@ -54,7 +54,7 @@ module Transformers
54
54
  "pt" => [AutoModelForQuestionAnswering],
55
55
  "default" => {
56
56
  "model" => {
57
- "pt" => ["distilbert/distilbert-base-cased-distilled-squad", "626af31"]
57
+ "pt" => ["distilbert/distilbert-base-cased-distilled-squad", "564e9b5"]
58
58
  }
59
59
  },
60
60
  "type" => "text"
@@ -64,7 +64,7 @@ module Transformers
64
64
  "pt" => [AutoModelForImageClassification],
65
65
  "default" => {
66
66
  "model" => {
67
- "pt" => ["google/vit-base-patch16-224", "5dca96d"]
67
+ "pt" => ["google/vit-base-patch16-224", "3f49326"]
68
68
  }
69
69
  },
70
70
  "type" => "image"
@@ -227,6 +227,7 @@ module Transformers
227
227
  " #{revision} (#{Utils::Hub::HUGGINGFACE_CO_RESOLVE_ENDPOINT}/#{model}).\n" +
228
228
  "Using a pipeline without specifying a model name and revision in production is not recommended."
229
229
  )
230
+ hub_kwargs[:revision] = revision
230
231
  if config.nil? && model.is_a?(String)
231
232
  config = AutoConfig.from_pretrained(model, _from_pipeline: task, **hub_kwargs, **model_kwargs)
232
233
  hub_kwargs[:_commit_hash] = config._commit_hash
data/lib/transformers/version.rb CHANGED
@@ -1,3 +1,3 @@
1
1
  module Transformers
2
- VERSION = "0.1.1"
2
+ VERSION = "0.1.2"
3
3
  end
metadata CHANGED
@@ -1,14 +1,14 @@
1
1
  --- !ruby/object:Gem::Specification
2
2
  name: transformers-rb
3
3
  version: !ruby/object:Gem::Version
4
- version: 0.1.1
4
+ version: 0.1.2
5
5
  platform: ruby
6
6
  authors:
7
7
  - Andrew Kane
8
8
  autorequire:
9
9
  bindir: bin
10
10
  cert_chain: []
11
- date: 2024-08-30 00:00:00.000000000 Z
11
+ date: 2024-09-10 00:00:00.000000000 Z
12
12
  dependencies:
13
13
  - !ruby/object:Gem::Dependency
14
14
  name: numo-narray
@@ -155,7 +155,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
155
155
  - !ruby/object:Gem::Version
156
156
  version: '0'
157
157
  requirements: []
158
- rubygems_version: 3.5.11
158
+ rubygems_version: 3.5.16
159
159
  signing_key:
160
160
  specification_version: 4
161
161
  summary: State-of-the-art transformers for Ruby