x-transformers 1.42.9__py3-none-any.whl → 1.42.10__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- x_transformers/autoregressive_wrapper.py +1 -1
- {x_transformers-1.42.9.dist-info → x_transformers-1.42.10.dist-info}/METADATA +1 -1
- {x_transformers-1.42.9.dist-info → x_transformers-1.42.10.dist-info}/RECORD +6 -6
- {x_transformers-1.42.9.dist-info → x_transformers-1.42.10.dist-info}/LICENSE +0 -0
- {x_transformers-1.42.9.dist-info → x_transformers-1.42.10.dist-info}/WHEEL +0 -0
- {x_transformers-1.42.9.dist-info → x_transformers-1.42.10.dist-info}/top_level.txt +0 -0
@@ -48,7 +48,7 @@ def align_right(t, lens, pad_id = 0):
|
|
48
48
|
batch_arange = torch.arange(batch, device = device, dtype = torch.long)[..., None]
|
49
49
|
prompt_len_arange = torch.arange(seq_len, device = device, dtype = torch.long)
|
50
50
|
|
51
|
-
t = F.pad(t, (max_pad_len, 0), value =
|
51
|
+
t = F.pad(t, (max_pad_len, 0), value = pad_id)
|
52
52
|
offset = max_pad_len - pad_lens
|
53
53
|
|
54
54
|
aligned = t[batch_arange, prompt_len_arange + offset[..., None]]
|
@@ -1,6 +1,6 @@
|
|
1
1
|
x_transformers/__init__.py,sha256=l0dom8ZYkRzFvnDdgzDboXqrI1tKav3beVE7TN2nHko,844
|
2
2
|
x_transformers/attend.py,sha256=SdWlV8Vp5DtpsOzAd0LRhm4VGrJf0lJCGiV2_j_CtoA,17284
|
3
|
-
x_transformers/autoregressive_wrapper.py,sha256=
|
3
|
+
x_transformers/autoregressive_wrapper.py,sha256=reLCno9Z9pchVU79tBF8OMo21LwSZ67KAeB83jqkyAc,10505
|
4
4
|
x_transformers/continuous.py,sha256=cIVEdhfei258__ziV7kQBrJMxCel54bExBTDrO9rfCI,6450
|
5
5
|
x_transformers/dpo.py,sha256=xt4OuOWhU8pN3OKN2LZAaC2NC8iiEnchqqcrPWVqf0o,3521
|
6
6
|
x_transformers/multi_input.py,sha256=tCh-fTJDj2ib4SMGtsa-AM8MxKzJAQSwqAXOu3HU2mg,9252
|
@@ -9,8 +9,8 @@ x_transformers/nonautoregressive_wrapper.py,sha256=2NU58hYMgn-4Jzg3mie-mXb0XH_dC
|
|
9
9
|
x_transformers/x_transformers.py,sha256=VxdA44EYQhVH1Rp7wreJ83I2e0Ea7VN_bFRE-iDXOI8,93833
|
10
10
|
x_transformers/xl_autoregressive_wrapper.py,sha256=CvZMJ6A6PA-Y_bQAhnORwjJBSl6Vjq2IdW5KTdk8NI8,4195
|
11
11
|
x_transformers/xval.py,sha256=7S00kCuab4tWQa-vf-z-XfzADjVj48MoFIr7VSIvttg,8575
|
12
|
-
x_transformers-1.42.
|
13
|
-
x_transformers-1.42.
|
14
|
-
x_transformers-1.42.
|
15
|
-
x_transformers-1.42.
|
16
|
-
x_transformers-1.42.
|
12
|
+
x_transformers-1.42.10.dist-info/LICENSE,sha256=As9u198X-U-vph5noInuUfqsAG2zX_oXPHDmdjwlPPY,1066
|
13
|
+
x_transformers-1.42.10.dist-info/METADATA,sha256=UlOzLgz1fhOIaz3eKDhcHgVI81EZySt4Ko3BiKJ7Jok,690
|
14
|
+
x_transformers-1.42.10.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
|
15
|
+
x_transformers-1.42.10.dist-info/top_level.txt,sha256=hO6KGpFuGucRNEtRfme4A_rGcM53AKwGP7RVlRIxS5Q,15
|
16
|
+
x_transformers-1.42.10.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|