diff --git a/src/pages/playground/components/ground-images.tsx b/src/pages/playground/components/ground-images.tsx index 32f4ddac..e627d7b9 100644 --- a/src/pages/playground/components/ground-images.tsx +++ b/src/pages/playground/components/ground-images.tsx @@ -135,8 +135,8 @@ const GroundImages: React.FC = forwardRef((props, ref) => { }; }); - const getNewImageSizeOptions = useCallback((metaData: any) => { - const { max_height, max_width } = metaData || {}; + const paramsConfig = useMemo(() => { + const { max_height, max_width } = modelMeta; if (!max_height || !max_width) { return ImageParamsConfig; } @@ -155,12 +155,7 @@ const GroundImages: React.FC = forwardRef((props, ref) => { value: `${max_width}x${max_height}` }); } - return newImageSizeOptions; - }, []); - - const paramsConfig = useMemo(() => { - const newImageSizeOptions = getNewImageSizeOptions(modelMeta); - let result = ImageParamsConfig.map((item) => { + let result: ParamsSchema[] = ImageParamsConfig.map((item: ParamsSchema) => { if (item.name === 'size') { return { ...item, @@ -173,7 +168,7 @@ const GroundImages: React.FC = forwardRef((props, ref) => { result = result.filter((item) => item.name !== 'size'); } return result; - }, [modelMeta, getNewImageSizeOptions]); + }, [modelMeta]); const generateNumber = (min: number, max: number) => { return Math.floor(Math.random() * (max - min + 1) + min); diff --git a/src/pages/playground/components/image-edit.tsx b/src/pages/playground/components/image-edit.tsx index 5e3eb6e3..9aec441c 100644 --- a/src/pages/playground/components/image-edit.tsx +++ b/src/pages/playground/components/image-edit.tsx @@ -155,8 +155,8 @@ const GroundImages: React.FC = forwardRef((props, ref) => { }; }; - const getNewImageSizeOptions = useCallback((metaData: any) => { - const { max_height, max_width } = metaData || {}; + const paramsConfig = useMemo(() => { + const { max_height, max_width } = modelMeta; if (!max_height || !max_width) { return ImageParamsConfig; } @@ -175,11 +175,6 @@ const GroundImages: React.FC = forwardRef((props, ref) => { value: `${max_width}x${max_height}` }); } - return newImageSizeOptions; - }, []); - - const paramsConfig = useMemo(() => { - const newImageSizeOptions = getNewImageSizeOptions(modelMeta); let result = ImageParamsConfig.map((item) => { if (item.name === 'size') { return { @@ -193,7 +188,7 @@ const GroundImages: React.FC = forwardRef((props, ref) => { result = result.filter((item) => item.name !== 'size'); } return result; - }, [modelMeta, getNewImageSizeOptions]); + }, [modelMeta]); const setImageSize = useCallback(() => { let size: Record = {