cerebras.modelzoo.common.run_gpt_inference

GPT Inference script built using the cstorch API

Functions

get_cluster_config
get_parser
inference_input_dataloader
main

previous

cerebras.modelzoo.common.run_cstorch_flow.GradScalerParams

next

cerebras.modelzoo.common.run_gpt_inference.get_cluster_config