Update src/model_utils.py
Browse files- src/model_utils.py +1 -1
src/model_utils.py
CHANGED
@@ -11,7 +11,7 @@ from parallelism_utils import estimate_zero1_model_states_mem_needs, estimate_ze
|
|
11 |
|
12 |
DTYPE_MODIFIER = {"float32": 1, "float16/bfloat16": 2, "int8": 4, "int4": 8}
|
13 |
PRECISION = {"Mixed precision": "mixed", "Single precision": "single"}
|
14 |
-
DTYPE = {"float32":
|
15 |
|
16 |
|
17 |
def extract_from_url(name: str):
|
|
|
11 |
|
12 |
DTYPE_MODIFIER = {"float32": 1, "float16/bfloat16": 2, "int8": 4, "int4": 8}
|
13 |
PRECISION = {"Mixed precision": "mixed", "Single precision": "single"}
|
14 |
+
DTYPE = {"float32": "float32", "float16/bfloat16": "float16"}
|
15 |
|
16 |
|
17 |
def extract_from_url(name: str):
|