float16

app.py

@@ -81,7 +81,7 @@ def calculate_memory(model_name:str, library:str, options:list, access_token:str
     for dtype in options:
         dtype_total_size = total_size
         dtype_largest_layer = largest_layer[0]
-        if dtype == "float16":
+        if dtype in ("float16", "fp16"):
             dtype_total_size /= 2
             dtype_largest_layer /= 2
         elif dtype == "int8":