MrYXJ commited on
Commit
c51def8
1 Parent(s): d49aff6

Update model_utils.py

Browse files
Files changed (1) hide show
  1. model_utils.py +1 -1
model_utils.py CHANGED
@@ -42,7 +42,7 @@ def calculate_flops_in_hugging_space(model_name: str,
42
  # print("access_token:", access_token)
43
  # print("input_shape:", input_shape)
44
  flops, macs, params, return_print = calculate_flops_hf(model_name=model_name,
45
- empty_model=None,
46
  input_shape=input_shape,
47
  access_token=access_token,
48
  output_as_string=False,
 
42
  # print("access_token:", access_token)
43
  # print("input_shape:", input_shape)
44
  flops, macs, params, return_print = calculate_flops_hf(model_name=model_name,
45
+ # empty_model=empty_model,
46
  input_shape=input_shape,
47
  access_token=access_token,
48
  output_as_string=False,