diff --git a/env.nu b/env.nu index 0d10c31..601a6e5 100644 --- a/env.nu +++ b/env.nu @@ -160,6 +160,12 @@ export-env { $env.COMFY_CLEAN_ROOT = ($env.COMFY_ROOT | path dirname | path join ComfyClean) path-add 'C:/Portable/TensorRT-8.6.0.12/lib' + + if $nu.os-info.family == 'windows' { + path-add 'G:\BIN\TensorRT-10.7.0.23\lib' + path-add 'G:\BIN\cudnn-windows-x86_64-9.6.0.74_cuda12-archive\bin' + } + path-add ($env.CUDA_ROOT | path join bin) overlay use ../../.venv/Scripts/activate.nu } diff --git a/nodes/image_utils.py b/nodes/image_utils.py index 4b5320f..06c3e70 100644 --- a/nodes/image_utils.py +++ b/nodes/image_utils.py @@ -172,11 +172,6 @@ class MTB_PickFromBatch: # Limit count to the available number of images in the batch count = min(count, batch_size) - if count < batch_size: - log.warning( - f"Requested {count} images, " - f"but only {batch_size} are available." - ) if from_direction == "end": selected_tensors = image[-count:] diff --git a/pyproject.toml b/pyproject.toml index 3f3cf75..93f0300 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,6 +38,7 @@ optional-dependencies = { mel = [ ], dev = [ "black[jupyter]", "codespell", + "marimo", "mypy", "pre-commit", "pytest",