├── .gitignore ├── .python-version ├── LICENSE ├── README.md ├── SCOP ├── README.md ├── __main__.py ├── data_processing.py ├── dataset_reader.py ├── filters.py ├── models.py ├── process_masks │ └── __main__.py └── visualize.py ├── TENOR ├── README.md ├── flux │ ├── .gitignore │ ├── README.md │ ├── compass_train_lora.py │ ├── image_datasets │ │ ├── __init__.py │ │ └── scop.py │ ├── models_licence │ │ └── LICENSE-FLUX1-dev │ ├── scripts │ │ └── train.sh │ ├── src │ │ └── flux │ │ │ ├── controlnet.py │ │ │ ├── math.py │ │ │ ├── model.py │ │ │ ├── modules │ │ │ ├── autoencoder.py │ │ │ ├── conditioner.py │ │ │ └── layers.py │ │ │ ├── sampling.py │ │ │ ├── util.py │ │ │ └── xflux_pipeline.py │ └── train_configs │ │ └── compass.yaml └── sd │ ├── README.md │ ├── scripts │ ├── sd14.sh │ ├── sd15.sh │ └── sd21.sh │ └── src │ ├── attn_utils.py │ ├── compass_train_unet.py │ ├── loss_utils.py │ └── scop.py ├── assets └── teaser.avif ├── pyproject.toml ├── setup_env.sh └── uv.lock /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/.gitignore -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | >=3.12, <3.13 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/README.md -------------------------------------------------------------------------------- /SCOP/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/README.md -------------------------------------------------------------------------------- /SCOP/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/__main__.py -------------------------------------------------------------------------------- /SCOP/data_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/data_processing.py -------------------------------------------------------------------------------- /SCOP/dataset_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/dataset_reader.py -------------------------------------------------------------------------------- /SCOP/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/filters.py -------------------------------------------------------------------------------- /SCOP/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/models.py -------------------------------------------------------------------------------- /SCOP/process_masks/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/process_masks/__main__.py -------------------------------------------------------------------------------- /SCOP/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/SCOP/visualize.py -------------------------------------------------------------------------------- /TENOR/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/README.md -------------------------------------------------------------------------------- /TENOR/flux/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/.gitignore -------------------------------------------------------------------------------- /TENOR/flux/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/README.md -------------------------------------------------------------------------------- /TENOR/flux/compass_train_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/compass_train_lora.py -------------------------------------------------------------------------------- /TENOR/flux/image_datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /TENOR/flux/image_datasets/scop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/image_datasets/scop.py -------------------------------------------------------------------------------- /TENOR/flux/models_licence/LICENSE-FLUX1-dev: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/models_licence/LICENSE-FLUX1-dev -------------------------------------------------------------------------------- /TENOR/flux/scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/scripts/train.sh -------------------------------------------------------------------------------- /TENOR/flux/src/flux/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/controlnet.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/math.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/model.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/modules/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/modules/autoencoder.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/modules/conditioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/modules/conditioner.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/modules/layers.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/sampling.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/util.py -------------------------------------------------------------------------------- /TENOR/flux/src/flux/xflux_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/src/flux/xflux_pipeline.py -------------------------------------------------------------------------------- /TENOR/flux/train_configs/compass.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/flux/train_configs/compass.yaml -------------------------------------------------------------------------------- /TENOR/sd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/README.md -------------------------------------------------------------------------------- /TENOR/sd/scripts/sd14.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/scripts/sd14.sh -------------------------------------------------------------------------------- /TENOR/sd/scripts/sd15.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/scripts/sd15.sh -------------------------------------------------------------------------------- /TENOR/sd/scripts/sd21.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/scripts/sd21.sh -------------------------------------------------------------------------------- /TENOR/sd/src/attn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/src/attn_utils.py -------------------------------------------------------------------------------- /TENOR/sd/src/compass_train_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/src/compass_train_unet.py -------------------------------------------------------------------------------- /TENOR/sd/src/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/src/loss_utils.py -------------------------------------------------------------------------------- /TENOR/sd/src/scop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/TENOR/sd/src/scop.py -------------------------------------------------------------------------------- /assets/teaser.avif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/assets/teaser.avif -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/setup_env.sh -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/blurgyy/CoMPaSS/HEAD/uv.lock --------------------------------------------------------------------------------