tests/test_checkpoint_io - hpcaitech/ColossalAI

Making large AI models cheaper, faster and more accessible

0 0 0 Python

	test_gemini_checkpoint_io.py	8.3 KB
	test_gemini_torch_compability.py	6.9 KB
	test_general_checkpoint_io.py	7.8 KB
	test_hybrid_parallel_plugin_checkpoint_io.py	5.8 KB
	test_low_level_zero_checkpoint_io.py	8.5 KB
	test_plugins_huggingface_compatibility.py	2.8 KB
	test_safetensors_async_io.py	5.8 KB
	test_torch_ddp_checkpoint_io.py	3.3 KB
	test_torch_fsdp_checkpoint_io.py	5.8 KB
	utils.py	607 B