Ali-C137 commited on
Commit
5b409eb
1 Parent(s): 2f8af2b

Update src/display/utils.py

Browse files

Revoke support for float32 precision

Files changed (1) hide show
  1. src/display/utils.py +3 -3
src/display/utils.py CHANGED
@@ -93,7 +93,7 @@ class WeightType(Enum):
93
  Delta = ModelDetails("Delta")
94
 
95
  class Precision(Enum):
96
- float32 = ModelDetails("float32")
97
  float16 = ModelDetails("float16")
98
  bfloat16 = ModelDetails("bfloat16")
99
  qt_8bit = ModelDetails("8bit")
@@ -102,8 +102,8 @@ class Precision(Enum):
102
  Unknown = ModelDetails("?")
103
 
104
  def from_str(precision):
105
- if precision in ["torch.float32", "float32"]:
106
- return Precision.float32
107
  if precision in ["torch.float16", "float16"]:
108
  return Precision.float16
109
  if precision in ["torch.bfloat16", "bfloat16"]:
 
93
  Delta = ModelDetails("Delta")
94
 
95
  class Precision(Enum):
96
+ # float32 = ModelDetails("float32")
97
  float16 = ModelDetails("float16")
98
  bfloat16 = ModelDetails("bfloat16")
99
  qt_8bit = ModelDetails("8bit")
 
102
  Unknown = ModelDetails("?")
103
 
104
  def from_str(precision):
105
+ # if precision in ["torch.float32", "float32"]:
106
+ # return Precision.float32
107
  if precision in ["torch.float16", "float16"]:
108
  return Precision.float16
109
  if precision in ["torch.bfloat16", "bfloat16"]: