.dockerignore
LICENSE.md
MANIFEST.in
README.md
docker-bake.hcl
docker-compose.yml
pyproject.toml
setup.py
.devcontainer/devcontainer.json
.devcontainer/docker-compose.extend.yml
.github/workflows/build-push-coreweave.yaml
.github/workflows/build-push-test.yaml
.github/workflows/build-push.yaml
.github/workflows/pypi.yml
configs/hook/loss_scale.example.yml
configs/sd15/sd15.example.yml
configs/sdxl/sdxl-te.example.yaml
configs/sdxl/sdxl.example.yaml
docker/Dockerfile
scripts/convert/sd15-ldm2neurosis.py
scripts/convert/sdxl-neurosis2sgm.py
scripts/convert/vae-neurosis2ldm.py
scripts/modify/sdxl-patch-te.py
src/neurosis/__init__.py
src/neurosis/__main__.py
src/neurosis/_version.py
src/neurosis/constants.py
src/neurosis.egg-info/PKG-INFO
src/neurosis.egg-info/SOURCES.txt
src/neurosis.egg-info/dependency_links.txt
src/neurosis.egg-info/entry_points.txt
src/neurosis.egg-info/requires.txt
src/neurosis.egg-info/top_level.txt
src/neurosis/data/__init__.py
src/neurosis/data/accessors.py
src/neurosis/data/files.py
src/neurosis/data/dinov2/LICENSE.dinov2.md
src/neurosis/data/dinov2/__init__.py
src/neurosis/data/dinov2/acquire.py
src/neurosis/data/fonts/NotoSansMono.ttf
src/neurosis/data/lpips/alex_lpips_v0.1.safetensors
src/neurosis/data/lpips/vgg_lpips_v0.1.safetensors
src/neurosis/dataset/__init__.py
src/neurosis/dataset/base.py
src/neurosis/dataset/loaders.py
src/neurosis/dataset/utils.py
src/neurosis/dataset/aspect/__init__.py
src/neurosis/dataset/aspect/base.py
src/neurosis/dataset/aspect/bucket.py
src/neurosis/dataset/aspect/lists.py
src/neurosis/dataset/aspect/sampler.py
src/neurosis/dataset/imagefolder/__init__.py
src/neurosis/dataset/imagefolder/aspect.py
src/neurosis/dataset/imagefolder/meme.py
src/neurosis/dataset/imagefolder/nobucket.py
src/neurosis/dataset/imagefolder/nocaption.py
src/neurosis/dataset/mongo/__init__.py
src/neurosis/dataset/mongo/aspect.py
src/neurosis/dataset/mongo/base.py
src/neurosis/dataset/mongo/nobucket.py
src/neurosis/dataset/mongo/nocaption.py
src/neurosis/dataset/mongo/settings.py
src/neurosis/dataset/processing/__init__.py
src/neurosis/dataset/processing/transform.py
src/neurosis/dataset/processing/caption/__init__.py
src/neurosis/dataset/processing/caption/clean.py
src/neurosis/dataset/processing/caption/meta.py
src/neurosis/dataset/processing/caption/rating.py
src/neurosis/fsdp/__init__.py
src/neurosis/fsdp/diffusionpolicy.py
src/neurosis/models/__init__.py
src/neurosis/models/autoencoder.py
src/neurosis/models/autoencoder_hf.py
src/neurosis/models/diffusion.py
src/neurosis/models/utils.py
src/neurosis/models/text_encoder/__init__.py
src/neurosis/models/text_encoder/clip.py
src/neurosis/models/text_encoder/clip_t5.py
src/neurosis/models/text_encoder/t5.py
src/neurosis/modules/__init__.py
src/neurosis/modules/attention.py
src/neurosis/modules/distributions.py
src/neurosis/modules/ema.py
src/neurosis/modules/guidance.py
src/neurosis/modules/layers.py
src/neurosis/modules/regularizers.py
src/neurosis/modules/autoencoding/__init__.py
src/neurosis/modules/autoencoding/losses/__init__.py
src/neurosis/modules/autoencoding/losses/discriminator_loss.py
src/neurosis/modules/autoencoding/losses/dreamsim.py
src/neurosis/modules/autoencoding/losses/latent_lpips.py
src/neurosis/modules/autoencoding/losses/vae_lpips_discr.py
src/neurosis/modules/autoencoding/losses/vqperceptual.py
src/neurosis/modules/autoencoding/regularizers/__init__.py
src/neurosis/modules/autoencoding/regularizers/base.py
src/neurosis/modules/autoencoding/regularizers/quantize.py
src/neurosis/modules/diffusion/__init__.py
src/neurosis/modules/diffusion/denoiser.py
src/neurosis/modules/diffusion/denoiser_preconditioning.py
src/neurosis/modules/diffusion/denoiser_weighting.py
src/neurosis/modules/diffusion/discretization.py
src/neurosis/modules/diffusion/loss.py
src/neurosis/modules/diffusion/model.py
src/neurosis/modules/diffusion/openaimodel.py
src/neurosis/modules/diffusion/util.py
src/neurosis/modules/diffusion/wrappers.py
src/neurosis/modules/diffusion/sampling/__init__.py
src/neurosis/modules/diffusion/sampling/sampling.py
src/neurosis/modules/diffusion/sampling/sigma_generators.py
src/neurosis/modules/diffusion/sampling/utils.py
src/neurosis/modules/encoders/__init__.py
src/neurosis/modules/encoders/classed.py
src/neurosis/modules/encoders/embedding.py
src/neurosis/modules/encoders/lowscale.py
src/neurosis/modules/encoders/metadata.py
src/neurosis/modules/encoders/misc.py
src/neurosis/modules/hooks/__init__.py
src/neurosis/modules/hooks/common.py
src/neurosis/modules/losses/__init__.py
src/neurosis/modules/losses/extractors.py
src/neurosis/modules/losses/functions.py
src/neurosis/modules/losses/perceptual.py
src/neurosis/modules/losses/types.py
src/neurosis/modules/losses/dreamsim/__init__.py
src/neurosis/modules/losses/dreamsim/common.py
src/neurosis/modules/losses/dreamsim/model.py
src/neurosis/modules/losses/dreamsim/utils.py
src/neurosis/modules/losses/dreamsim/vit.py
src/neurosis/modules/losses/patchgan/LICENSE
src/neurosis/modules/losses/patchgan/__init__.py
src/neurosis/modules/losses/patchgan/model.py
src/neurosis/modules/sampling/__init__.py
src/neurosis/modules/sampling/common.py
src/neurosis/modules/sampling/discrete.py
src/neurosis/modules/sampling/edmc.py
src/neurosis/modules/sampling/scaling.py
src/neurosis/modules/sampling/schedule.py
src/neurosis/optimizers/LICENSE.adafactor
src/neurosis/optimizers/__init__.py
src/neurosis/optimizers/adafactor.py
src/neurosis/optimizers/came.py
src/neurosis/optimizers/hybrid.py
src/neurosis/optimizers/types.py
src/neurosis/optimizers/utils.py
src/neurosis/schedulers/__init__.py
src/neurosis/schedulers/base.py
src/neurosis/schedulers/cosine.py
src/neurosis/schedulers/hf.py
src/neurosis/schedulers/linear.py
src/neurosis/schedulers/warmup.py
src/neurosis/torch/__init__.py
src/neurosis/torch/hooks.py
src/neurosis/torch/stringtensor.py
src/neurosis/trainer/__init__.py
src/neurosis/trainer/checkpoint_io.py
src/neurosis/trainer/cli.py
src/neurosis/trainer/common.py
src/neurosis/trainer/sdxl.py
src/neurosis/trainer/util.py
src/neurosis/trainer/callbacks/__init__.py
src/neurosis/trainer/callbacks/checkpoint.py
src/neurosis/trainer/callbacks/exception.py
src/neurosis/trainer/callbacks/image_logger.py
src/neurosis/trainer/callbacks/progress.py
src/neurosis/trainer/callbacks/refimg_logger.py
src/neurosis/trainer/callbacks/setup.py
src/neurosis/trainer/callbacks/stats.py
src/neurosis/trainer/callbacks/system.py
src/neurosis/trainer/callbacks/wandb.py
src/neurosis/trainer/profile/__init__.py
src/neurosis/trainer/profile/profiler.py
src/neurosis/utils/__init__.py
src/neurosis/utils/misc.py
src/neurosis/utils/sgm.py
src/neurosis/utils/system.py
src/neurosis/utils/text.py
src/neurosis/utils/blessup/__init__.py
src/neurosis/utils/blessup/__main__.py
src/neurosis/utils/blessup/bless.py
src/neurosis/utils/image/__init__.py
src/neurosis/utils/image/convert.py
src/neurosis/utils/image/grid.py
src/neurosis/utils/image/label.py
src/neurosis/utils/image/pil.py
src/neurosis/utils/image/vae.py
src/neurosis/utils/summary/__init__.py
src/neurosis/utils/summary/model.py