├── .env.default ├── .gitignore ├── LICENSE ├── README.md ├── SETUP.md ├── config ├── .gitignore ├── dataset │ ├── build_emb_db.json │ ├── build_splits.json │ ├── dataset_processor.json │ ├── encode.json │ ├── import.json │ ├── integrity_check.json │ ├── label.json │ └── normalize.json ├── models │ ├── default │ │ ├── format.json │ │ ├── model_index.json │ │ ├── unet.json │ │ ├── unet_accelerate.yaml │ │ ├── unet_train.json │ │ ├── upload_to_hf.sh │ │ ├── vae.json │ │ ├── vae_accelerate.yaml │ │ └── vae_train.json │ ├── edm2_dae_d3a │ │ ├── dae.json │ │ ├── dae_accelerate.yaml │ │ ├── dae_train.json │ │ ├── embedding.json │ │ ├── format.json │ │ └── model_index.json │ ├── edm2_dae_e1b │ │ ├── dae.json │ │ ├── dae_accelerate.yaml │ │ ├── dae_train.json │ │ ├── embedding.json │ │ ├── format.json │ │ └── model_index.json │ ├── edm2_ddec_mclt_b1a │ │ ├── dae.json │ │ ├── ddec.json │ │ ├── ddec_accelerate.yaml │ │ ├── ddec_train.json │ │ ├── embedding.json │ │ ├── format.json │ │ └── model_index.json │ └── edm2_ddec_mclt_b2a │ │ ├── dae.json │ │ ├── ddec.json │ │ ├── ddec_accelerate.yaml │ │ ├── ddec_train.json │ │ ├── embedding.json │ │ ├── format.json │ │ └── model_index.json ├── sampling │ ├── nicegui_app.json │ └── presets │ │ └── default.json └── tests │ ├── dae_mdct_test.json │ ├── dae_test.json │ ├── dataloader.json │ ├── embedding.json │ ├── mclt.json │ ├── mdct_format.json │ ├── mdct_psd_format.json │ ├── mp_fourier.json │ ├── ms_mdct_dual_format.json │ ├── ms_mdct_dual_format_2.json │ ├── sigma_sampler.json │ ├── spectrogram.json │ ├── spectrogram_format.json │ └── unet_test.json ├── dataset └── README.md ├── debug └── README.md ├── environment.yml ├── environment_blackwell.yml ├── models └── README.md └── src ├── create_new_model.py ├── dataset ├── __init__.py ├── dataset_processor.py ├── processes │ ├── build_emb_db.py │ ├── build_splits.py │ ├── dedupe.py │ ├── encode.py │ ├── import.py │ ├── integrity_check.py │ ├── label.py │ └── normalize.py └── utils │ ├── compare_folders.py │ ├── list_file_types.py │ ├── scrape_joshw.py │ ├── scrape_zophars.py │ ├── spc_fix.py │ └── unzip.py ├── modules ├── __init__.py ├── daes │ ├── __init__.py │ ├── dae.py │ ├── dae_edm2_d3.py │ ├── dae_edm2_d4.py │ ├── dae_edm2_g1.py │ ├── dae_edm2_g2.py │ ├── dae_edm2_g3.py │ ├── dae_edm2_h1.py │ ├── dae_edm2_i1.py │ ├── dae_edm2_i3.py │ ├── dae_edm2_i4.py │ ├── dae_edm2_i5.py │ ├── dae_edm2_j1.py │ ├── dae_edm2_j2.py │ ├── dae_edm2_j3.py │ ├── dae_edm2_j4.py │ ├── dae_edm2_j5.py │ ├── dae_edm2_k1.py │ ├── dae_edm2_m1.py │ ├── dae_edm2_n1.py │ ├── dae_edm2_o1.py │ ├── dae_edm2_p1.py │ ├── dae_edm2_p2.py │ ├── dae_edm2_p3.py │ ├── dae_edm2_p4.py │ └── old │ │ ├── dae_edm2_a1.py │ │ ├── dae_edm2_b1.py │ │ ├── dae_edm2_b2.py │ │ ├── dae_edm2_c1.py │ │ ├── dae_edm2_d1.py │ │ ├── dae_edm2_d2.py │ │ ├── dae_edm2_e1.py │ │ └── dae_edm2_f1.py ├── discs │ ├── __init__.py │ ├── disc.py │ └── disc_j3.py ├── embeddings │ ├── __init__.py │ ├── clap.py │ └── embedding.py ├── formats │ ├── __init__.py │ ├── format.py │ ├── frequency_scale.py │ ├── mdct.py │ ├── mdct_psd.py │ ├── ms_mdct_dual.py │ ├── ms_mdct_dual_2.py │ ├── old │ │ ├── mclt.py │ │ ├── phase_recovery.py │ │ └── spectrogram.py │ └── raw.py ├── module.py ├── mp_tools.py ├── mss.py ├── old │ └── vaes │ │ ├── __init__.py │ │ ├── vae.py │ │ ├── vae_edm2.py │ │ └── vae_edm2_d1.py ├── rope.py ├── sliding_attention.py ├── unets │ ├── __init__.py │ ├── old │ │ ├── dit_edm2.py │ │ ├── dit_edm2_b.py │ │ ├── dit_edm2_c.py │ │ ├── dit_edm2_e.py │ │ ├── dit_edm2_f.py │ │ ├── unet_edm2.py │ │ ├── unet_edm2_b.py │ │ ├── unet_edm2_b2.py │ │ ├── unet_edm2_b4a.py │ │ ├── unet_edm2_b4b.py │ │ ├── unet_edm2_b4c.py │ │ ├── unet_edm2_b4p.py │ │ ├── unet_edm2_c.py │ │ ├── unet_edm2_c1a.py │ │ ├── unet_edm2_d1.py │ │ ├── unet_edm2_ddec_d1.py │ │ ├── unet_edm2_ddec_mclt.py │ │ ├── unet_edm2_ddec_mdct_b3.py │ │ ├── unet_edm2_ddec_mdct_b4.py │ │ ├── unet_edm2_e.py │ │ ├── unet_edm2_f.py │ │ └── unet_edm2_mixer.py │ ├── unet.py │ ├── unet_edm2_b4.py │ ├── unet_edm2_b4_2.py │ ├── unet_edm2_b4_3.py │ ├── unet_edm2_b4_3d.py │ ├── unet_edm2_b4_p3d.py │ ├── unet_edm2_d1.py │ ├── unet_edm2_ddec_b1.py │ ├── unet_edm2_ddec_d1.py │ ├── unet_edm2_ddec_i1.py │ ├── unet_edm2_ddec_i3.py │ ├── unet_edm2_ddec_i4.py │ ├── unet_edm2_ddec_i5.py │ ├── unet_edm2_ddec_k1.py │ ├── unet_edm2_ddec_mclt_b1.py │ ├── unet_edm2_ddec_mdct_b2.py │ ├── unet_edm2_ddec_mdct_c1.py │ ├── unet_edm2_ddec_mdct_d1.py │ ├── unet_edm2_p1_ddec.py │ ├── unet_edm2_p2_ddec.py │ ├── unet_edm2_p3_ddec.py │ └── unet_edm2_p4_ddec.py └── utils │ ├── combine_models.py │ └── convert_unet_to_inpainting.py ├── pipelines ├── __init__.py └── dual_diffusion_pipeline.py ├── sample.py ├── sampling ├── __init__.py ├── model_server.py ├── nicegui_app.py ├── nicegui_audio_editor.js ├── nicegui_audio_editor.py ├── nicegui_custom_audio.js ├── nicegui_elements.py ├── nicegui_model_explorer.py ├── schedule.py └── tmp │ ├── nicegui_dragdrop_test.py │ ├── nicegui_drawer_test.py │ └── nicegui_scroll_area_test.py ├── tests ├── __init__.py ├── dae.py ├── dae_2.py ├── dae_mdct.py ├── dae_nt.py ├── dataloader.py ├── dataset_processor.py ├── embedding.py ├── load_config.py ├── mdct.py ├── mdct_psd.py ├── mp_fourier.py ├── ms_mdct_dual.py ├── ms_mdct_dual_2.py ├── old │ ├── mclt.py │ └── spectrogram.py ├── resample.py ├── sigma_sampler.py └── unet.py ├── train.py ├── training ├── __init__.py ├── dataset.py ├── ema.py ├── loss │ ├── __init__.py │ ├── difference_of_gaussians.py │ ├── equivariance.py │ ├── multiscale_spectral.py │ ├── spectral_regularization.py │ └── wavelet.py ├── module_trainers │ ├── __init__.py │ ├── dae_p1_trainer.py │ ├── dae_trainer_d3.py │ ├── dae_trainer_d3_2.py │ ├── dae_trainer_g1.py │ ├── dae_trainer_g3.py │ ├── dae_trainer_j1.py │ ├── dae_trainer_m1.py │ ├── dae_trainer_n1.py │ ├── ddec_mclt_trainer_b1.py │ ├── ddec_p1_trainer.py │ ├── ddec_p3_trainer.py │ ├── ddec_p4_trainer.py │ ├── ddec_trainer.py │ ├── ddec_trainer_i1.py │ ├── ddec_trainer_k1.py │ ├── ddec_trainer_nt.py │ ├── module_trainer.py │ ├── old │ │ ├── dae_trainer.py │ │ ├── dae_trainer_b1.py │ │ ├── dae_trainer_c1.py │ │ ├── dae_trainer_d1.py │ │ ├── dae_trainer_d2.py │ │ ├── dae_trainer_d3b.py │ │ ├── dae_trainer_e1.py │ │ ├── dae_trainer_f1.py │ │ ├── ddec_mclt_trainer.py │ │ ├── ddec_mdct_trainer_b2.py │ │ ├── ddec_mdct_trainer_b3.py │ │ ├── ddec_spec_trainer_d1.py │ │ ├── ddec_vae_trainer.py │ │ ├── unet_trainer.py │ │ ├── unet_trainer_b4.py │ │ ├── vae_trainer.py │ │ ├── vae_trainer_c.py │ │ └── vae_trainer_d.py │ ├── unet_trainer.py │ ├── unet_trainer_p3.py │ └── unet_trainer_p4.py ├── muon.py ├── nor_muon.py ├── sigma_sampler.py └── trainer.py └── utils ├── __init__.py ├── compare_dirs.py ├── config.py ├── dual_diffusion_utils.py ├── mclt.py ├── mdct ├── __init__.py ├── functional.py ├── torch_mdct.py └── windows.py ├── resample.py └── roseus_colormap.py /.env.default: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/.env.default -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/README.md -------------------------------------------------------------------------------- /SETUP.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/SETUP.md -------------------------------------------------------------------------------- /config/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/.gitignore -------------------------------------------------------------------------------- /config/dataset/build_emb_db.json: -------------------------------------------------------------------------------- 1 | { 2 | 3 | } -------------------------------------------------------------------------------- /config/dataset/build_splits.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/build_splits.json -------------------------------------------------------------------------------- /config/dataset/dataset_processor.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/dataset_processor.json -------------------------------------------------------------------------------- /config/dataset/encode.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/encode.json -------------------------------------------------------------------------------- /config/dataset/import.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/import.json -------------------------------------------------------------------------------- /config/dataset/integrity_check.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/integrity_check.json -------------------------------------------------------------------------------- /config/dataset/label.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/label.json -------------------------------------------------------------------------------- /config/dataset/normalize.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/dataset/normalize.json -------------------------------------------------------------------------------- /config/models/default/format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/format.json -------------------------------------------------------------------------------- /config/models/default/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/model_index.json -------------------------------------------------------------------------------- /config/models/default/unet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/unet.json -------------------------------------------------------------------------------- /config/models/default/unet_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/unet_accelerate.yaml -------------------------------------------------------------------------------- /config/models/default/unet_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/unet_train.json -------------------------------------------------------------------------------- /config/models/default/upload_to_hf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/upload_to_hf.sh -------------------------------------------------------------------------------- /config/models/default/vae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/vae.json -------------------------------------------------------------------------------- /config/models/default/vae_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/vae_accelerate.yaml -------------------------------------------------------------------------------- /config/models/default/vae_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/default/vae_train.json -------------------------------------------------------------------------------- /config/models/edm2_dae_d3a/dae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_d3a/dae.json -------------------------------------------------------------------------------- /config/models/edm2_dae_d3a/dae_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_d3a/dae_accelerate.yaml -------------------------------------------------------------------------------- /config/models/edm2_dae_d3a/dae_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_d3a/dae_train.json -------------------------------------------------------------------------------- /config/models/edm2_dae_d3a/embedding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_d3a/embedding.json -------------------------------------------------------------------------------- /config/models/edm2_dae_d3a/format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_d3a/format.json -------------------------------------------------------------------------------- /config/models/edm2_dae_d3a/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_d3a/model_index.json -------------------------------------------------------------------------------- /config/models/edm2_dae_e1b/dae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_e1b/dae.json -------------------------------------------------------------------------------- /config/models/edm2_dae_e1b/dae_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_e1b/dae_accelerate.yaml -------------------------------------------------------------------------------- /config/models/edm2_dae_e1b/dae_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_e1b/dae_train.json -------------------------------------------------------------------------------- /config/models/edm2_dae_e1b/embedding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_e1b/embedding.json -------------------------------------------------------------------------------- /config/models/edm2_dae_e1b/format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_e1b/format.json -------------------------------------------------------------------------------- /config/models/edm2_dae_e1b/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_dae_e1b/model_index.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/dae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/dae.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/ddec.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/ddec.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/ddec_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/ddec_accelerate.yaml -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/ddec_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/ddec_train.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/embedding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/embedding.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/format.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b1a/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b1a/model_index.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/dae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/dae.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/ddec.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/ddec.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/ddec_accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/ddec_accelerate.yaml -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/ddec_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/ddec_train.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/embedding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/embedding.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/format.json -------------------------------------------------------------------------------- /config/models/edm2_ddec_mclt_b2a/model_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/models/edm2_ddec_mclt_b2a/model_index.json -------------------------------------------------------------------------------- /config/sampling/nicegui_app.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/sampling/nicegui_app.json -------------------------------------------------------------------------------- /config/sampling/presets/default.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/sampling/presets/default.json -------------------------------------------------------------------------------- /config/tests/dae_mdct_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/dae_mdct_test.json -------------------------------------------------------------------------------- /config/tests/dae_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/dae_test.json -------------------------------------------------------------------------------- /config/tests/dataloader.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/dataloader.json -------------------------------------------------------------------------------- /config/tests/embedding.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/embedding.json -------------------------------------------------------------------------------- /config/tests/mclt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/mclt.json -------------------------------------------------------------------------------- /config/tests/mdct_format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/mdct_format.json -------------------------------------------------------------------------------- /config/tests/mdct_psd_format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/mdct_psd_format.json -------------------------------------------------------------------------------- /config/tests/mp_fourier.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/mp_fourier.json -------------------------------------------------------------------------------- /config/tests/ms_mdct_dual_format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/ms_mdct_dual_format.json -------------------------------------------------------------------------------- /config/tests/ms_mdct_dual_format_2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/ms_mdct_dual_format_2.json -------------------------------------------------------------------------------- /config/tests/sigma_sampler.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/sigma_sampler.json -------------------------------------------------------------------------------- /config/tests/spectrogram.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/spectrogram.json -------------------------------------------------------------------------------- /config/tests/spectrogram_format.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/spectrogram_format.json -------------------------------------------------------------------------------- /config/tests/unet_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/config/tests/unet_test.json -------------------------------------------------------------------------------- /dataset/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/dataset/README.md -------------------------------------------------------------------------------- /debug/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/debug/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/environment.yml -------------------------------------------------------------------------------- /environment_blackwell.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/environment_blackwell.yml -------------------------------------------------------------------------------- /models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/models/README.md -------------------------------------------------------------------------------- /src/create_new_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/create_new_model.py -------------------------------------------------------------------------------- /src/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataset/dataset_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/dataset_processor.py -------------------------------------------------------------------------------- /src/dataset/processes/build_emb_db.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/build_emb_db.py -------------------------------------------------------------------------------- /src/dataset/processes/build_splits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/build_splits.py -------------------------------------------------------------------------------- /src/dataset/processes/dedupe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/dedupe.py -------------------------------------------------------------------------------- /src/dataset/processes/encode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/encode.py -------------------------------------------------------------------------------- /src/dataset/processes/import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/import.py -------------------------------------------------------------------------------- /src/dataset/processes/integrity_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/integrity_check.py -------------------------------------------------------------------------------- /src/dataset/processes/label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/label.py -------------------------------------------------------------------------------- /src/dataset/processes/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/processes/normalize.py -------------------------------------------------------------------------------- /src/dataset/utils/compare_folders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/utils/compare_folders.py -------------------------------------------------------------------------------- /src/dataset/utils/list_file_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/utils/list_file_types.py -------------------------------------------------------------------------------- /src/dataset/utils/scrape_joshw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/utils/scrape_joshw.py -------------------------------------------------------------------------------- /src/dataset/utils/scrape_zophars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/utils/scrape_zophars.py -------------------------------------------------------------------------------- /src/dataset/utils/spc_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/utils/spc_fix.py -------------------------------------------------------------------------------- /src/dataset/utils/unzip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/dataset/utils/unzip.py -------------------------------------------------------------------------------- /src/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/daes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/daes/dae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_d3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_d3.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_d4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_d4.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_g1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_g1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_g2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_g2.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_g3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_g3.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_h1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_h1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_i1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_i1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_i3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_i3.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_i4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_i4.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_i5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_i5.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_j1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_j1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_j2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_j2.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_j3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_j3.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_j4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_j4.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_j5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_j5.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_k1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_k1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_m1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_m1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_n1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_n1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_o1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_o1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_p1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_p1.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_p2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_p2.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_p3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_p3.py -------------------------------------------------------------------------------- /src/modules/daes/dae_edm2_p4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/dae_edm2_p4.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_a1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_a1.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_b1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_b1.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_b2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_b2.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_c1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_c1.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_d1.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_d2.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_e1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_e1.py -------------------------------------------------------------------------------- /src/modules/daes/old/dae_edm2_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/daes/old/dae_edm2_f1.py -------------------------------------------------------------------------------- /src/modules/discs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/discs/disc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/discs/disc.py -------------------------------------------------------------------------------- /src/modules/discs/disc_j3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/discs/disc_j3.py -------------------------------------------------------------------------------- /src/modules/embeddings/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/embeddings/clap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/embeddings/clap.py -------------------------------------------------------------------------------- /src/modules/embeddings/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/embeddings/embedding.py -------------------------------------------------------------------------------- /src/modules/formats/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/formats/format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/format.py -------------------------------------------------------------------------------- /src/modules/formats/frequency_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/frequency_scale.py -------------------------------------------------------------------------------- /src/modules/formats/mdct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/mdct.py -------------------------------------------------------------------------------- /src/modules/formats/mdct_psd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/mdct_psd.py -------------------------------------------------------------------------------- /src/modules/formats/ms_mdct_dual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/ms_mdct_dual.py -------------------------------------------------------------------------------- /src/modules/formats/ms_mdct_dual_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/ms_mdct_dual_2.py -------------------------------------------------------------------------------- /src/modules/formats/old/mclt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/old/mclt.py -------------------------------------------------------------------------------- /src/modules/formats/old/phase_recovery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/old/phase_recovery.py -------------------------------------------------------------------------------- /src/modules/formats/old/spectrogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/old/spectrogram.py -------------------------------------------------------------------------------- /src/modules/formats/raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/formats/raw.py -------------------------------------------------------------------------------- /src/modules/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/module.py -------------------------------------------------------------------------------- /src/modules/mp_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/mp_tools.py -------------------------------------------------------------------------------- /src/modules/mss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/mss.py -------------------------------------------------------------------------------- /src/modules/old/vaes/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/old/vaes/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/old/vaes/vae.py -------------------------------------------------------------------------------- /src/modules/old/vaes/vae_edm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/old/vaes/vae_edm2.py -------------------------------------------------------------------------------- /src/modules/old/vaes/vae_edm2_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/old/vaes/vae_edm2_d1.py -------------------------------------------------------------------------------- /src/modules/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/rope.py -------------------------------------------------------------------------------- /src/modules/sliding_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/sliding_attention.py -------------------------------------------------------------------------------- /src/modules/unets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/modules/unets/old/dit_edm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/dit_edm2.py -------------------------------------------------------------------------------- /src/modules/unets/old/dit_edm2_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/dit_edm2_b.py -------------------------------------------------------------------------------- /src/modules/unets/old/dit_edm2_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/dit_edm2_c.py -------------------------------------------------------------------------------- /src/modules/unets/old/dit_edm2_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/dit_edm2_e.py -------------------------------------------------------------------------------- /src/modules/unets/old/dit_edm2_f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/dit_edm2_f.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_b.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_b2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_b2.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_b4a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_b4a.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_b4b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_b4b.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_b4c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_b4c.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_b4p.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_b4p.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_c.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_c1a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_c1a.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_d1.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_ddec_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_ddec_d1.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_ddec_mclt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_ddec_mclt.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_ddec_mdct_b3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_ddec_mdct_b3.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_ddec_mdct_b4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_ddec_mdct_b4.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_e.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_f.py -------------------------------------------------------------------------------- /src/modules/unets/old/unet_edm2_mixer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/old/unet_edm2_mixer.py -------------------------------------------------------------------------------- /src/modules/unets/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_b4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_b4.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_b4_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_b4_2.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_b4_3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_b4_3.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_b4_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_b4_3d.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_b4_p3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_b4_p3d.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_d1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_b1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_b1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_d1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_i1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_i1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_i3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_i3.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_i4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_i4.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_i5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_i5.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_k1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_k1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_mclt_b1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_mclt_b1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_mdct_b2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_mdct_b2.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_mdct_c1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_mdct_c1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_ddec_mdct_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_ddec_mdct_d1.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_p1_ddec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_p1_ddec.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_p2_ddec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_p2_ddec.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_p3_ddec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_p3_ddec.py -------------------------------------------------------------------------------- /src/modules/unets/unet_edm2_p4_ddec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/unets/unet_edm2_p4_ddec.py -------------------------------------------------------------------------------- /src/modules/utils/combine_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/utils/combine_models.py -------------------------------------------------------------------------------- /src/modules/utils/convert_unet_to_inpainting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/modules/utils/convert_unet_to_inpainting.py -------------------------------------------------------------------------------- /src/pipelines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/pipelines/dual_diffusion_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/pipelines/dual_diffusion_pipeline.py -------------------------------------------------------------------------------- /src/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sample.py -------------------------------------------------------------------------------- /src/sampling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/sampling/model_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/model_server.py -------------------------------------------------------------------------------- /src/sampling/nicegui_app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/nicegui_app.py -------------------------------------------------------------------------------- /src/sampling/nicegui_audio_editor.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/nicegui_audio_editor.js -------------------------------------------------------------------------------- /src/sampling/nicegui_audio_editor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/nicegui_audio_editor.py -------------------------------------------------------------------------------- /src/sampling/nicegui_custom_audio.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/nicegui_custom_audio.js -------------------------------------------------------------------------------- /src/sampling/nicegui_elements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/nicegui_elements.py -------------------------------------------------------------------------------- /src/sampling/nicegui_model_explorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/nicegui_model_explorer.py -------------------------------------------------------------------------------- /src/sampling/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/schedule.py -------------------------------------------------------------------------------- /src/sampling/tmp/nicegui_dragdrop_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/tmp/nicegui_dragdrop_test.py -------------------------------------------------------------------------------- /src/sampling/tmp/nicegui_drawer_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/tmp/nicegui_drawer_test.py -------------------------------------------------------------------------------- /src/sampling/tmp/nicegui_scroll_area_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/sampling/tmp/nicegui_scroll_area_test.py -------------------------------------------------------------------------------- /src/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/tests/dae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/dae.py -------------------------------------------------------------------------------- /src/tests/dae_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/dae_2.py -------------------------------------------------------------------------------- /src/tests/dae_mdct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/dae_mdct.py -------------------------------------------------------------------------------- /src/tests/dae_nt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/dae_nt.py -------------------------------------------------------------------------------- /src/tests/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/dataloader.py -------------------------------------------------------------------------------- /src/tests/dataset_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/dataset_processor.py -------------------------------------------------------------------------------- /src/tests/embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/embedding.py -------------------------------------------------------------------------------- /src/tests/load_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/load_config.py -------------------------------------------------------------------------------- /src/tests/mdct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/mdct.py -------------------------------------------------------------------------------- /src/tests/mdct_psd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/mdct_psd.py -------------------------------------------------------------------------------- /src/tests/mp_fourier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/mp_fourier.py -------------------------------------------------------------------------------- /src/tests/ms_mdct_dual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/ms_mdct_dual.py -------------------------------------------------------------------------------- /src/tests/ms_mdct_dual_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/ms_mdct_dual_2.py -------------------------------------------------------------------------------- /src/tests/old/mclt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/old/mclt.py -------------------------------------------------------------------------------- /src/tests/old/spectrogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/old/spectrogram.py -------------------------------------------------------------------------------- /src/tests/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/resample.py -------------------------------------------------------------------------------- /src/tests/sigma_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/sigma_sampler.py -------------------------------------------------------------------------------- /src/tests/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/tests/unet.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/train.py -------------------------------------------------------------------------------- /src/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/dataset.py -------------------------------------------------------------------------------- /src/training/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/ema.py -------------------------------------------------------------------------------- /src/training/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/training/loss/difference_of_gaussians.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/loss/difference_of_gaussians.py -------------------------------------------------------------------------------- /src/training/loss/equivariance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/loss/equivariance.py -------------------------------------------------------------------------------- /src/training/loss/multiscale_spectral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/loss/multiscale_spectral.py -------------------------------------------------------------------------------- /src/training/loss/spectral_regularization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/loss/spectral_regularization.py -------------------------------------------------------------------------------- /src/training/loss/wavelet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/loss/wavelet.py -------------------------------------------------------------------------------- /src/training/module_trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/training/module_trainers/dae_p1_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_p1_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_d3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_d3.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_d3_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_d3_2.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_g1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_g1.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_g3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_g3.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_j1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_j1.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_m1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_m1.py -------------------------------------------------------------------------------- /src/training/module_trainers/dae_trainer_n1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/dae_trainer_n1.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_mclt_trainer_b1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_mclt_trainer_b1.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_p1_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_p1_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_p3_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_p3_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_p4_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_p4_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_trainer_i1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_trainer_i1.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_trainer_k1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_trainer_k1.py -------------------------------------------------------------------------------- /src/training/module_trainers/ddec_trainer_nt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/ddec_trainer_nt.py -------------------------------------------------------------------------------- /src/training/module_trainers/module_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/module_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_b1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_b1.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_c1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_c1.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_d1.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_d2.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_d3b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_d3b.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_e1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_e1.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/dae_trainer_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/dae_trainer_f1.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/ddec_mclt_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/ddec_mclt_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/ddec_mdct_trainer_b2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/ddec_mdct_trainer_b2.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/ddec_mdct_trainer_b3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/ddec_mdct_trainer_b3.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/ddec_spec_trainer_d1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/ddec_spec_trainer_d1.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/ddec_vae_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/ddec_vae_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/unet_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/unet_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/unet_trainer_b4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/unet_trainer_b4.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/vae_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/vae_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/vae_trainer_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/vae_trainer_c.py -------------------------------------------------------------------------------- /src/training/module_trainers/old/vae_trainer_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/old/vae_trainer_d.py -------------------------------------------------------------------------------- /src/training/module_trainers/unet_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/unet_trainer.py -------------------------------------------------------------------------------- /src/training/module_trainers/unet_trainer_p3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/unet_trainer_p3.py -------------------------------------------------------------------------------- /src/training/module_trainers/unet_trainer_p4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/module_trainers/unet_trainer_p4.py -------------------------------------------------------------------------------- /src/training/muon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/muon.py -------------------------------------------------------------------------------- /src/training/nor_muon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/nor_muon.py -------------------------------------------------------------------------------- /src/training/sigma_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/sigma_sampler.py -------------------------------------------------------------------------------- /src/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/training/trainer.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/compare_dirs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/compare_dirs.py -------------------------------------------------------------------------------- /src/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/config.py -------------------------------------------------------------------------------- /src/utils/dual_diffusion_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/dual_diffusion_utils.py -------------------------------------------------------------------------------- /src/utils/mclt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/mclt.py -------------------------------------------------------------------------------- /src/utils/mdct/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/mdct/__init__.py -------------------------------------------------------------------------------- /src/utils/mdct/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/mdct/functional.py -------------------------------------------------------------------------------- /src/utils/mdct/torch_mdct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/mdct/torch_mdct.py -------------------------------------------------------------------------------- /src/utils/mdct/windows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/mdct/windows.py -------------------------------------------------------------------------------- /src/utils/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/resample.py -------------------------------------------------------------------------------- /src/utils/roseus_colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/parlance-zz/dualdiffusion/HEAD/src/utils/roseus_colormap.py --------------------------------------------------------------------------------