x-transformers 2.1.18__py3-none-any.whl → 2.1.20__py3-none-any.whl

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 (5) hide show
  1. x_transformers/belief_state_wrapper.py +2 -2
  2. {x_transformers-2.1.18.dist-info → x_transformers-2.1.20.dist-info}/METADATA +1 -1
  3. {x_transformers-2.1.18.dist-info → x_transformers-2.1.20.dist-info}/RECORD +5 -5
  4. {x_transformers-2.1.18.dist-info → x_transformers-2.1.20.dist-info}/WHEEL +0 -0
  5. {x_transformers-2.1.18.dist-info → x_transformers-2.1.20.dist-info}/licenses/LICENSE +0 -0
x_transformers/belief_state_wrapper.py CHANGED
@@ -245,7 +245,7 @@ class BeliefStateWrapper(Module):
245
245
 
246
246
  seq_arange = arange(seq_len, device = device)
247
247
 
248
- fb_pairs = cartesian_prod(seq_arange, seq_arange)
248
+ fb_pairs = cartesian_prod(seq_arange, seq_arange + 1) # plus one for suffix token
249
249
 
250
250
  # filter down to valid pairs, as in figure 11
251
251
  # f - forward, b - backward, i - indices
@@ -271,7 +271,7 @@ class BeliefStateWrapper(Module):
271
271
 
272
272
  fi, bi = fb_pairs.unbind(dim = -1)
273
273
 
274
- labels_fi, labels_bi = (fi + 1), bi
274
+ labels_fi, labels_bi = (fi + 1), (bi - 1)
275
275
 
276
276
  forward_labels, backward_labels = seq[:, labels_fi], seq[:, labels_bi]
277
277
 
{x_transformers-2.1.18.dist-info → x_transformers-2.1.20.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-transformers
3
- Version: 2.1.18
3
+ Version: 2.1.20
4
4
  Summary: X-Transformers
5
5
  Project-URL: Homepage, https://pypi.org/project/x-transformers/
6
6
  Project-URL: Repository, https://github.com/lucidrains/x-transformers
{x_transformers-2.1.18.dist-info → x_transformers-2.1.20.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  x_transformers/__init__.py,sha256=NDoiBivau559WQ0FvXG4ssU3Il9aoHmTIUFN_1juz0s,911
2
2
  x_transformers/attend.py,sha256=-5BWWhFsp7tvZTdN91Ay5SqOjyj9uOs-122vFvoO6b4,17253
3
3
  x_transformers/autoregressive_wrapper.py,sha256=reLCno9Z9pchVU79tBF8OMo21LwSZ67KAeB83jqkyAc,10505
4
- x_transformers/belief_state_wrapper.py,sha256=MkZqVpwEZJfZGTCB24n2DvGbnFSZefk9LZ-HA5LqGS0,9803
4
+ x_transformers/belief_state_wrapper.py,sha256=22jTxhNIKJuQFU8iRanOMpDdyqT_GiCZ2MAprxz6CGo,9841
5
5
  x_transformers/continuous.py,sha256=p0sCAiH1na236ygwgL1Yyhu36eZBf9cZvoW1JyP_fFE,7073
6
6
  x_transformers/dpo.py,sha256=xt4OuOWhU8pN3OKN2LZAaC2NC8iiEnchqqcrPWVqf0o,3521
7
7
  x_transformers/multi_input.py,sha256=tCh-fTJDj2ib4SMGtsa-AM8MxKzJAQSwqAXOu3HU2mg,9252
@@ -10,7 +10,7 @@ x_transformers/nonautoregressive_wrapper.py,sha256=2NU58hYMgn-4Jzg3mie-mXb0XH_dC
10
10
  x_transformers/x_transformers.py,sha256=fqgtIs6__JpLWMnJa8AY5OW3AJ2GR1B5p-9TsWdiOIU,110425
11
11
  x_transformers/xl_autoregressive_wrapper.py,sha256=CvZMJ6A6PA-Y_bQAhnORwjJBSl6Vjq2IdW5KTdk8NI8,4195
12
12
  x_transformers/xval.py,sha256=7S00kCuab4tWQa-vf-z-XfzADjVj48MoFIr7VSIvttg,8575
13
- x_transformers-2.1.18.dist-info/METADATA,sha256=bW46vXghSJjACgIgOlf5UYzX9fZz6ryEAoevC0DnE-0,87571
14
- x_transformers-2.1.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
15
- x_transformers-2.1.18.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
16
- x_transformers-2.1.18.dist-info/RECORD,,
13
+ x_transformers-2.1.20.dist-info/METADATA,sha256=YU5P-lgqBdEofFNiMZH1YIbgH8FddCS-l4K-n1o2h7o,87571
14
+ x_transformers-2.1.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
15
+ x_transformers-2.1.20.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
16
+ x_transformers-2.1.20.dist-info/RECORD,,