Update api/ltx_server.py
Browse files- api/ltx_server.py +2 -0
api/ltx_server.py
CHANGED
|
@@ -486,6 +486,8 @@ class VideoService:
|
|
| 486 |
print(f"[DEBUG] Aplicando política de precisão: {prec}")
|
| 487 |
if prec == "float8_e4m3fn":
|
| 488 |
self.runtime_autocast_dtype = torch.bfloat16
|
|
|
|
|
|
|
| 489 |
force_promote = os.getenv("LTXV_FORCE_BF16_ON_FP8", "0") == "1"
|
| 490 |
print(f"[DEBUG] FP8 detectado. force_promote={force_promote}")
|
| 491 |
if force_promote and hasattr(torch, "float8_e4m3fn"):
|
|
|
|
| 486 |
print(f"[DEBUG] Aplicando política de precisão: {prec}")
|
| 487 |
if prec == "float8_e4m3fn":
|
| 488 |
self.runtime_autocast_dtype = torch.bfloat16
|
| 489 |
+
|
| 490 |
+
LTXV_FORCE_BF16_ON_FP8 = 1
|
| 491 |
force_promote = os.getenv("LTXV_FORCE_BF16_ON_FP8", "0") == "1"
|
| 492 |
print(f"[DEBUG] FP8 detectado. force_promote={force_promote}")
|
| 493 |
if force_promote and hasattr(torch, "float8_e4m3fn"):
|