chore: image custom params

src/pages/llmodels/components/advance-config.tsx

@@ -35,7 +35,7 @@ interface AdvanceConfigProps {
   gpuOptions: Array<any>;
   action: PageActionType;
   source: string;
-  modelTask: string;
+  modelTask: Record<string, any>;
 }
 
 const AdvanceConfig: React.FC<AdvanceConfigProps> = (props) => {

src/pages/llmodels/components/data-form.tsx

@@ -23,6 +23,7 @@ import {
   modelTaskMap,
   ollamaModelOptions
 } from '../config';
+import { HuggingFaceModels, ModelScopeModels } from '../config/audio-catalog';
 import { FormData, GPUListItem } from '../config/types';
 import AdvanceConfig from './advance-config';
 
@@ -120,11 +121,30 @@ const DataForm: React.FC<DataFormProps> = forwardRef((props, ref) => {
     []
   );
 
+  const identifyModelTask = () => {
+    let data = null;
+    if (props.source === modelSourceMap.huggingface_value) {
+      data = HuggingFaceModels.find(
+        (item) => `${item.org}/${item.name}` === props.selectedModel.name
+      );
+    }
+    if (props.source === modelSourceMap.modelscope_value) {
+      data = ModelScopeModels.find(
+        (item) => `${item.org}/${item.name}` === props.selectedModel.name
+      );
+    }
+    if (data) {
+      return modelTaskMap.audio;
+    }
+    return '';
+  };
   const handleOnSelectModel = () => {
     let name = _.split(props.selectedModel.name, '/').slice(-1)[0];
     const reg = /(-gguf)$/i;
     name = _.toLower(name).replace(reg, '');
 
+    const modelTaskType = identifyModelTask();
+
     const modelTask =
       HuggingFaceTaskMap.audio.includes(props.selectedModel.task) ||
       ModelscopeTaskMap.audio.includes(props.selectedModel.task)
@@ -133,7 +153,7 @@ const DataForm: React.FC<DataFormProps> = forwardRef((props, ref) => {
 
     setModelTask({
       value: props.selectedModel.task,
-      type: modelTask,
+      type: modelTaskType || modelTask,
       text2speech:
         HuggingFaceTaskMap[modelTaskMap.textToSpeech] ===
           props.selectedModel.task ||
@@ -351,8 +371,6 @@ const DataForm: React.FC<DataFormProps> = forwardRef((props, ref) => {
     if (gpu) {
       onOk({
         ..._.omit(formdata, ['scheduleType']),
-        speech_to_text: modelTask.speech2text,
-        text_to_speech: modelTask.text2speech,
         gpu_selector: {
           gpu_name: gpu.name,
           gpu_index: gpu.index,
@@ -361,9 +379,7 @@ const DataForm: React.FC<DataFormProps> = forwardRef((props, ref) => {
       });
     } else {
       onOk({
-        ..._.omit(formdata, ['scheduleType']),
-        speech_to_text: modelTask.speech2text,
-        text_to_speech: modelTask.text2speech
+        ..._.omit(formdata, ['scheduleType'])
       });
     }
   };

src/pages/llmodels/components/model-card.tsx

@@ -130,12 +130,6 @@ const ModelCard: React.FC<{
   // huggingface model card data
   const getHuggingfaceModelDetail = async () => {
     try {
-      const configjson = await loadConfig(
-        props.selectedModel.name,
-        'main'
-      ).catch(() => {
-        return null;
-      });
       const [modelcard, readme] = await Promise.all([
         queryHuggingfaceModelDetail(
           { repo: props.selectedModel.name },
@@ -183,11 +177,6 @@ const ModelCard: React.FC<{
 
   const getModelScopeModelDetail = async () => {
     try {
-      const configjson = await loadModelscopeModelConfig(
-        props.selectedModel.name
-      ).catch(() => {
-        return null;
-      });
       const data = await queryModelScopeModelDetail(
         {
           name: props.selectedModel.name
@@ -200,7 +189,6 @@ const ModelCard: React.FC<{
         ...data?.Data,
         name: `${data.Data?.Path}/${data.Data?.Name}`
       });
-      console.log('modelData++++++++++++', configjson, data?.Data);
       setReadmeText(data?.Data?.ReadMeContent);
       const isGGUF = some(
         data?.Data?.Tags,

src/pages/llmodels/components/search-model.tsx

@@ -10,7 +10,6 @@ import {
   ModelSortType,
   ModelscopeTaskMap,
   modelSourceMap,
-  modelTaskMap,
   ollamaModelOptions
 } from '../config';
 import SearchStyle from '../style/search-result.less';
@@ -233,22 +232,15 @@ const SearchModel: React.FC<SearchInputProps> = (props) => {
           modelSource={modelSource}
         ></SearchInput>
         <div className={SearchStyle.filter}>
-          {/* <span>
+          <span>
             <span className="value">
               {intl.formatMessage(
                 { id: 'models.search.result' },
                 { count: dataSource.repoOptions.length }
               )}
             </span>
-          </span> */}
-          <span
-            style={{
-              flex: 1,
-              display: 'flex',
-              justifyContent: 'space-between',
-              alignItems: 'center'
-            }}
-          >
+          </span>
+          <span>
             <Checkbox
               onChange={handleFilterGGUFChange}
               className="m-r-5"
@@ -271,43 +263,20 @@ const SearchModel: React.FC<SearchInputProps> = (props) => {
                 <InfoCircleOutlined className="m-l-4" />
               </Tooltip>
             </Checkbox>
-            <span className="flex gap-6">
-              <Select
-                allowClear
-                value={filterTaskRef.current}
-                onChange={handleFilterTaskChange}
-                options={[
-                  {
-                    label: intl.formatMessage({
-                      id: 'playground.audio.texttospeech'
-                    }),
-                    value: modelTaskMap.textToSpeech
-                  },
-                  {
-                    label: intl.formatMessage({
-                      id: 'playground.audio.speechtotext'
-                    }),
-                    value: modelTaskMap.speechToText
-                  }
-                ]}
-                size="middle"
-                style={{ width: '140px' }}
-              ></Select>
-              <Select
-                value={dataSource.sortType}
-                onChange={handleSortChange}
-                labelRender={({ label }) => {
-                  return (
-                    <span>
-                      {intl.formatMessage({ id: 'model.deploy.sort' })}: {label}
-                    </span>
-                  );
-                }}
-                options={modelFilesSortOptions.current}
-                size="middle"
-                style={{ width: '140px' }}
-              ></Select>
-            </span>
+            <Select
+              value={dataSource.sortType}
+              onChange={handleSortChange}
+              labelRender={({ label }) => {
+                return (
+                  <span>
+                    {intl.formatMessage({ id: 'model.deploy.sort' })}: {label}
+                  </span>
+                );
+              }}
+              options={modelFilesSortOptions.current}
+              size="middle"
+              style={{ width: '150px' }}
+            ></Select>
           </span>
         </div>
       </>