Update app.py

app.py

@@ -145,14 +145,14 @@ import_custom_nodes()
 downloadandloadcogvideomodel = NODE_CLASS_MAPPINGS[
     "DownloadAndLoadCogVideoModel"
 ]()
-downloadandloadcogvideomodel_1 = downloadandloadcogvideomodel.loadmodel(
-    model="THUDM/CogVideoX-5b",
-    precision="bf16",
-    quantization="disabled",
-    enable_sequential_cpu_offload=True,
-    attention_mode="sdpa",
-    load_device="main_device",
-)
+# downloadandloadcogvideomodel_1 = downloadandloadcogvideomodel.loadmodel(
+#     model="THUDM/CogVideoX-5b",
+#     precision="bf16",
+#     quantization="disabled",
+#     enable_sequential_cpu_offload=True,
+#     attention_mode="sdpa",
+#     load_device="main_device",
+# )
 loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
 cliploader = NODE_CLASS_MAPPINGS["CLIPLoader"]()
 cliploader_20 = cliploader.load_clip(
@@ -170,7 +170,8 @@ cr_upscale_image = NODE_CLASS_MAPPINGS["CR Upscale Image"]()
 vhs_videocombine = NODE_CLASS_MAPPINGS["VHS_VideoCombine"]()
 
 #Add all the models that load a safetensors file
-model_loaders = [downloadandloadcogvideomodel_1, cliploader_20]
+model_loaders = [cliploader_20]
+# model_loaders = [downloadandloadcogvideomodel_1, cliploader_20]
 
 # Check which models are valid and how to best load them
 valid_models = [
@@ -196,14 +197,14 @@ def generate_video(positive_prompt, num_frames, input_image):
             # downloadandloadcogvideomodel = NODE_CLASS_MAPPINGS[
             #     "DownloadAndLoadCogVideoModel"
             # ]()
-            # downloadandloadcogvideomodel_1 = downloadandloadcogvideomodel.loadmodel(
-            #     model="THUDM/CogVideoX-5b",
-            #     precision="bf16",
-            #     quantization="disabled",
-            #     enable_sequential_cpu_offload=True,
-            #     attention_mode="sdpa",
-            #     load_device="main_device",
-            # )
+            downloadandloadcogvideomodel_1 = downloadandloadcogvideomodel.loadmodel(
+                model="THUDM/CogVideoX-5b",
+                precision="bf16",
+                quantization="disabled",
+                enable_sequential_cpu_offload=True,
+                attention_mode="sdpa",
+                load_device="main_device",
+            )
     
             # loadimage = NODE_CLASS_MAPPINGS["LoadImage"]()
             loadimage_8 = loadimage.load_image(image=input_image)