add error for now

2024-09-18 21:27:57 +03:00
parent 1bd74f96b5
commit 9174d3e7dc
1 changed files with 3 additions and 0 deletions
@@ -57,6 +57,9 @@ class DownloadAndLoadCogVideoModel:
        offload_device = mm.unet_offload_device()
        mm.soft_empty_cache()

+        if "I2V" in model and fp8_transformer != "disabled":
+            raise NotImplementedError("fp8_transformer is not implemented yet for I2V -model")
+
        dtype = {"bf16": torch.bfloat16, "fp16": torch.float16, "fp32": torch.float32}[precision]

        if "Fun" in model: