tensorneko 0.3.10__py3-none-any.whl → 0.3.12__py3-none-any.whl

tensorneko/layer/conv.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from functools import partial
 from typing import Union, Optional, Callable, Tuple, TypeVar, Type, Generic
 
 from torch import Tensor
@@ -7,7 +6,6 @@ from torch.nn import Conv1d as PtConv1d, Conv2d as PtConv2d, Conv3d as PtConv3d,
 from torch.nn.modules.conv import _ConvNd as PtConvNd
 
 from ..neko_module import NekoModule
-from ..util import F
 
 C = TypeVar("C", bound=PtConvNd)
 

tensorneko/util/__init__.py

@@ -4,7 +4,7 @@ from tensorneko_util.util import dispatch, AverageMeter, tensorneko_util_path
 from tensorneko_util.util.fp import Seq, AbstractSeq, curry, F, Stream, return_option, Option, Monad, Eval, _, __
 from tensorneko_util.util import ref, Timer, Singleton
 from tensorneko_util.util.eventbus import Event, EventBus, EventHandler, subscribe
-from tensorneko_util.util import download_file, WindowMerger, Registry
+from tensorneko_util.util import download_file, download_file_thread, download_files_thread, WindowMerger, Registry
 from . import type
 from .configuration import Configuration
 from .misc import reduce_dict_by, summarize_dict_by, with_printed_shape, is_bad_num, count_parameters, compose, \
@@ -73,6 +73,8 @@ __all__ = [
     "try_until_success",
     "sample_indexes",
     "download_file",
+    "download_file_thread",
+    "download_files_thread",
     "WindowMerger",
     "Registry",
 ]

tensorneko/version.txt

@@ -1 +1 @@
-0.3.10
+0.3.12

{tensorneko-0.3.10.dist-info → tensorneko-0.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tensorneko
-Version: 0.3.10
+Version: 0.3.12
 Summary: Tensor Neural Engine Kompanion. An util library based on PyTorch and PyTorch Lightning.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
@@ -33,7 +33,7 @@ Requires-Dist: pillow >=8.1
 Requires-Dist: av >=8.0.3
 Requires-Dist: numpy >=1.20.1
 Requires-Dist: einops >=0.3.0
-Requires-Dist: tensorneko-util ==0.3.10
+Requires-Dist: tensorneko-util ==0.3.12
 Requires-Dist: pysoundfile >=0.9.0 ; platform_system == "Windows"
 
 <h1 style="text-align: center">TensorNeko</h1> 
@@ -72,6 +72,11 @@ To use the library without PyTorch and PyTorch Lightning, you can install the ut
 pip install tensorneko_util
 ```
 
+Some cpu bound functions are implemented by `pyo3`, and you can install the optimized version with below command.
+```shell
+pip install tensorneko_lib
+```
+
 ## Neko Layers, Modules and Architectures
 
 Build an MLP with linear layers. The activation and normalization will be placed in the hidden layers.

{tensorneko-0.3.10.dist-info → tensorneko-0.3.12.dist-info}/RECORD

@@ -2,7 +2,7 @@ tensorneko/__init__.py,sha256=lLVC4StQ5q9OFJTceNrJbj-CIFHFMRvFQyGYJTjuRM4,812
 tensorneko/neko_model.py,sha256=QTbdOAg9ki0ix6mDL_Qu8Wmd5WJOoUFF3M1SXEp3KGc,10551
 tensorneko/neko_module.py,sha256=qELXvguSjWo_NvcRQibiFl0Qauzd9JWLSnT4dbGNS3Y,1473
 tensorneko/neko_trainer.py,sha256=JC8qoKSZ5ngz3grf3S0SjvIFVktDIP_GExth5aFfbGA,10074
-tensorneko/version.txt,sha256=apxBfLVXDP1RFR7rEa3Nef0WFNWj80QwbCsJ5ZStFiM,6
+tensorneko/version.txt,sha256=_hgtyo8QjKpIz7Q25ZNWKcEobB5vDhbZZ42IVY16MxA,6
 tensorneko/arch/__init__.py,sha256=w4lTUeyBIZelrnSjlBFWUF0erzOmBFl9FqeWQuSOyKs,248
 tensorneko/arch/auto_encoder.py,sha256=j6PWWyaNYaYNtw_zZ9ikzhCASqe9viXR3JGBIXSK92Y,2137
 tensorneko/arch/binary_classifier.py,sha256=1MkEbReXKLdDksRG5Rsife40grJk08EVDcNKp54Xvb4,2316
@@ -46,7 +46,7 @@ tensorneko/layer/__init__.py,sha256=HHCBzwR-8UEVsrUKz8j_dhQSVGRASg0flLc_pyG1JN0,
 tensorneko/layer/aggregation.py,sha256=ykH6u-NLJx4Yesu_BLWa6T-vWIYFzJXJV1Txrbz3mPE,1177
 tensorneko/layer/attention.py,sha256=sNC6gZgZaeHqUAfzrPh0Vefp7T6nc1gFvrreuOL7wUg,6360
 tensorneko/layer/concatenate.py,sha256=edQqDvC-8qBOqhqCu96fh4l75uS9GoQmJFZmG8HR_gg,1574
-tensorneko/layer/conv.py,sha256=c0XLcf2hRvyv6hMj4irIAYM0JyPzpv7tJBIW0Dl_sw8,5904
+tensorneko/layer/conv.py,sha256=RwwS3GIQnx8uZL33ZLH7Ai2XRmC2JnLfZp2fqOy7R1A,5853
 tensorneko/layer/linear.py,sha256=NKIzSmSodk0wftu-f-6aSPazJ3X-eduRkZdLMMBjN_8,3283
 tensorneko/layer/log.py,sha256=UN7xgfzC7tF4P856taRtVgVHjc5CTTZqXYODvw6DhKQ,716
 tensorneko/layer/masked_conv2d.py,sha256=4DWjQ_Wc8lncYfk2Iur_M5V0mByvaQ2JZlOGRm29XLE,4796
@@ -72,7 +72,7 @@ tensorneko/preprocess/enum.py,sha256=Wp5qFaUjea5XU4o3N0WxUd-qfzI-m5vr4ZWSqWjELb4
 tensorneko/preprocess/pad.py,sha256=b4IbbhGNRotZ7weZcKA7hfDqSixPo5KjM6khnqzaeUA,3238
 tensorneko/preprocess/resize.py,sha256=hitMlzVnN6n_8nEJwxy4C4ErZrTwpM86QGnYewsrmf8,3469
 tensorneko/preprocess/face_detector/__init__.py,sha256=_ktIfUZqGTX0hk7RBgKf-zHwG2n9KRH4RS7rjuOI8Bo,262
-tensorneko/util/__init__.py,sha256=YCXP1tHeFKEF_NhZP9ZzRDeRx7p5oKOoJ0y7LBAKcL4,2070
+tensorneko/util/__init__.py,sha256=ztsBVn3NLJanwcYyNYyLtUnws6xLcgP3oHMpkjUQh1o,2172
 tensorneko/util/configuration.py,sha256=xXeAjDh1FCNTmSPwDdkL-uH-ULfzFF6Fg0LT7gsZ6nQ,2510
 tensorneko/util/dispatched_misc.py,sha256=_0Go7XezdYB7bpMnCs1MDD_6mPNoWP5qt8DoKuPxynI,997
 tensorneko/util/misc.py,sha256=LEvACtGDOX43iK86A8-Cek0S9rbXFR0AtTP1edE3XDI,4701
@@ -84,8 +84,8 @@ tensorneko/visualization/log_graph.py,sha256=NvOwWVc_petXWYdgaHosPFLa43sHBeacbYc
 tensorneko/visualization/matplotlib.py,sha256=xs9Ssc44ojZX65QU8-fftA7Ug_pBuZ3TBtM8vETNq9w,1568
 tensorneko/visualization/image_browser/__init__.py,sha256=AtykhAE3bXQS6SOWbeYFeeUE9ts9XOFMvrL31z0LoMg,63
 tensorneko/visualization/watcher/__init__.py,sha256=Nq752qIYvfRUZ8VctKQRSqhxh5KmFbWcqPfZlijVx6s,379
-tensorneko-0.3.10.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
-tensorneko-0.3.10.dist-info/METADATA,sha256=hHoyqdXL1Eb-npgPwhTi5GjsPShdnh6V6kIRX6XKIQ8,18906
-tensorneko-0.3.10.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-tensorneko-0.3.10.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
-tensorneko-0.3.10.dist-info/RECORD,,
+tensorneko-0.3.12.dist-info/LICENSE,sha256=Vd75kwgJpVuMnCRBWasQzceMlXt4YQL13ikBLy8G5h0,1067
+tensorneko-0.3.12.dist-info/METADATA,sha256=uUgd_ZqIZDu5O7cLceWJnQOW8FVGV6dXdjOK9nXux-Q,19061
+tensorneko-0.3.12.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+tensorneko-0.3.12.dist-info/top_level.txt,sha256=sZHwlP0iyk7_zHuhRHzSBkdY9yEgyC48f6UVuZ6CvqE,11
+tensorneko-0.3.12.dist-info/RECORD,,

{tensorneko-0.3.10.dist-info → tensorneko-0.3.12.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any