fix: image size options

This commit is contained in:
jialin
2025-01-13 16:34:08 +08:00
parent 73e41b0de0
commit 9974fa831c
2 changed files with 7 additions and 17 deletions
@@ -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> = {