diff --git a/calflops/flops_counter_hf.py b/calflops/flops_counter_hf.py index 5d4e21e..6e4eb63 100644 --- a/calflops/flops_counter_hf.py +++ b/calflops/flops_counter_hf.py @@ -29,7 +29,7 @@ def calculate_flops_hf(model_name, empty_model=None, input_shape=None, trust_remote_code=True, - access_token="", + access_token=None, forward_mode="forward", include_backPropagation=False, compute_bp_factor=2.0,