Ouzhang
/

hku_diffusion_dllm

Model card Files Files and versions

hku_diffusion_dllm / reference /code /flm /configs /strategy /fsdp.yaml

Ouzhang's picture

Add files using upload-large-folder tool

31dc8dc verified about 1 month ago

History Blame Contribute Delete

142 Bytes

	# TODO(yair): Currenly not compatible with grad clipping
	_target_: lightning.pytorch.strategies.FSDPStrategy
	sharding_strategy: SHARD_GRAD_OP