add

app.py

@@ -40,7 +40,7 @@ def calculate_flops_hf(model_name, input_shape, access_token, bp_factor, output_
     model = get_mode_from_hf(model_name=model_name, library="auto", access_token=access_token)
     data, return_print = calculate_flops_in_hugging_space(model_name=model_name, empty_model=model,
                                                           access_token=access_token, input_shape=input_shape,
-                                                          bp_factor=bp_factor, output_unit=output_unit)
+                                                          bp_factor=bp_factor)#, output_unit=output_unit)
     print(f"Data: {data}")
 
     total_flops = data[0]['Forward+Backward FLOPs'] * 10 ** 9  # Convert GFLOPs to FLOPs
@@ -99,10 +99,10 @@ elif calculation_method == "Hugging Face Model":
     input_shape = (batch_size, max_seq_length)
     access_token = st.text_input("Hugging Face Access Token:", "")
     bp_factor = st.number_input("Backward Pass Factor (BP Factor):", min_value=0.0, value=1.0, step=0.1)
-    output_unit = st.selectbox("Output Unit:", ["auto", "MFLOPs", "GFLOPs", "TFLOPs", "PFLOPs"])
+    #output_unit = st.selectbox("Output Unit:", ["auto", "MFLOPs", "GFLOPs", "TFLOPs", "PFLOPs"])
 
     if st.button("Calculate FLOPs (Hugging Face Model)"):
-        result = calculate_flops_hf(model_name, input_shape, access_token, bp_factor, output_unit)
+        result = calculate_flops_hf(model_name, input_shape, access_token, bp_factor) #output_unit)
         st.write(f"Total FLOPs: {result['total_flops']:.2e} FLOPs")
         st.write(f"Meets high impact capabilities criteria: {result['meets_criteria']}")
         st.write("Detailed FLOPs Data:")

model_util.py

@@ -15,8 +15,8 @@ def calculate_flops_in_hugging_space(model_name: str,
                                      empty_model: torch.nn.modules,
                                      access_token: str,
                                      input_shape: tuple,
-                                     bp_factor: float,
-                                     output_unit: str):
+                                     bp_factor: float):
+                                     #output_unit: str):
     "Calculates the FLOPs and Params usage for a model init on `meta` device"
 
     try:
@@ -35,20 +35,20 @@ def calculate_flops_in_hugging_space(model_name: str,
     fw_bp_flops = flops * (1.0 + bp_factor)
     fw_bp_macs = macs * (1.0 + bp_factor)
 
-    if output_unit == "":
-        pass
-    elif output_unit == "auto":
-        params = params_to_string(params, units=None, precision=3)
-        flops = flops_to_string(flops, units=None, precision=3)
-        macs = macs_to_string(macs, units=None, precision=3)
-        fw_bp_flops = flops_to_string(fw_bp_flops, units=None, precision=3)
-        fw_bp_macs = macs_to_string(fw_bp_macs, units=None, precision=3)
-    elif output_unit == "T" or output_unit == "G" or output_unit == "M" or output_unit == "K" or output_unit == "m" or output_unit == "u":
-        params = params_to_string(params, units=output_unit, precision=3)
-        flops = flops_to_string(flops, units=output_unit, precision=3)
-        macs = macs_to_string(macs, units=output_unit, precision=3)
-        fw_bp_flops = flops_to_string(fw_bp_flops, units=output_unit, precision=3)
-        fw_bp_macs = macs_to_string(fw_bp_macs, units=output_unit, precision=3)
+    # if output_unit == "":
+    #     pass
+    # elif output_unit == "auto":
+    #     params = params_to_string(params, units=None, precision=3)
+    #     flops = flops_to_string(flops, units=None, precision=3)
+    #     macs = macs_to_string(macs, units=None, precision=3)
+    #     fw_bp_flops = flops_to_string(fw_bp_flops, units=None, precision=3)
+    #     fw_bp_macs = macs_to_string(fw_bp_macs, units=None, precision=3)
+    # elif output_unit == "T" or output_unit == "G" or output_unit == "M" or output_unit == "K" or output_unit == "m" or output_unit == "u":
+    #     params = params_to_string(params, units=output_unit, precision=3)
+    #     flops = flops_to_string(flops, units=output_unit, precision=3)
+    #     macs = macs_to_string(macs, units=output_unit, precision=3)
+    #     fw_bp_flops = flops_to_string(fw_bp_flops, units=output_unit, precision=3)
+    #     fw_bp_macs = macs_to_string(fw_bp_macs, units=output_unit, precision=3)
 
     return_lines = return_print.split("\n")
     return_start = False