fix: image size options

main
jialin 1 year ago
parent 73e41b0de0
commit 9974fa831c

@ -135,8 +135,8 @@ const GroundImages: React.FC<MessageProps> = 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<MessageProps> = 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<MessageProps> = 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);

@ -155,8 +155,8 @@ const GroundImages: React.FC<MessageProps> = 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<MessageProps> = 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<MessageProps> = forwardRef((props, ref) => {
result = result.filter((item) => item.name !== 'size');
}
return result;
}, [modelMeta, getNewImageSizeOptions]);
}, [modelMeta]);
const setImageSize = useCallback(() => {
let size: Record<string, string | number> = {

Loading…
Cancel
Save