├── .gitmodules ├── LICENSE ├── LucidDreamer ├── .gitignore ├── .gitmodules ├── GAUSSIAN_SPLATTING_LICENSE.md ├── LICENSE.txt ├── README.md ├── arguments │ └── __init__.py ├── configs │ └── stableid.yaml ├── gaussian_renderer │ ├── __init__.py │ └── network_gui.py ├── guidance │ ├── perpneg_utils.py │ ├── sd_step.py │ └── sd_utils.py ├── lora_diffusion │ ├── __init__.py │ ├── cli_lora_add.py │ ├── cli_lora_pti.py │ ├── cli_pt_to_safetensors.py │ ├── cli_svd.py │ ├── dataset.py │ ├── lora.py │ ├── lora_manager.py │ ├── preprocess_files.py │ ├── safe_open.py │ ├── to_ckpt_v2.py │ ├── utils.py │ └── xformers_utils.py ├── scene │ ├── __init__.py │ ├── cameras.py │ ├── dataset_readers.py │ └── gaussian_model.py ├── train.py ├── train.sh └── utils │ ├── camera_utils.py │ ├── general_utils.py │ ├── graphics_utils.py │ ├── image_utils.py │ ├── loss_utils.py │ ├── pointe_utils.py │ ├── sh_utils.py │ └── system_utils.py ├── README.md ├── assets ├── burger.gif ├── golden_crown.gif ├── makeup.gif └── oil_painting.gif ├── datasets_face ├── face_id.py ├── identity_space.yaml ├── pose_input │ └── 1.png ├── selected_celeb_names.txt └── test_data_demo │ ├── 00059.png │ ├── 00673.png │ ├── 00790.png │ └── 01247.png ├── models ├── celeb_embeddings.py ├── embedding_manager.py ├── face_parsing │ ├── infer_onehot.py │ ├── model.py │ └── resnet.py └── id_embedding │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── helpers.cpython-38.pyc │ ├── iresnet.cpython-38.pyc │ └── meta_net.cpython-38.pyc │ ├── helpers.py │ ├── iresnet.py │ └── meta_net.py ├── modelscope_t2v_files ├── text_to_video_synthesis_model.py └── tmp.pt ├── requirements_LucidDreamer.txt ├── requirements_StableIdentity.txt ├── requirements_modelscope.txt ├── test.ipynb ├── test_with_controlnet_openpose.ipynb ├── test_with_modelscope.ipynb ├── train.py ├── train_for_testset.sh └── utils.py /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LICENSE -------------------------------------------------------------------------------- /LucidDreamer/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/.gitignore -------------------------------------------------------------------------------- /LucidDreamer/.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/.gitmodules -------------------------------------------------------------------------------- /LucidDreamer/GAUSSIAN_SPLATTING_LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/GAUSSIAN_SPLATTING_LICENSE.md -------------------------------------------------------------------------------- /LucidDreamer/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/LICENSE.txt -------------------------------------------------------------------------------- /LucidDreamer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/README.md -------------------------------------------------------------------------------- /LucidDreamer/arguments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/arguments/__init__.py -------------------------------------------------------------------------------- /LucidDreamer/configs/stableid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/configs/stableid.yaml -------------------------------------------------------------------------------- /LucidDreamer/gaussian_renderer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/gaussian_renderer/__init__.py -------------------------------------------------------------------------------- /LucidDreamer/gaussian_renderer/network_gui.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/gaussian_renderer/network_gui.py -------------------------------------------------------------------------------- /LucidDreamer/guidance/perpneg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/guidance/perpneg_utils.py -------------------------------------------------------------------------------- /LucidDreamer/guidance/sd_step.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/guidance/sd_step.py -------------------------------------------------------------------------------- /LucidDreamer/guidance/sd_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/guidance/sd_utils.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/__init__.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/cli_lora_add.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/cli_lora_add.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/cli_lora_pti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/cli_lora_pti.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/cli_pt_to_safetensors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/cli_pt_to_safetensors.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/cli_svd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/cli_svd.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/dataset.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/lora.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/lora_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/lora_manager.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/preprocess_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/preprocess_files.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/safe_open.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/safe_open.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/to_ckpt_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/to_ckpt_v2.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/utils.py -------------------------------------------------------------------------------- /LucidDreamer/lora_diffusion/xformers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/lora_diffusion/xformers_utils.py -------------------------------------------------------------------------------- /LucidDreamer/scene/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/scene/__init__.py -------------------------------------------------------------------------------- /LucidDreamer/scene/cameras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/scene/cameras.py -------------------------------------------------------------------------------- /LucidDreamer/scene/dataset_readers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/scene/dataset_readers.py -------------------------------------------------------------------------------- /LucidDreamer/scene/gaussian_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/scene/gaussian_model.py -------------------------------------------------------------------------------- /LucidDreamer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/train.py -------------------------------------------------------------------------------- /LucidDreamer/train.sh: -------------------------------------------------------------------------------- 1 | export CUDA_VISIBLE_DEVICES="0" 2 | python train.py --opt './configs/bagel.yaml' 3 | -------------------------------------------------------------------------------- /LucidDreamer/utils/camera_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/camera_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/general_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/general_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/graphics_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/graphics_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/image_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/loss_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/pointe_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/pointe_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/sh_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/sh_utils.py -------------------------------------------------------------------------------- /LucidDreamer/utils/system_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/LucidDreamer/utils/system_utils.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/README.md -------------------------------------------------------------------------------- /assets/burger.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/assets/burger.gif -------------------------------------------------------------------------------- /assets/golden_crown.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/assets/golden_crown.gif -------------------------------------------------------------------------------- /assets/makeup.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/assets/makeup.gif -------------------------------------------------------------------------------- /assets/oil_painting.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/assets/oil_painting.gif -------------------------------------------------------------------------------- /datasets_face/face_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/face_id.py -------------------------------------------------------------------------------- /datasets_face/identity_space.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/identity_space.yaml -------------------------------------------------------------------------------- /datasets_face/pose_input/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/pose_input/1.png -------------------------------------------------------------------------------- /datasets_face/selected_celeb_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/selected_celeb_names.txt -------------------------------------------------------------------------------- /datasets_face/test_data_demo/00059.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/test_data_demo/00059.png -------------------------------------------------------------------------------- /datasets_face/test_data_demo/00673.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/test_data_demo/00673.png -------------------------------------------------------------------------------- /datasets_face/test_data_demo/00790.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/test_data_demo/00790.png -------------------------------------------------------------------------------- /datasets_face/test_data_demo/01247.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/datasets_face/test_data_demo/01247.png -------------------------------------------------------------------------------- /models/celeb_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/celeb_embeddings.py -------------------------------------------------------------------------------- /models/embedding_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/embedding_manager.py -------------------------------------------------------------------------------- /models/face_parsing/infer_onehot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/face_parsing/infer_onehot.py -------------------------------------------------------------------------------- /models/face_parsing/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/face_parsing/model.py -------------------------------------------------------------------------------- /models/face_parsing/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/face_parsing/resnet.py -------------------------------------------------------------------------------- /models/id_embedding/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/id_embedding/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/id_embedding/__pycache__/helpers.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/__pycache__/helpers.cpython-38.pyc -------------------------------------------------------------------------------- /models/id_embedding/__pycache__/iresnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/__pycache__/iresnet.cpython-38.pyc -------------------------------------------------------------------------------- /models/id_embedding/__pycache__/meta_net.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/__pycache__/meta_net.cpython-38.pyc -------------------------------------------------------------------------------- /models/id_embedding/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/helpers.py -------------------------------------------------------------------------------- /models/id_embedding/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/iresnet.py -------------------------------------------------------------------------------- /models/id_embedding/meta_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/models/id_embedding/meta_net.py -------------------------------------------------------------------------------- /modelscope_t2v_files/text_to_video_synthesis_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/modelscope_t2v_files/text_to_video_synthesis_model.py -------------------------------------------------------------------------------- /modelscope_t2v_files/tmp.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/modelscope_t2v_files/tmp.pt -------------------------------------------------------------------------------- /requirements_LucidDreamer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/requirements_LucidDreamer.txt -------------------------------------------------------------------------------- /requirements_StableIdentity.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/requirements_StableIdentity.txt -------------------------------------------------------------------------------- /requirements_modelscope.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/requirements_modelscope.txt -------------------------------------------------------------------------------- /test.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/test.ipynb -------------------------------------------------------------------------------- /test_with_controlnet_openpose.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/test_with_controlnet_openpose.ipynb -------------------------------------------------------------------------------- /test_with_modelscope.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/test_with_modelscope.ipynb -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/train.py -------------------------------------------------------------------------------- /train_for_testset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/train_for_testset.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qinghew/StableIdentity/HEAD/utils.py --------------------------------------------------------------------------------