Fabrice-TIERCELIN
commited on
" rather than '
Browse files- hyvideo/constants.py +3 -3
hyvideo/constants.py
CHANGED
@@ -17,9 +17,9 @@ __all__ = [
|
|
17 |
]
|
18 |
|
19 |
PRECISION_TO_TYPE = {
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
}
|
24 |
|
25 |
# =================== Constant Values =====================
|
|
|
17 |
]
|
18 |
|
19 |
PRECISION_TO_TYPE = {
|
20 |
+
"fp32": torch.float32,
|
21 |
+
"fp16": torch.float16,
|
22 |
+
"bf16": torch.bfloat16,
|
23 |
}
|
24 |
|
25 |
# =================== Constant Values =====================
|