Skip to content

tinker_cookbook.hyperparam_utils.get_full_finetune_param_count

tinker_cookbook.hyperparam_utils.get_full_finetune_param_count(model_name)

Get the total parameter count for a model by reading safetensors headers.

Parameters:

  • model_name (str) – HuggingFace model identifier.

Returns: Total number of parameters as a float.