x-transformers 1.42.9__tar.gz → 1.42.10__tar.gz
Sign up to get free protection for your applications and to get access to all the features.
- {x_transformers-1.42.9/x_transformers.egg-info → x_transformers-1.42.10}/PKG-INFO +1 -1
- {x_transformers-1.42.9 → x_transformers-1.42.10}/setup.py +1 -1
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/autoregressive_wrapper.py +1 -1
- {x_transformers-1.42.9 → x_transformers-1.42.10/x_transformers.egg-info}/PKG-INFO +1 -1
- {x_transformers-1.42.9 → x_transformers-1.42.10}/LICENSE +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/README.md +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/setup.cfg +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/tests/test_x_transformers.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/__init__.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/attend.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/continuous.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/dpo.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/multi_input.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/neo_mlp.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/nonautoregressive_wrapper.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/x_transformers.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/xl_autoregressive_wrapper.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/xval.py +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers.egg-info/SOURCES.txt +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers.egg-info/dependency_links.txt +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers.egg-info/requires.txt +0 -0
- {x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers.egg-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]]
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
{x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/nonautoregressive_wrapper.py
RENAMED
File without changes
|
File without changes
|
{x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers/xl_autoregressive_wrapper.py
RENAMED
File without changes
|
File without changes
|
File without changes
|
{x_transformers-1.42.9 → x_transformers-1.42.10}/x_transformers.egg-info/dependency_links.txt
RENAMED
File without changes
|
File without changes
|
File without changes
|