├── LICENSE.txt ├── README.md ├── accelerate_config.yaml ├── assets ├── model.png └── qwen-pruning.png ├── deepspeed_config.json ├── requirement.txt ├── script ├── inference.py ├── inference_10B.py ├── inference_12B.py ├── inference_controlnet.py ├── inference_realism_lora.py ├── inference_txt_ffn.py ├── inference_txt_ffn.sh ├── pipeline_save.py ├── pipeline_save.sh ├── train_phase.py ├── train_phase.sh ├── train_phase_ehs_ffn_proj_initlinear.py └── train_phase_ehs_ffn_proj_initlinear.sh └── src ├── custom_dataset_qwen.py ├── transformer_qwenimage_10B.py ├── transformer_qwenimage_attn.py ├── transformer_qwenimage_attn_ehs_ffn_proj.py ├── transformer_qwenimage_no_txt_no_ffn.py └── transformer_qwenimage_pruning.py /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/README.md -------------------------------------------------------------------------------- /accelerate_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/accelerate_config.yaml -------------------------------------------------------------------------------- /assets/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/assets/model.png -------------------------------------------------------------------------------- /assets/qwen-pruning.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/assets/qwen-pruning.png -------------------------------------------------------------------------------- /deepspeed_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/deepspeed_config.json -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/requirement.txt -------------------------------------------------------------------------------- /script/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference.py -------------------------------------------------------------------------------- /script/inference_10B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference_10B.py -------------------------------------------------------------------------------- /script/inference_12B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference_12B.py -------------------------------------------------------------------------------- /script/inference_controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference_controlnet.py -------------------------------------------------------------------------------- /script/inference_realism_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference_realism_lora.py -------------------------------------------------------------------------------- /script/inference_txt_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference_txt_ffn.py -------------------------------------------------------------------------------- /script/inference_txt_ffn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/inference_txt_ffn.sh -------------------------------------------------------------------------------- /script/pipeline_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/pipeline_save.py -------------------------------------------------------------------------------- /script/pipeline_save.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/pipeline_save.sh -------------------------------------------------------------------------------- /script/train_phase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/train_phase.py -------------------------------------------------------------------------------- /script/train_phase.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/train_phase.sh -------------------------------------------------------------------------------- /script/train_phase_ehs_ffn_proj_initlinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/train_phase_ehs_ffn_proj_initlinear.py -------------------------------------------------------------------------------- /script/train_phase_ehs_ffn_proj_initlinear.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/script/train_phase_ehs_ffn_proj_initlinear.sh -------------------------------------------------------------------------------- /src/custom_dataset_qwen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/src/custom_dataset_qwen.py -------------------------------------------------------------------------------- /src/transformer_qwenimage_10B.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/src/transformer_qwenimage_10B.py -------------------------------------------------------------------------------- /src/transformer_qwenimage_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/src/transformer_qwenimage_attn.py -------------------------------------------------------------------------------- /src/transformer_qwenimage_attn_ehs_ffn_proj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/src/transformer_qwenimage_attn_ehs_ffn_proj.py -------------------------------------------------------------------------------- /src/transformer_qwenimage_no_txt_no_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/src/transformer_qwenimage_no_txt_no_ffn.py -------------------------------------------------------------------------------- /src/transformer_qwenimage_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPPO-Mente-Lab/Qwen-Image-Pruning/HEAD/src/transformer_qwenimage_pruning.py --------------------------------------------------------------------------------