Update model_utils.py

model_utils.py

@@ -37,6 +37,10 @@ def calculate_flops_in_hugging_space(model_name: str,
     "Calculates the FLOPs and Params usage for a model init on `meta` device"
 
     try:
+        print("model_name:", model_name)
+        print("empty_model:", empty_model)
+        print("access_token:", access_token)
+        print("input_shape:", input_shape)
         flops, macs, params, return_print  = calculate_flops_hf(model_name=model_name,
                                                                 empty_model=empty_model,
                                                                 access_token=access_token,