cerebras.modelzoo.common.run_gpt_inference.load_from_checkpoint_file#

cerebras.modelzoo.common.run_gpt_inference.load_from_checkpoint_file(checkpoint_path)[source]#

Loads state dict from checkpoint path and checks for version compatibilty.

cerebras.modelzoo.common.run_gpt_inference.inference_input_dataloader