diff --git a/nodes.py b/nodes.py index e1b90ba..5a57a5c 100644 --- a/nodes.py +++ b/nodes.py @@ -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: