├── LICENSE ├── acc_config.yaml ├── images ├── baseline.webp ├── decomposition.webp └── more_reward.webp ├── loss.py ├── modules ├── adapters.py └── lora.py ├── readme.md ├── requirements.txt ├── run_unet.sh ├── sample.py ├── tools.py ├── unet_align_ct5.py ├── unet_lcm.py ├── unet_reward.py └── unet_reward_lcm.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/LICENSE -------------------------------------------------------------------------------- /acc_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/acc_config.yaml -------------------------------------------------------------------------------- /images/baseline.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/images/baseline.webp -------------------------------------------------------------------------------- /images/decomposition.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/images/decomposition.webp -------------------------------------------------------------------------------- /images/more_reward.webp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/images/more_reward.webp -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/loss.py -------------------------------------------------------------------------------- /modules/adapters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/modules/adapters.py -------------------------------------------------------------------------------- /modules/lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/modules/lora.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_unet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/run_unet.sh -------------------------------------------------------------------------------- /sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/sample.py -------------------------------------------------------------------------------- /tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/tools.py -------------------------------------------------------------------------------- /unet_align_ct5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/unet_align_ct5.py -------------------------------------------------------------------------------- /unet_lcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/unet_lcm.py -------------------------------------------------------------------------------- /unet_reward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/unet_reward.py -------------------------------------------------------------------------------- /unet_reward_lcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/luping-liu/LongAlign/HEAD/unet_reward_lcm.py --------------------------------------------------------------------------------