Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

pixelart fix np array as input

Browse Source
This commit is contained in:
Disty0
2025-08-22 22:53:25 +03:00
parent 35d45c5088
commit a073c63e09
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/postprocess/pixelart.py
+5 -5
View File
@@ -171,12 +171,12 @@ def process_image_input(images: PipelineImageInput) -> torch.ByteTensor:
img = torch.from_numpy(np.asarray(img).copy()).unsqueeze(0)
combined_images.append(img)
elif isinstance(img, np.ndarray):
if len(img.shape) == 3:
img = img.unsqueeze(0)
img = torch.from_numpy(img)
if img.ndim == 3:
img = img.unsqueeze(0)
combined_images.append(img)
elif isinstance(img, torch.Tensor):
if len(img.shape) == 3:
if img.ndim == 3:
img = img.unsqueeze(0)
combined_images.append(img)
else:
@@ -186,11 +186,11 @@ def process_image_input(images: PipelineImageInput) -> torch.ByteTensor:
combined_images = torch.from_numpy(np.asarray(images).copy()).unsqueeze(0)
elif isinstance(images, np.ndarray):
combined_images = torch.from_numpy(images)
if len(combined_images.shape) == 3:
if combined_images.ndim == 3:
combined_images = combined_images.unsqueeze(0)
elif isinstance(images, torch.Tensor):
combined_images = images
if len(combined_images.shape) == 3:
if combined_images.ndim == 3:
combined_images = combined_images.unsqueeze(0)
else:
raise RuntimeError(f"Invalid input! Given: {type(images)} should be in ('torch.Tensor', 'np.ndarray', 'PIL.Image.Image')")