From db697fea11c59491cb6f0b9f2eb66c3e2c9d4ab1 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 12 Nov 2024 00:22:26 +0200
Subject: [PATCH] Update model_loading.py

---
 model_loading.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/model_loading.py b/model_loading.py
index 1fcfbb6..08ddcee 100644
--- a/model_loading.py
+++ b/model_loading.py
@@ -113,6 +113,9 @@ class DownloadAndLoadCogVideoModel:
     def loadmodel(self, model, precision, fp8_transformer="disabled", compile="disabled", 
                   enable_sequential_cpu_offload=False, pab_config=None, block_edit=None, lora=None, compile_args=None, 
                   attention_mode="sdpa", load_device="main_device"):
+        
+        if precision == "fp16" and "1.5" in model:
+            raise ValueError("1.5 models do not work in fp16")
 
         device = mm.get_torch_device()
         offload_device = mm.unet_offload_device()