Merge pull request #5 from mihaibirsan/main

Allow image selection for arbitrary sizes
This commit is contained in:
SLAPaper Pang
2024-01-10 18:02:25 +08:00
committed by GitHub
+1 -1
View File
@@ -95,7 +95,7 @@ class ImageSelector:
if selected_index:
print(f"ImageSelector: selected: {len(selected_index)} images")
return (images[selected_index, :, :, :], )
return (images[selected_index], )
print(f"ImageSelector: selected no images, passthrough")
return (images, )