transformers-rb 0.1.1 → 0.1.2

Sign up to get free protection for your applications and to get access to all the features.
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