MrYXJ commited on
Commit
d77727f
1 Parent(s): 4d8e2ae

Update model_utils.py

Browse files
Files changed (1) hide show
  1. model_utils.py +1 -5
model_utils.py CHANGED
@@ -37,12 +37,8 @@ def calculate_flops_in_hugging_space(model_name: str,
37
  "Calculates the FLOPs and Params usage for a model init on `meta` device"
38
 
39
  try:
40
- # print("model_name:", model_name)
41
- # print("empty_model:", empty_model)
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,
 
37
  "Calculates the FLOPs and Params usage for a model init on `meta` device"
38
 
39
  try:
 
 
 
 
40
  flops, macs, params, return_print = calculate_flops_hf(model_name=model_name,
41
+ empty_model=empty_model,
42
  input_shape=input_shape,
43
  access_token=access_token,
44
  output_as_string=False,