neurosis

View on PyPIReverse Dependencies (1)

0.1.3 neurosis-0.1.3-py3-none-any.whl

Wheel Details

Project: neurosis
Version: 0.1.3
Filename: neurosis-0.1.3-py3-none-any.whl
Download: [link]
Size: 253125
MD5: 172b0a14179f0b857dfa2b9cde547999
SHA256: ee100f3afee731eb49374f09cceb4b2e48c50b500474d860440a94b025cc8113
Uploaded: 2024-07-13 06:22:44 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: neurosis
Version: 0.1.3
Summary: A neural network trainer (for weebs)
Author-Email: Andi Powers-Holmes <aholmes[at]omnom.net>
License: The extensive modifications made to the original code are released/relicensed under: GNU General Public License v3 Copyright (c) 2023, Andi Powers-Holmes This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program. If not, see <http://www.gnu.org/licenses/>. --- In addition to the above, Andi-Powers Holmes grants the following additional permissions: Stability AI and its affiliates are granted a non-exclusive, worldwide, royalty-free, non-transferable, non-sublicensable, non-revocable license to use, reproduce, modify, and distribute the Software, and to permit persons to whom the Software is furnished to do so, subject to the conditions of the MIT License the original code was released under (see below). As a further note, should the GPLv3 license be found to be incompatible with the MIT License, the MIT License shall take precedence, and the GPLv3 license shall be considered null and void. Any person or entity who wishes to use this software in a manner not permitted by the above license(s) can contact Andi Powers-Holmes at the email address listed above (or open a GitHub issue) to negotiate a separate license agreement. Aside: In general I'm happy to hand out MIT licenses, but I'd like to keep track of who's using this so it doesn't get wrapped up in some proprietary for-profit software package without credit and without giving back to the community. --- Based upon code from Stability AI's [generative-models/sgm](https://github.com/Stability-AI/generative-models) repository, originally released under the following license: MIT License Copyright (c) 2023 Stability AI Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
Keywords: deep-learning,diffusers,machine-learning,neural-networks,pytorch-lightning,pytorch,stable-diffusion,transformers
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.10
Requires-Dist: accelerate (>=0.26.1)
Requires-Dist: chardet (==5.1.0)
Requires-Dist: colorama (<0.5.0,>=0.4.3)
Requires-Dist: colorcet (<4.0.0,>=3.0.1)
Requires-Dist: crc32c (>=2.3)
Requires-Dist: diffusers (>=0.26.1)
Requires-Dist: einops (>=0.6.1)
Requires-Dist: fsspec (>=2024.4.0)
Requires-Dist: jsonargparse[argcomplete,fsspec,omegaconf,signatures] (<5.0.0,>=4.27.1)
Requires-Dist: kornia (>=0.7.0)
Requires-Dist: lightning[extra] (>=2.2.1)
Requires-Dist: natsort (>=8.4.0)
Requires-Dist: ninja (>=1.11.0)
Requires-Dist: numpy (>=1.24.3)
Requires-Dist: open-clip-torch (>=2.2.0)
Requires-Dist: opencv-contrib-python-headless (<4.7.0,>=4.6.0.66)
Requires-Dist: pandas (>=2.0.0)
Requires-Dist: pillow (<11.0.0,>=10.0.0)
Requires-Dist: pudb (>=2022.1.3)
Requires-Dist: pydantic (<2.6.0,>=2.5.0)
Requires-Dist: pydantic-settings (<2.2.0,>=2.0.0)
Requires-Dist: pytorch-optimizer (>=2.12.0)
Requires-Dist: rich (<14.0.0,>=13.0.0)
Requires-Dist: safetensors (>=0.4.2)
Requires-Dist: scipy (<1.12.0,>=1.11.1)
Requires-Dist: sentencepiece (>=0.1.99)
Requires-Dist: shellingham (<2.0.0,>=1.5.0)
Requires-Dist: tokenizers (>=0.13.3)
Requires-Dist: torch (>=2.1.0)
Requires-Dist: torchmetrics (>=1.3.0)
Requires-Dist: torchvision
Requires-Dist: tqdm (>=4.65.0)
Requires-Dist: transformers (>=4.36.0)
Requires-Dist: typer (<1.0.0,>=0.9.0)
Requires-Dist: wandb (<0.18.0,>=0.17.0)
Requires-Dist: neurosis[dev]; extra == "all"
Requires-Dist: neurosis[extra]; extra == "all"
Requires-Dist: neurosis[mongo]; extra == "all"
Requires-Dist: neurosis[multinode]; extra == "all"
Requires-Dist: neurosis[vae]; extra == "all"
Requires-Dist: ruff (>=0.0.289); extra == "dev"
Requires-Dist: setuptools-scm (>=8.0.0); extra == "dev"
Requires-Dist: pre-commit (>=3.0.0); extra == "dev"
Requires-Dist: tabulate (>=0.8.9); extra == "dev"
Requires-Dist: bitsandbytes (>=0.41.0); extra == "extra"
Requires-Dist: datasets (>=2.13.0); extra == "extra"
Requires-Dist: hf-transfer (<1.0.0,>=0.1.4); extra == "extra"
Requires-Dist: nvidia-ml-py (<13.0.0,>=12.0.0); extra == "extra"
Requires-Dist: adlfs (>=2024.4.0); extra == "extra"
Requires-Dist: s3fs (>=2024.4.0); extra == "extra"
Requires-Dist: pyarrow (<17.0.0,>=13.0.0); extra == "mongo"
Requires-Dist: pymongoarrow (<1.6.0,>=1.4.0); extra == "mongo"
Requires-Dist: zstandard (>=0.22.0); extra == "mongo"
Requires-Dist: torchelastic (<0.3.0,>=0.2.2); extra == "multinode"
Requires-Dist: neurosis[dev]; extra == "novae"
Requires-Dist: neurosis[extra]; extra == "novae"
Requires-Dist: neurosis[mongo]; extra == "novae"
Requires-Dist: neurosis[multinode]; extra == "novae"
Provides-Extra: all
Provides-Extra: dev
Provides-Extra: extra
Provides-Extra: mongo
Provides-Extra: multinode
Provides-Extra: novae
Description-Content-Type: text/markdown
License-File: LICENSE.md
[Description omitted; length: 3833 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (70.3.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
neurosis/__init__.py sha256=edMGBwmPRndLNT-WGvzPUlzfR5hRHEN64XjWuTsbLTo 1912
neurosis/__main__.py sha256=LcC9HhYy821EXqaVlJDf3zJDZpkmIN64EMtNBDnjnO0 77
neurosis/_version.py sha256=L5DCMp1QAlSqy-8bW7d51bLubTxNjZGYc5fMQkb752U 411
neurosis/constants.py sha256=UfUYByaXUsaX7dJXuDTItE-uyb9p7GVDeE_-f_5_N1w 328
neurosis/data/__init__.py sha256=0tTmBFKNTTBdNoPipgWwMPxcG5l7_RNVwwa4ayN-FDA 213
neurosis/data/accessors.py sha256=bFaq3YkoWVyFlpVtFHK_Cj6EnmeY_XaMZcQwv5HxQYs 1141
neurosis/data/files.py sha256=wkf9bmz8N5UBBbzv6yzWiUWKEz6kzIMJiCfU3ao_yiw 839
neurosis/data/dinov2/LICENSE.dinov2.md sha256=JqFlpJ5PiK10zH24XUw-x4g9vLXxE604iq8XGBwowF0 10609
neurosis/data/dinov2/__init__.py sha256=sQR2oU6siHA93mkABva4xRoOg9yHscKdAm-uTN5-Yto 145
neurosis/data/dinov2/acquire.py sha256=blROf4lj8OvnHcm4LxT3OW3TcdgOj4VXJrzUa1U4eyQ 3973
neurosis/data/fonts/NotoSansMono.ttf sha256=D-tJevaZF94wXu2eqcbekFqmhjwTGaS0M8Vaa8QaPlY 24288
neurosis/data/lpips/alex_lpips_v0.1.safetensors sha256=JV1kVCktFVoGn2k6ERdprOwAPViGF7JQ8kd1NpyBbgQ 5040
neurosis/data/lpips/vgg_lpips_v0.1.safetensors sha256=T7sBIri3xBmDjhL52UFKWQkWdXdzodyDXO963vheiTM 6320
neurosis/dataset/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neurosis/dataset/base.py sha256=HtEy8Bo7l02TKtqf92r-MgytOK-Vh91SKNelPFrQ_BY 1362
neurosis/dataset/loaders.py sha256=FQUFFO7E0PoiDcbiXJ6UV-HxfXRhtngZk7iPTeFOtm0 5585
neurosis/dataset/utils.py sha256=PBtJ6Ix_TaAsLZdyeWPfTk-p8n_70Zqys9mODkRqkw0 6143
neurosis/dataset/aspect/__init__.py sha256=ojoNC8t4qo_8TvZ4WYNENBZ67a_dt8jILi1HuERZgfE 435
neurosis/dataset/aspect/base.py sha256=rNqpsaZtxU2BnnNQyzBX-CkoOXGOGd-SOOd4V7ekg1U 2098
neurosis/dataset/aspect/bucket.py sha256=l4yZC_xqyTniREJ7qfz1KJA7SJ13gKkZ6yJvIEW29N8 9612
neurosis/dataset/aspect/lists.py sha256=74r1xGVAFY2w-Rorb0v2ZUFUWFH3XAnKdzRN7IXQvLw 7335
neurosis/dataset/aspect/sampler.py sha256=nN1ic1AVyRc9f40mRqyH7uCTHBeCW5aT300R_INsG6g 2932
neurosis/dataset/imagefolder/__init__.py sha256=OzTSVDlU7GVQmZQ--s4xnNSD1Wxw6qsaxcBrPcw-Wxk 446
neurosis/dataset/imagefolder/aspect.py sha256=yCCoYR9CJEHOq7PbKaxg-jVATUHhMA9fg98dPWkfJlI 9630
neurosis/dataset/imagefolder/meme.py sha256=tjyJ_EGiA8edTX39hJ648PIria9AFZ0J7BHQIuI7KPc 10301
neurosis/dataset/imagefolder/nobucket.py sha256=59brqnYXr1_CligWxtVz2qCySgAEubji3ojhjZDn6qI 6695
neurosis/dataset/imagefolder/nocaption.py sha256=9mdH-e1fo_D2n6cXXAPXqoFQAL8SzZB4JTjt2dtOXcQ 4275
neurosis/dataset/mongo/__init__.py sha256=bOpWaC1bnbAkj8UahvyhLc6FO0RSMuUOYxdwMqXxFGI 497
neurosis/dataset/mongo/aspect.py sha256=I36POyI2U6slm9ySLsNUZj2rcKTH1T35if13EIYJlZg 12049
neurosis/dataset/mongo/base.py sha256=ua5UhbOFuP4o9mszzBn6uXmta1cS7NciQ3fpXlE4t0U 10143
neurosis/dataset/mongo/nobucket.py sha256=6CNmNhWgteBHSIIFezexb-s4IShVWTNlR8hGTeBJ2Yw 7782
neurosis/dataset/mongo/nocaption.py sha256=C4YxL-aNJU9FjyVyPnBjgeaW00nNHTbekXboWeKVsvA 5895
neurosis/dataset/mongo/settings.py sha256=F5XnVgzUAirDbDVLakPsbyiwhklWGSsZGVDmCqIHVAE 4919
neurosis/dataset/processing/__init__.py sha256=f1nAjStRGT5Uf6DrzuzxbP7Bka1hvFgcIIxCdu49MYs 73
neurosis/dataset/processing/transform.py sha256=fbmOQaW7C7uc-XVmShb_va1CHJOVTK6c-Fel_BoORRg 916
neurosis/dataset/processing/caption/__init__.py sha256=ZNBacLLVdD2HUhNS5Qlba-nXMeK5oLnkDtJgSGNikc8 150
neurosis/dataset/processing/caption/clean.py sha256=YHMJxaUwLdPv7Fs9MJbqHBtOtCOf9dfMRKZ52ArW2Zk 2061
neurosis/dataset/processing/caption/meta.py sha256=qUZ7bwx7eAJdn5ClkWUr0vPhNdv9SnLVVj4Dq7bLMKo 1389
neurosis/dataset/processing/caption/rating.py sha256=TdCPZWwmiF947wl4dHBAgOCwKO_5qI2Oi2DMEARu_PA 5678
neurosis/fsdp/__init__.py sha256=jnHw8oxvdlly79hVs8i8S-aWDl8bbOjqjpvPehxXinM 179
neurosis/fsdp/diffusionpolicy.py sha256=Vh5IEkTXqkQKRJogjyapXAetNyTOtKaezI_b1OShNTQ 5748
neurosis/models/__init__.py sha256=4BEsw3rJNjTD-GwCqIeC946w_YUbKgPHNfAq5kCwVcg 817
neurosis/models/autoencoder.py sha256=v8nWOoAWkzjUHXOM0XDE9J5NCWNPrNPmTISY-XKOl7s 25173
neurosis/models/autoencoder_hf.py sha256=hYBHuv8ziHjJSYqMQmJZR5YsrA36VR2m3_cbZr8uKog 9862
neurosis/models/diffusion.py sha256=KTLAKJ2i3ozuRs677zMlj3iH5KEqtCZzjpTKpBjcnac 17569
neurosis/models/utils.py sha256=KF6Gacv_Qzpl3MUQnP9rBE3iWMZKOBJxMv1m-_KgYbY 1116
neurosis/models/text_encoder/__init__.py sha256=PsT36I2W-_KNGvybkB-ieQy2XaB9yhxHzXJ3w5EfNYk 386
neurosis/models/text_encoder/clip.py sha256=S3MAg4mWA9WZluzCrMi-PZLV47E2_eO6pjdarZ_zryE 20358
neurosis/models/text_encoder/clip_t5.py sha256=zRSQmig19tOh3Oc5K7Mr0S4VzmxvGZoo7SGFYlSVZR8 1174
neurosis/models/text_encoder/t5.py sha256=pFBJ_-wB3G5lW9BZP4njRB49vvY3w0B0KaiLkIvtzVQ 3458
neurosis/modules/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neurosis/modules/attention.py sha256=kkAtkGfGjxzOXw_Y4WNG76OXta69S5uMHYYZG5M-nhk 24550
neurosis/modules/distributions.py sha256=mPouQBBLER8EXXrWTUpYJlviYXusc95rpofc0vYbUhE 2347
neurosis/modules/ema.py sha256=BVJYRvK2GLKaHmgXnpJCGmdEwHR44p1sA1Yubohoz78 12047
neurosis/modules/guidance.py sha256=vo2x-o37KX8_-LQrakBg-IClL2h7tNjXz0tsAOdlpdY 2923
neurosis/modules/layers.py sha256=gsDfA-SziOFB37loT_1KEiDGuDOk-FvR5jNz6GqTsXU 2768
neurosis/modules/regularizers.py sha256=_BGvm9QxurAuX26b9j2MK4JVs7Gb0RDygF7H8KGQ5bE 1765
neurosis/modules/autoencoding/__init__.py sha256=g7XAeT9Bxz-rEoOZHVyhOiELfeS8-cSd4qpoIPPv1Vk 395
neurosis/modules/autoencoding/losses/__init__.py sha256=9FDj0sUKuogfhoJZyE29Y_uzOFxiDal2qse7yax7nQI 678
neurosis/modules/autoencoding/losses/discriminator_loss.py sha256=JrHV3LdFDrjTbRHJOt6ERRmGjAsl79Xafr7uBIQEVFI 12329
neurosis/modules/autoencoding/losses/dreamsim.py sha256=f0eCMd6bsrpZoB5yZCR39mDiKkg0SQIbgo0nluEkYEE 5597
neurosis/modules/autoencoding/losses/latent_lpips.py sha256=IhDvoK9Pn1B_YvJSDo0BDbWO1U9clEfS1JH3EZF57Jk 2927
neurosis/modules/autoencoding/losses/vae_lpips_discr.py sha256=Bna5jFRd725Pc3NaRPQRiJYuv6ipOsOmiRf0TT-sL9s 14722
neurosis/modules/autoencoding/losses/vqperceptual.py sha256=HyUv--6_nEATRArPLE03C7wn9ifr2BmYbOr5baHK2FM 5857
neurosis/modules/autoencoding/regularizers/__init__.py sha256=1m-v4mszxpMhnX9TeiCurnHxC1h13ZMGIjJA3NZlxW0 791
neurosis/modules/autoencoding/regularizers/base.py sha256=8aK5OEkGKiBxraNhEatWTn54mEC3Hop_S0SGD1_5ugI 1275
neurosis/modules/autoencoding/regularizers/quantize.py sha256=ym3yBNLSiDwuA9gQZoh1oxQqCgEfIEz2wnHR8DwFBD4 16831
neurosis/modules/diffusion/__init__.py sha256=fqRYrm54QyKQ5eBVj0ABjIo8YpMUVKMk5GCsGMyasmk 2471
neurosis/modules/diffusion/denoiser.py sha256=jlzfloXvncZFTizdR0KFFBXv9LPZ1wh4WewlRFN7QUk 2993
neurosis/modules/diffusion/denoiser_preconditioning.py sha256=6Ci4ZTqGRcXlJSmajLg9F37QpXfsplzQSOP99AW0RbY 3440
neurosis/modules/diffusion/denoiser_weighting.py sha256=b5cAsP8E3Muzi8sMQ3EK2JTdiZgR4Op2zMYMGgO4odQ 2958
neurosis/modules/diffusion/discretization.py sha256=8kBFlCevzHcgJnDCtjY6Lh54Xda_LEcx2J8MFF-kvqU 6049
neurosis/modules/diffusion/loss.py sha256=LwQq6mKgzWcGrpvFkhcxzofnU-mzOqkcgPdaR4XXJZI 5576
neurosis/modules/diffusion/model.py sha256=VSRP26MLbbwcj7NGc8eFg7YYVcpghAL1ICTcvZDCnvk 26241
neurosis/modules/diffusion/openaimodel.py sha256=VXNv_ZivTGo_65gfWWConXzAhgjuJr3-jsgmnZaUWYY 31877
neurosis/modules/diffusion/util.py sha256=BgO8Rpzj5RMIMhIS-EoQnU9BeN6imrb0ZFw7y7r5vBo 10661
neurosis/modules/diffusion/wrappers.py sha256=K6r6eFPvqiXwS0AJANfqmFmEUFZzpWiXi83kiUc0ngg 1038
neurosis/modules/diffusion/sampling/__init__.py sha256=qG6Muuyv3kOqxcTZXjpoetLfR925jB9lKZNcgJzLwlA 893
neurosis/modules/diffusion/sampling/sampling.py sha256=qM-igL5diwl8TOXEd8ExnBVOT5sCch1YTP96znvTxiw 14373
neurosis/modules/diffusion/sampling/sigma_generators.py sha256=WremHXb7JO8IbZMHs8-kkeslgAt0s7eW2MPsgIoalzs 4808
neurosis/modules/diffusion/sampling/utils.py sha256=n3U6aXoIAhQchhMrGcYtj_xOty8vV-PkwknaM8PaDog 3178
neurosis/modules/encoders/__init__.py sha256=HzWvJvHIZqo-Tw5Vdxp7-myB0dy7f-aSxWxJNUX6owQ 524
neurosis/modules/encoders/classed.py sha256=DdG-4mXrHo8UUEHrdfAcjxBEz0_jiFTILwkFIWt_HQ8 1346
neurosis/modules/encoders/embedding.py sha256=gXY0Qr3_SCYRPYz-FLNS7EkHiGFV_ojqM-ky53cgZts 8852
neurosis/modules/encoders/lowscale.py sha256=AN9bYSVUq6rN00sOAh96QrSBPjS3PrgKYFNAr5dwRyE 3985
neurosis/modules/encoders/metadata.py sha256=oABdsiWJVxaNPMfnvSmYw1lzBAvynFpawbLEPJzPXx8 1691
neurosis/modules/encoders/misc.py sha256=OMhajMkNnd7G4qWDRegEuXW4xd7cnglyTQ1rzHiU8Os 231
neurosis/modules/hooks/__init__.py sha256=IhloCoR26oM3Az8rMVGQfiKFsxWbjIIyg5IrEi7ixDs 60
neurosis/modules/hooks/common.py sha256=T_vea33GXqeSbEF9bqGBsQHCedZ1TofwmlTK6jc8scg 1356
neurosis/modules/losses/__init__.py sha256=T74Nf4yZTqegbZarRehTKVFIKkOEZYfC6FCJYdlrNOQ 378
neurosis/modules/losses/extractors.py sha256=w4_TzCojKVCFGEUaBxtWgwBrQ2X4cyHquvWLwB5_H5E 1357
neurosis/modules/losses/functions.py sha256=SRIvz5pEFqlupUeny0ipw6EbniB3__-wq_EFVSKiDxM 4142
neurosis/modules/losses/perceptual.py sha256=jk43fUrO2-CIki7Zl9uwniXjHQX9l7TfFoTPxwLmnuY 7581
neurosis/modules/losses/types.py sha256=eQa95Qtb5doyiLWbfeIcv5y4wYWhXHTbWKQhRaVVH6Y 521
neurosis/modules/losses/dreamsim/__init__.py sha256=dts9D6jHA846fz3lP2A8vnZB7bVOoFKR-ju1mS11fRU 257
neurosis/modules/losses/dreamsim/common.py sha256=4T9q8skumGDHymR4cHiM8BI_7LhrtFQLx_DolFiQkHQ 1003
neurosis/modules/losses/dreamsim/model.py sha256=ySv7YLcCn4_Yxz7GyDsVoQEPEYnpyUU1K1sLj_Gb6GI 5671
neurosis/modules/losses/dreamsim/utils.py sha256=21I51EAYPvxXxLbQwsOtqTn5q0FRfGQvJrDJSy-aDpE 5789
neurosis/modules/losses/dreamsim/vit.py sha256=e7ieNhz3J6bDyAQ2YXEVxhhosxpxLQcMcRxyqN3mOaA 12946
neurosis/modules/losses/patchgan/LICENSE sha256=VHTZEPx1YQfxSfteoI5Vc2TRYHrNpOeaA7jo96YnlZM 3565
neurosis/modules/losses/patchgan/__init__.py sha256=-maiq_3YJSMyvM9tPsH8H_lvuo5J-Ttl8xfEWLtCVZg 115
neurosis/modules/losses/patchgan/model.py sha256=bjcJajnsF2jeZZtBVJMG_QvMSQOrms6DfHdqCAT9a_0 3224
neurosis/modules/sampling/__init__.py sha256=bIfx-kgAOCMmFjf3LkB6hMWxudMQ_Gm6frUXNwUmjxI 78
neurosis/modules/sampling/common.py sha256=wKfdvu5xzzFv6pOo9IkMgNDem8QWnfv9KM8-5RrnGtA 2343
neurosis/modules/sampling/discrete.py sha256=NU_i0s7_jbA-1QmqS08EQuFBu5X3gZ2YzwoQmKaNbLs 1680
neurosis/modules/sampling/edmc.py sha256=Ooy0OLGjaMLFSJLjl3G-3yXVE_3gKeiOM8n9NyU_QnI 1609
neurosis/modules/sampling/scaling.py sha256=VKMpH0lzMTOWYlZIvBr5ZPiX2fyvDrMBvli5yx9g_cU 1647
neurosis/modules/sampling/schedule.py sha256=Gu2NrVbaAZT5_MZV1gT7UYGP62NweCH0E_y1rK6DT6E 2536
neurosis/optimizers/LICENSE.adafactor sha256=tIi0DRhX9jBGyVzPL-9Yhw3K-Wi2kIt9_6-sVKVieVA 1091
neurosis/optimizers/__init__.py sha256=7GqBr1H-eNAo2NUfZUY9OcfQuK1o2F6m-KgFo40KLo4 210
neurosis/optimizers/adafactor.py sha256=iEIIK_tf9QeIHpdSonjQjXNCIMY7YT-yWNnTKyXQZWQ 11556
neurosis/optimizers/came.py sha256=d5VV-2Ero0cngs1yJMjNy4HH0EYfcRapFNMwQuao3GU 8583
neurosis/optimizers/hybrid.py sha256=AT1SMBjT7FMt5Tm3ucHSfkYToYp8GHs3hydACrwqsIU 7145
neurosis/optimizers/types.py sha256=p8uRGwUVNBcuUtCwOsyrm5jz8sHXkuQYWtju3i18GSs 415
neurosis/optimizers/utils.py sha256=Q0Qu3Q9WxetmhTtn6wkkq0hKuMq8LQTLC6eUG01Y2Ss 589
neurosis/schedulers/__init__.py sha256=7dOEUyTJ8GvvGtdpmzCngmWG5DIkRPXBRi9EGgTwNA0 356
neurosis/schedulers/base.py sha256=w3kTlhxnjQVVcFPrLg4BJjkH6rDf62NCjB_osLmWfmQ 1853
neurosis/schedulers/cosine.py sha256=rr59OIOJ3DzVY2RK0bh4TQjriSjcCMvH_J6AKXQttcI 8281
neurosis/schedulers/hf.py sha256=c80dd-XLYfK-P4Euxy17enABNpsEx7X6kzx1fA_duxo 1558
neurosis/schedulers/linear.py sha256=bszFzCLpczwIBBUv7hbCkILNWKpEl-eU6imiKSvjZzM 3191
neurosis/schedulers/warmup.py sha256=XDJfSK8tcUWFGufcI-wRlbidaKalciKduqiU9tkJ2Ds 7114
neurosis/torch/__init__.py sha256=rxZPwP0CYnLfYEUurWp7BgQ5rlx5-0M2JHWhHe8QPHA 202
neurosis/torch/hooks.py sha256=r7owx-l4AvREH4zd3BAnf9wtrvmxNSH_0vmAP4t8lIQ 271
neurosis/torch/stringtensor.py sha256=eAfIHiXAdffzQs_vDRIWGYRRsLabKbfDG07wLL802V4 2373
neurosis/trainer/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neurosis/trainer/checkpoint_io.py sha256=mY6gDKVgj8wLr2WNQ0zFrqtE6vlJvfTw41d8Hcf3XN4 2135
neurosis/trainer/cli.py sha256=qn7iqqH3hJzEdgj4HvZfj5n7GCoELDdX4RIqL4T8_fk 5787
neurosis/trainer/common.py sha256=cG1iTqDs6itMPGPN7wAOism0r-bD8noAyLj-UnZrXmc 977
neurosis/trainer/sdxl.py sha256=eAiGwrRpJUDhPXyXj_sE-JbrCNRPYMQKLJSjtglZwD0 198
neurosis/trainer/util.py sha256=4Kl_A3u21fmTP9UkLA-JXAqN0Kt4MoQcqSxVt-WF6Ls 3913
neurosis/trainer/callbacks/__init__.py sha256=w3-FD1DsNBEguBr5p7vkDKjikj9gdFc3t14uSPQZ9gY 651
neurosis/trainer/callbacks/checkpoint.py sha256=Uw1QpqTqj40ThItRJ8-EgFNXbabUUIOolrmf7GogEm8 9050
neurosis/trainer/callbacks/exception.py sha256=NrTwSqwjMQ7oa2kBqf2KJ6ZvrbLP5Zj5GChYbVTJY7w 2938
neurosis/trainer/callbacks/image_logger.py sha256=KUNAkMYPt3xHJxZ5lJPBGiVi0mk1Dnd5l5fY3PICfuU 15325
neurosis/trainer/callbacks/progress.py sha256=yHyKkmIYjmCBPqtCy9EoBI8Lvr9U-UcQ631-T361L2I 1870
neurosis/trainer/callbacks/refimg_logger.py sha256=5ZlcPHlat5hhqyNCPF-0C3rTTf2a8dsf2WjmdKlWl4o 9651
neurosis/trainer/callbacks/setup.py sha256=mupeVFadQI0uYxFdlbaOYSdUCBr6drMjg7n5KTTa1uU 1520
neurosis/trainer/callbacks/stats.py sha256=Q9I3A_eNhRFjgGM2POEOf5zA2USdEPvH2AFI2ZIWMfM 4360
neurosis/trainer/callbacks/system.py sha256=_qUp-NehkalfM7yA_KokLFhHQtBj0-X-507U3QAXSI0 4979
neurosis/trainer/callbacks/wandb.py sha256=Y9heHQcaLNUoy8J2qRT4LNAIgB0n9YLz7POiTvO-okU 1288
neurosis/trainer/profile/__init__.py sha256=KkeFpgdN_dPcXuX0xBjqq8A0Hgsc79FfQfCoJKMmJTs 166
neurosis/trainer/profile/profiler.py sha256=OJWOLrhkAPgx7HTvov-V7aCrGaTnIbSi6Xdob2--y-g 10732
neurosis/utils/__init__.py sha256=M_a-D4BEuhzsLx0XgVe-RWdJwrHv8fMFJsi8-AxgPOE 1447
neurosis/utils/misc.py sha256=-oZ9Lbh8y_4oxlpnSM8gC9VL2NW81qZyNHicFp1fZcM 3304
neurosis/utils/sgm.py sha256=y_6_6-J8uiXPshPSZetUtKwQ361zzwX0bzKeQI1Cj5w 5375
neurosis/utils/system.py sha256=cOdoImPvKSTLJC9qpaT3zRGgapOgYtk1kriN-S0o_Mg 2414
neurosis/utils/text.py sha256=Kjxe2c_7lsUIYKsVv4xa7jdROWqhso2mjEQQKKBlqac 450
neurosis/utils/blessup/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neurosis/utils/blessup/__main__.py sha256=UwLeWmHf0-olhL_Zop3sUmyev-P8Rl3w0HzvPoMOaMk 61
neurosis/utils/blessup/bless.py sha256=4fmAiyG0zhbfJb9r3YESDEIHGytpu9ob0kuK7jAsJek 5842
neurosis/utils/image/__init__.py sha256=nUAhNgaSpuGY7q6V56PawjR1CEt3K51gaEa6QH_Wniw 562
neurosis/utils/image/convert.py sha256=UASOrws4-P222XpXkmw0ASG5iAJu6rAgCTY4RqwKijk 5042
neurosis/utils/image/grid.py sha256=GNuqemFqPcI5-1Ul7SdikBDI4Ehd3tigjQN-2I3ZFSA 5959
neurosis/utils/image/label.py sha256=LO9RC5hsmTLmEfd1lNTUfPgsfb3iljTr8N_sv3ZCqQ8 1506
neurosis/utils/image/pil.py sha256=1SmLcuAveiI_An77nQuqHtOBgAnvx4aS6caqdrhi-KA 906
neurosis/utils/image/vae.py sha256=wwAttjQ6inkKd96M9ilgABid_1BOA1VweplQzWX_jdA 549
neurosis/utils/summary/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
neurosis/utils/summary/model.py sha256=VxNKOiqVsuCdDjFITVnpwnk213_5hW4TvROcHN0eTCk 8886
neurosis-0.1.3.dist-info/LICENSE.md sha256=WqLw3x5k_uOV-IauHzyUrOUHycJ4deZ12FHgDUaDMsk 3139
neurosis-0.1.3.dist-info/METADATA sha256=51EkF2JgdGLoB78Ck6d7-ZBJaabwfij1OPTSbBc7Ei0 10568
neurosis-0.1.3.dist-info/WHEEL sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk 91
neurosis-0.1.3.dist-info/entry_points.txt sha256=yoF3nsFdhLV0yDWm4RXskOI6-PzlZ8oL4XRCR_rW4ZU 55
neurosis-0.1.3.dist-info/top_level.txt sha256=OtmY1MZTQnj-vp1WJ79CPs4XuN3mi9s90XSaQglrbqs 9
neurosis-0.1.3.dist-info/RECORD

top_level.txt

neurosis

entry_points.txt

neurosis = neurosis.trainer.cli:main