From 260fdb39a01ee8a10174ee00a40cfd17b8af6690 Mon Sep 17 00:00:00 2001
From: Prakarsh Kaushik <66624139+RUFFY-369@users.noreply.github.com>
Date: Wed, 7 May 2025 17:58:34 +0530
Subject: [PATCH] fix:Update batched_denoising.py

---
 src/batched_denoising/batched_denoising.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/batched_denoising/batched_denoising.py b/src/batched_denoising/batched_denoising.py
index 04a155a..5d2e727 100644
--- a/src/batched_denoising/batched_denoising.py
+++ b/src/batched_denoising/batched_denoising.py
@@ -1,5 +1,5 @@
 import torch
-from control_base import ControlNodeBase
+from .control_base import ControlNodeBase
 import comfy.model_management
 import comfy.samplers
 import math
@@ -776,4 +776,4 @@ class StreamScheduler(ControlNodeBase):
         selected_sigmas = torch.tensor(
             selected_sigmas, dtype=torch.float32, device=device
         )
-        return (selected_sigmas,)
\ No newline at end of file
+        return (selected_sigmas,)