├── README.md └── SeMani-Trans ├── README.md ├── clip_openai ├── __init__.py ├── bpe_simple_vocab_16e6.txt.gz ├── clip.py ├── model.py └── simple_tokenizer.py ├── dalle_pytorch ├── __init__.py ├── attention.py ├── clip_loss.py ├── dalle_pytorch.py ├── data │ └── bpe_simple_vocab_16e6.txt ├── distributed_backends │ ├── __init__.py │ ├── deepspeed_backend.py │ ├── distributed_backend.py │ ├── dummy_backend.py │ └── horovod_backend.py ├── distributed_utils.py ├── generate_engine.py ├── lamb.py ├── loader.py ├── lr_scheduler.py ├── reversible.py ├── tokenizer.py ├── transformer.py ├── vae.py └── val_engine.py ├── evaluate.py ├── finetune.py ├── taming ├── data │ ├── ade20k.py │ ├── annotated_objects_coco.py │ ├── annotated_objects_dataset.py │ ├── base.py │ ├── coco.py │ ├── conditional_builder │ │ ├── objects_bbox.py │ │ ├── objects_center_points.py │ │ └── utils.py │ ├── custom.py │ ├── faceshq.py │ ├── helper_types.py │ ├── image_transforms.py │ ├── imagenet.py │ ├── sflckr.py │ └── utils.py ├── lr_scheduler.py ├── main.py ├── models │ ├── cond_transformer.py │ ├── dummy_cond_stage.py │ └── vqgan.py ├── modules │ ├── diffusionmodules │ │ └── model.py │ ├── discriminator │ │ └── model.py │ ├── losses │ │ ├── __init__.py │ │ ├── lpips.py │ │ ├── segmentation.py │ │ └── vqperceptual.py │ ├── misc │ │ └── coord.py │ ├── transformer │ │ ├── mingpt.py │ │ └── permuter.py │ ├── util.py │ └── vqvae │ │ └── quantize.py └── util.py ├── train.py └── util ├── __init__.py └── html.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/README.md -------------------------------------------------------------------------------- /SeMani-Trans/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/README.md -------------------------------------------------------------------------------- /SeMani-Trans/clip_openai/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /SeMani-Trans/clip_openai/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/clip_openai/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /SeMani-Trans/clip_openai/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/clip_openai/clip.py -------------------------------------------------------------------------------- /SeMani-Trans/clip_openai/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/clip_openai/model.py -------------------------------------------------------------------------------- /SeMani-Trans/clip_openai/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/clip_openai/simple_tokenizer.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/__init__.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/attention.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/clip_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/clip_loss.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/dalle_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/dalle_pytorch.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/data/bpe_simple_vocab_16e6.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/data/bpe_simple_vocab_16e6.txt -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/distributed_backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/distributed_backends/__init__.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/distributed_backends/deepspeed_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/distributed_backends/deepspeed_backend.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/distributed_backends/distributed_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/distributed_backends/distributed_backend.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/distributed_backends/dummy_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/distributed_backends/dummy_backend.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/distributed_backends/horovod_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/distributed_backends/horovod_backend.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/distributed_utils.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/generate_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/generate_engine.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/lamb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/lamb.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/loader.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/lr_scheduler.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/reversible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/reversible.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/tokenizer.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/transformer.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/vae.py -------------------------------------------------------------------------------- /SeMani-Trans/dalle_pytorch/val_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/dalle_pytorch/val_engine.py -------------------------------------------------------------------------------- /SeMani-Trans/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/evaluate.py -------------------------------------------------------------------------------- /SeMani-Trans/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/finetune.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/ade20k.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/annotated_objects_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/annotated_objects_coco.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/annotated_objects_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/annotated_objects_dataset.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/base.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/coco.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/conditional_builder/objects_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/conditional_builder/objects_bbox.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/conditional_builder/objects_center_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/conditional_builder/objects_center_points.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/conditional_builder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/conditional_builder/utils.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/custom.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/faceshq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/faceshq.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/helper_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/helper_types.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/image_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/image_transforms.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/imagenet.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/sflckr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/sflckr.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/data/utils.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/lr_scheduler.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/main.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/models/cond_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/models/cond_transformer.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/models/dummy_cond_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/models/dummy_cond_stage.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/models/vqgan.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/discriminator/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/discriminator/model.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/losses/__init__.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/losses/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/losses/lpips.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/losses/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/losses/segmentation.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/misc/coord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/misc/coord.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/transformer/mingpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/transformer/mingpt.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/transformer/permuter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/transformer/permuter.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/util.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/modules/vqvae/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/modules/vqvae/quantize.py -------------------------------------------------------------------------------- /SeMani-Trans/taming/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/taming/util.py -------------------------------------------------------------------------------- /SeMani-Trans/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/train.py -------------------------------------------------------------------------------- /SeMani-Trans/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SeMani-Trans/util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Yikai-Wang/SeMani/HEAD/SeMani-Trans/util/html.py --------------------------------------------------------------------------------