add error for now

This commit is contained in:
kijai 2024-09-18 21:27:57 +03:00
parent 1bd74f96b5
commit 9174d3e7dc

View File

@ -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: