Support diffusion models with scaled fp8 weights.

2025-09-11 03:58:22 +00:00 · 2024-10-19 23:47:42 -04:00
parent 73e3a9e676
commit a68bbafddb
7 changed files with 95 additions and 23 deletions
--- a/comfy/supported_models_base.py
+++ b/comfy/supported_models_base.py
@@ -49,6 +49,7 @@ class BASE:

    manual_cast_dtype = None
    custom_operations = None
+    scaled_fp8 = False
    optimizations = {"fp8": False}

    @classmethod