x-transformers 2.1.17__py3-none-any.whl → 2.1.19__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 +1 -3
  2. {x_transformers-2.1.17.dist-info → x_transformers-2.1.19.dist-info}/METADATA +1 -1
  3. {x_transformers-2.1.17.dist-info → x_transformers-2.1.19.dist-info}/RECORD +5 -5
  4. {x_transformers-2.1.17.dist-info → x_transformers-2.1.19.dist-info}/WHEEL +0 -0
  5. {x_transformers-2.1.17.dist-info → x_transformers-2.1.19.dist-info}/licenses/LICENSE +0 -0
x_transformers/belief_state_wrapper.py CHANGED
@@ -69,8 +69,6 @@ class BeliefStateWrapper(Module):
69
69
  dim = forward_decoder.emb_dim
70
70
  num_tokens = forward_decoder.num_tokens
71
71
 
72
- self.to_forward_logits = nn.Linear(dim, num_tokens, bias = False)
73
-
74
72
  # the suffix token
75
73
 
76
74
  self.suffix_token = nn.Parameter(torch.zeros(dim))
@@ -283,7 +281,7 @@ class BeliefStateWrapper(Module):
283
281
 
284
282
  fb_embeds = cat((
285
283
  forward_embeds[:, fi],
286
- backward_embeds[:, bi]
284
+ backward_embeds[:, bi + 1] # needs plus one for auto matically added suffix token
287
285
  ), dim = -1)
288
286
 
289
287
  logits = self.text_head(fb_embeds)
{x_transformers-2.1.17.dist-info → x_transformers-2.1.19.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-transformers
3
- Version: 2.1.17
3
+ Version: 2.1.19
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.17.dist-info → x_transformers-2.1.19.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=7zi_inz75jZVQE4zFVA2H45yR2V0RLTgX_aGO86rN9s,9878
4
+ x_transformers/belief_state_wrapper.py,sha256=wmIyoMOFhd8L7LQ3cbMEo9PB6_uvvfLe9XSerYfcGbo,9862
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.17.dist-info/METADATA,sha256=PzCTyZz0yqv6vXVpzZwtWyu6sjR-kKwcgVPAsrn5TnI,87571
14
- x_transformers-2.1.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
15
- x_transformers-2.1.17.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
16
- x_transformers-2.1.17.dist-info/RECORD,,
13
+ x_transformers-2.1.19.dist-info/METADATA,sha256=zFJb_g8vWMw2DuUuGNyFmUah3LkIKFwhkdEA28u8Fp4,87571
14
+ x_transformers-2.1.19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
15
+ x_transformers-2.1.19.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
16
+ x_transformers-2.1.19.dist-info/RECORD,,