fix:Update batched_denoising.py

2025-05-07 17:58:34 +05:30
parent 89d2d955b1
commit 260fdb39a0
1 changed files with 2 additions and 2 deletions
@@ -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,)
+        return (selected_sigmas,)