diff --git a/invokeai/frontend/web/src/features/gallery/components/ImageGrid/GalleryImageGrid.tsx b/invokeai/frontend/web/src/features/gallery/components/ImageGrid/GalleryImageGrid.tsx index 858eeedaa3..8b44b39ae9 100644 --- a/invokeai/frontend/web/src/features/gallery/components/ImageGrid/GalleryImageGrid.tsx +++ b/invokeai/frontend/web/src/features/gallery/components/ImageGrid/GalleryImageGrid.tsx @@ -118,6 +118,20 @@ const GalleryImageGrid = () => { ); }, [dispatch, imageNames.length, galleryView]); + useEffect(() => { + // Set up gallery scroler + const { current: root } = rootRef; + if (scroller && root) { + initialize({ + target: root, + elements: { + viewport: scroller, + }, + }); + } + return () => osInstance()?.destroy(); + }, [scroller, initialize, osInstance]); + const handleEndReached = useMemo(() => { if (areMoreAvailable) { return handleLoadMoreImages;