deeplotx 0.8.1__py3-none-any.whl → 0.8.2__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 (6) hide show
  1. deeplotx/nn/attention.py +1 -1
  2. {deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/METADATA +1 -1
  3. {deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/RECORD +6 -6
  4. {deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/WHEEL +0 -0
  5. {deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/licenses/LICENSE +0 -0
  6. {deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/top_level.txt +0 -0
deeplotx/nn/attention.py CHANGED
@@ -36,7 +36,7 @@ class Attention(BaseNeuralNetwork):
36
36
  attn = torch.matmul(q, k.transpose(-2, -1))
37
37
  attn = attn / (self._feature_dim ** 0.5)
38
38
  attn = attn.masked_fill(mask == 0, -1e9) if mask is not None else attn
39
- return torch.softmax(attn, dim=-1)
39
+ return torch.softmax(attn, dtype=self.dtype, dim=-1)
40
40
 
41
41
  @override
42
42
  def forward(self, x: torch.Tensor, y: torch.Tensor | None = None, mask: torch.Tensor | None = None) -> torch.Tensor:
{deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: deeplotx
3
- Version: 0.8.1
3
+ Version: 0.8.2
4
4
  Summary: Easy-2-use long text NLP toolkit.
5
5
  Requires-Python: >=3.10
6
6
  Description-Content-Type: text/markdown
{deeplotx-0.8.1.dist-info → deeplotx-0.8.2.dist-info}/RECORD RENAMED
@@ -4,7 +4,7 @@ deeplotx/encoder/encoder.py,sha256=oSBdA-MiwMKNfTFJWR-RdvNS0G0qfX-Qchwy4LuwB00,3
4
4
  deeplotx/encoder/long_text_encoder.py,sha256=PFR6jjGyg1N58TQlKsPaNQEd-EDl13Hyhu7A1KtGBbA,3743
5
5
  deeplotx/encoder/longformer_encoder.py,sha256=A8FXqd4mdHxSn_o_R689XtpT73ISDT788EgMQRGLC2g,1822
6
6
  deeplotx/nn/__init__.py,sha256=01I_yqx9GTa4wy3uNyAqhtxp66tDqxgMLC4Ky5Vnkrg,651
7
- deeplotx/nn/attention.py,sha256=HZ4nfFtkk7AnJ9nuoDSK6zIlIhZ_hbpZc3o6JQIBqJ8,2861
7
+ deeplotx/nn/attention.py,sha256=R-i-Rd7gnsh6hwXDeYfqLQOJvfSZIGfQbFzRlC91XLo,2879
8
8
  deeplotx/nn/auto_regression.py,sha256=uISx29t_zkDGS8s2wvGB6wOGYZitQ4hQ7wyoQl4lcqY,857
9
9
  deeplotx/nn/base_neural_network.py,sha256=FjQEDFH810fJS7JV3aLgJZnaMqC6DH--wlBvuj-ghTc,5900
10
10
  deeplotx/nn/feed_forward.py,sha256=4ozj7EDalO9pb6JUhZtsJqE0r8bIHFApHRt2zTrl4ho,2931
@@ -27,8 +27,8 @@ deeplotx/trainer/text_binary_classification_trainer.py,sha256=QMLR4cC8NCUP-v7SOY
27
27
  deeplotx/util/__init__.py,sha256=5CH4MTeSgsmCe3LPMfvKoSBpwh6jDSBuHVElJvzQzgs,90
28
28
  deeplotx/util/hash.py,sha256=qbNU3RLBWGQYFVte9WZBAkZ1BkdjCXiKLDaKPN54KFk,662
29
29
  deeplotx/util/read_file.py,sha256=ptzouvEQeeW8KU5BrWNJlXw-vFXVrpS9SkAUxsu6A8A,612
30
- deeplotx-0.8.1.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
31
- deeplotx-0.8.1.dist-info/METADATA,sha256=zMKRLmdsEibLnN_hAx3OM7AbX3SiM7X1-8w4eFJGxNY,12251
32
- deeplotx-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
33
- deeplotx-0.8.1.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
34
- deeplotx-0.8.1.dist-info/RECORD,,
30
+ deeplotx-0.8.2.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
31
+ deeplotx-0.8.2.dist-info/METADATA,sha256=nYbTzZTrgKWDcUMv_QFQDxcnuTgxt1Icg1SWzUbLXks,12251
32
+ deeplotx-0.8.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
33
+ deeplotx-0.8.2.dist-info/top_level.txt,sha256=hKg4pVDXZ-WWxkRfJFczRIll1Sv7VyfKCmzHLXbuh1U,9
34
+ deeplotx-0.8.2.dist-info/RECORD,,