Refactor import statements for VEnhancer modules and inference_utils.py

nodes/multi_gpu_loader.py

@@ -1,6 +1,6 @@
 from typing import Dict, Any, Tuple
-from src.VEnhancer.configs.distributred_venhancer_config import DistributedConfig
-from src.VEnhancer.enhance_a_video_MultiGPU import DistributedVEnhancer
+from VEnhancer.configs.distributred_venhancer_config import DistributedConfig
+from VEnhancer.enhance_a_video_MultiGPU import DistributedVEnhancer
 
 
 class MultiGPUVEnhancerLoader:

nodes/single_gpu_loader.py

@@ -1,8 +1,8 @@
 """Node for loading VEnhancer model in single GPU mode."""
 
 from typing import Dict, Any, Tuple
-from src.VEnhancer.configs.venhnacer_config import VEnhancerConfig
-from src.VEnhancer.enhance_a_video import VEnhancer
+from VEnhancer.configs.venhnacer_config import VEnhancerConfig
+from VEnhancer.enhance_a_video import VEnhancer
 
 class SingleGPUVEnhancerLoader:
     @classmethod

pyproject.toml

@@ -1,11 +1,7 @@
-[build-system]
-requires = ["setuptools>=61.0"]
-build-backend = "setuptools.build_meta"
-
 [project]
 name = "ve-enhancer-comfyui"
 version = "0.1.0"
-description = "ComfyUI workflow for VE Enhancer Inference"
+description = "ComfyUI workflow for VEnhancer Inference"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
@@ -26,12 +22,16 @@ dependencies = [
     "accelerate>=0.20.0",
     "safetensors>=0.3.0",
     "sentencepiece",
-    "bitsandbytes"
+    "bitsandbytes",
+    "pydantic-settings"
 ]
 
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
 [tool.setuptools]
-packages = ["nodes"]  # Only include the nodes package
-exclude = ["VEnhancer*"]  # Exclude the VEnhancer package
+packages = ["VEnhancer", "nodes"]
 
 [tool.ruff]
 line-length = 120