From bc443d459e24dbdff2c9f9f7d31591340ddb78d8 Mon Sep 17 00:00:00 2001
From: Vikramjeet <singh.vikram.1782000@gmail.com>
Date: Wed, 13 Nov 2024 19:56:48 +0530
Subject: [PATCH] Refactor import statements for VEnhancer modules and
 inference_utils.py

---
 nodes/multi_gpu_loader.py  |  4 ++--
 nodes/single_gpu_loader.py |  4 ++--
 pyproject.toml             | 16 ++++++++--------
 3 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/nodes/multi_gpu_loader.py b/nodes/multi_gpu_loader.py
index 1e4ddab..01420cc 100644
--- a/nodes/multi_gpu_loader.py
+++ b/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:
diff --git a/nodes/single_gpu_loader.py b/nodes/single_gpu_loader.py
index 2d62b68..811d80c 100644
--- a/nodes/single_gpu_loader.py
+++ b/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
diff --git a/pyproject.toml b/pyproject.toml
index 530d449..3f60352 100644
--- a/pyproject.toml
+++ b/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
\ No newline at end of file