feat: add version constant and get_version() helper

__init__.py

@@ -1,3 +1,10 @@
 from .comfyui.comfyui_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
 
+__version__ = "1.0.1"
+
 __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
+
+
+def get_version():
+    """Return the VideoX-Fun version."""
+    return __version__

comfyui/flux2/nodes.py

@@ -598,10 +598,10 @@ def convert_mistral3_to_diffusers(state_dict):
             [os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "models/Diffusion_Transformer")] # Possible folder names to check
         try:
             tokenizer_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="flux2_tokenizer")
-        except:
+        except Exception:
             try:
                 tokenizer_path = os.path.join(search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="FLUX.2-dev"), "tokenizer")
-            except:
+            except Exception:
                 tokenizer_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Mistral-Nemo-Instruct-2407")
 
         tokenizer = PixtralProcessor.from_pretrained(tokenizer_path)

comfyui/qwenimage/nodes.py

@@ -476,10 +476,10 @@ def loadmodel(self, model_name, precision,):
             [os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "models/Diffusion_Transformer")] # Possible folder names to check
         try:
             tokenizer_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="qwen2_tokenizer")
-        except:
+        except Exception:
             try:
                 tokenizer_path = os.path.join(search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Qwen-Image"), "tokenizer")
-            except:
+            except Exception:
                 tokenizer_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Qwen2.5-VL-7B-Instruct")
 
         tokenizer = Qwen2Tokenizer.from_pretrained(tokenizer_path)
@@ -504,10 +504,10 @@ def loadmodel(self, ):
             [os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "models/Diffusion_Transformer")] # Possible folder names to check
         try:
             processor_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="qwen2_processor")
-        except:
+        except Exception:
             try:
                 processor_path = os.path.join(search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Qwen-Image-Edit"), "processor")
-            except:
+            except Exception:
                 processor_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Qwen2.5-VL-7B-Instruct")
 
         # Get processor

comfyui/z_image/nodes.py

@@ -435,10 +435,10 @@ def loadmodel(self, model_name, precision,):
             [os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "models/Diffusion_Transformer")] # Possible folder names to check
         try:
             tokenizer_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="qwen3_tokenizer")
-        except:
+        except Exception:
             try:
                 tokenizer_path = os.path.join(search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Z-Image-Turbo"), "tokenizer")
-            except:
+            except Exception:
                 tokenizer_path = search_sub_dir_in_possible_folders(possible_folders, sub_dir_name="Qwen3-4B")
 
         tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)

scripts/cogvideox_fun/train.py

@@ -1001,7 +1001,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/cogvideox_fun/train_control.py

@@ -961,7 +961,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/cogvideox_fun/train_lora.py

@@ -1008,7 +1008,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/cogvideox_fun/train_reward_lora.py

@@ -969,7 +969,7 @@ def save_model_hook(models, weights, output_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/fantasytalking/train.py

@@ -1013,7 +1013,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/flux/train.py

@@ -1023,7 +1023,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/flux/train_lora.py

@@ -1021,7 +1021,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/flux2/train.py

@@ -1115,7 +1115,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/flux2/train_lora.py

@@ -1082,7 +1082,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/flux2_fun/train_control.py

@@ -1137,7 +1137,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/flux2_fun/train_control_distill.py

@@ -1127,7 +1127,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/hunyuanvideo/train.py

@@ -1136,7 +1136,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/hunyuanvideo/train_lora.py

@@ -1137,7 +1137,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/longcatvideo/train.py

@@ -1017,7 +1017,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/longcatvideo/train_avatar.py

@@ -1057,7 +1057,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/longcatvideo/train_avatar_lora.py

@@ -1044,7 +1044,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/longcatvideo/train_lora.py

@@ -1000,7 +1000,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/qwenimage/train.py

@@ -924,7 +924,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/qwenimage/train_edit.py

@@ -960,7 +960,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/qwenimage/train_edit_lora.py

@@ -962,7 +962,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/qwenimage/train_lora.py

@@ -918,7 +918,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/qwenimage_fun/train_control.py

@@ -954,7 +954,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/qwenimage_instantx/train_control.py

@@ -947,7 +947,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/turbodiffusion/train_distill.py

@@ -1071,7 +1071,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1/train.py

@@ -1079,7 +1079,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1/train_distill.py

@@ -1070,7 +1070,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1/train_distill_lora.py

@@ -1130,7 +1130,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1/train_lora.py

@@ -1068,7 +1068,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1/train_reward_lora.py

@@ -978,7 +978,7 @@ def save_model_hook(models, weights, output_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1_fun/train.py

@@ -1037,7 +1037,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1_fun/train_control.py

@@ -983,7 +983,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1_fun/train_control_lora.py

@@ -981,7 +981,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1_fun/train_lora.py

@@ -1030,7 +1030,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1_fun/train_reward_lora.py

@@ -990,7 +990,7 @@ def save_model_hook(models, weights, output_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.1_vace/train.py

@@ -968,7 +968,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.2/train.py

@@ -1097,7 +1097,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )

scripts/wan2.2/train_animate.py

@@ -1061,7 +1061,7 @@ def load_model_hook(models, input_dir):
     elif args.use_came:
         try:
             from came_pytorch import CAME
-        except:
+        except Exception:
             raise ImportError(
                 "Please install came_pytorch to use CAME. You can do so by running `pip install came_pytorch`"
             )