x-transformers 2.3.23__py3-none-any.whl → 2.3.25__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/autoregressive_wrapper.py +2 -2
  2. {x_transformers-2.3.23.dist-info → x_transformers-2.3.25.dist-info}/METADATA +1 -1
  3. {x_transformers-2.3.23.dist-info → x_transformers-2.3.25.dist-info}/RECORD +5 -5
  4. {x_transformers-2.3.23.dist-info → x_transformers-2.3.25.dist-info}/WHEEL +0 -0
  5. {x_transformers-2.3.23.dist-info → x_transformers-2.3.25.dist-info}/licenses/LICENSE +0 -0
x_transformers/autoregressive_wrapper.py CHANGED
@@ -324,7 +324,7 @@ class AutoregressiveWrapper(Module):
324
324
  kwargs.update(self_attn_kv_mask = mask)
325
325
 
326
326
  out, cache = self.net(
327
- x,
327
+ inp,
328
328
  return_intermediates = True,
329
329
  return_attn_z_loss = add_attn_z_loss,
330
330
  return_next_embed_pred = add_next_embed_loss,
@@ -356,7 +356,7 @@ class AutoregressiveWrapper(Module):
356
356
  loss = loss + cache.attn_z_loss
357
357
 
358
358
  if add_next_embed_loss:
359
- mask = inp[:, :-1] != ignore_index
359
+ mask = target != ignore_index
360
360
  embed_pred = next_embed_pred[:, :-1]
361
361
  cont_targets = init_embeds[:, 1:].detach()
362
362
 
{x_transformers-2.3.23.dist-info → x_transformers-2.3.25.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: x-transformers
3
- Version: 2.3.23
3
+ Version: 2.3.25
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.3.23.dist-info → x_transformers-2.3.25.dist-info}/RECORD RENAMED
@@ -1,6 +1,6 @@
1
1
  x_transformers/__init__.py,sha256=h3I2ejobgEdy8H7NgV-rP8UaBCnd16-MysvDXH9GMEA,985
2
2
  x_transformers/attend.py,sha256=Ax34Rw56xXAWfFPqtZ_B8iKEW2EfQdbVoc9uFjfeNjA,17404
3
- x_transformers/autoregressive_wrapper.py,sha256=BWFaO-3YWzCcEfp-EC1ZkdckqDpPIOQG6_uyyP6AmhM,11753
3
+ x_transformers/autoregressive_wrapper.py,sha256=O-yRiFfDy_V7h1PKEK71xRJ8DatYSRLxLnBoQ-3tPpk,11750
4
4
  x_transformers/belief_state_wrapper.py,sha256=YLUMk6t2MhFBEw5lHDDHJHcoCxTIkHvxTNY__GGZEKU,13374
5
5
  x_transformers/continuous.py,sha256=hpb1sSbt3k2LNzzjrjSd8F5xOIbKj7IluV9MBEAFLkw,13031
6
6
  x_transformers/dpo.py,sha256=xt4OuOWhU8pN3OKN2LZAaC2NC8iiEnchqqcrPWVqf0o,3521
@@ -11,7 +11,7 @@ x_transformers/nonautoregressive_wrapper.py,sha256=2NU58hYMgn-4Jzg3mie-mXb0XH_dC
11
11
  x_transformers/x_transformers.py,sha256=7phSZvP1_SDRIkVMwVR4cz1dFU2UlR2Wf1HJHEQlcQg,116222
12
12
  x_transformers/xl_autoregressive_wrapper.py,sha256=CvZMJ6A6PA-Y_bQAhnORwjJBSl6Vjq2IdW5KTdk8NI8,4195
13
13
  x_transformers/xval.py,sha256=AwwYUm8yDAtKQyKJDIhYMsiLTJ_skh3scUFMjp5sda8,8597
14
- x_transformers-2.3.23.dist-info/METADATA,sha256=xRMZP1TSYdcbc0F5GX-WcaHhAbQPdGeFIbjHBZYG9_0,89897
15
- x_transformers-2.3.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
- x_transformers-2.3.23.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
17
- x_transformers-2.3.23.dist-info/RECORD,,
14
+ x_transformers-2.3.25.dist-info/METADATA,sha256=PhbwY8z-kRRrC65XN1fJ3cSpnZEV0J4yH-dCEb3JM_o,89897
15
+ x_transformers-2.3.25.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
16
+ x_transformers-2.3.25.dist-info/licenses/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
17
+ x_transformers-2.3.25.dist-info/RECORD,,