jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
21 lines
744 B
Python
21 lines
744 B
Python
import pathlib
|
|
from PIL import Image
|
|
import usdu_utils
|
|
|
|
|
|
def save_image(tensor, path: pathlib.Path):
|
|
"""The goto function to save a tensor image to the sampled images directory."""
|
|
assert tensor.ndim == 3 or (tensor.ndim == 4 and tensor.shape[0] == 1), (
|
|
f"Expected a 3D tensor (H, W, C) or (1, H, W, C), got {tensor.ndim=}"
|
|
)
|
|
if tensor.ndim == 3:
|
|
tensor = tensor.unsqueeze(0)
|
|
image = usdu_utils.tensor_to_pil(tensor.cpu())
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
image.save(path, quality=75, optimize=True)
|
|
|
|
|
|
def load_image(path: pathlib.Path, device=None):
|
|
"""Load an image from disk and convert it to a tensor."""
|
|
return usdu_utils.pil_to_tensor(Image.open(path)).to(device=device)
|