diff --git a/src/renderer/components/ModelZoo/LocalModelsTable.tsx b/src/renderer/components/ModelZoo/LocalModelsTable.tsx index d678ce7d5105469af17c75444b8ddcf01b2cf081..0a8404c4ca81350ace0d64a1123ed88a411dcc2a 100644 --- a/src/renderer/components/ModelZoo/LocalModelsTable.tsx +++ b/src/renderer/components/ModelZoo/LocalModelsTable.tsx @@ -11,6 +11,7 @@ import { Option, } from '@mui/joy'; import { + ArrowRightToLineIcon, ArrowDownIcon, FlaskRoundIcon, InfoIcon, @@ -184,6 +185,14 @@ const LocalModelsTable = ({ marginRight: '5px', }} /> + ) : (row?.source && row?.source != "transformerlab") ? ( + <ArrowRightToLineIcon + color="var(--joy-palette-success-700)" + style={{ + verticalAlign: 'middle', + marginRight: '5px', + }} + /> ) : ( '' )}{' '} @@ -191,38 +200,38 @@ const LocalModelsTable = ({ </Typography> </td> <td> - <Typography style={{overflow: 'hidden'}}> - {' '} - {row?.json_data?.architecture == 'MLX' && ( - <> - <TinyMLXLogo /> - - </> - )} - {row?.json_data?.architecture == 'GGUF' && ( - <> - <img - src="https://avatars.githubusercontent.com/ggerganov" - width="24" - valign="middle" - style={{ borderRadius: '50%' }} - />{' '} - - </> - )} - {[ - 'FalconForCausalLM', - 'Gemma2ForCausalLM', - 'GPTBigCodeForCausalLM', - 'LlamaForCausalLM', - 'MistralForCausalLM', - 'Phi3ForCausalLM', - 'Qwen2ForCausalLM', - 'T5ForConditionalGeneration' - ].includes(row?.json_data?.architecture) && ( - <>🤗 </> - )} - {row?.json_data?.architecture} + <Typography style={{ overflow: 'hidden' }}> + {' '} + {row?.json_data?.architecture == 'MLX' && ( + <> + <TinyMLXLogo /> + + </> + )} + {row?.json_data?.architecture == 'GGUF' && ( + <> + <img + src="https://avatars.githubusercontent.com/ggerganov" + width="24" + valign="middle" + style={{ borderRadius: '50%' }} + />{' '} + + </> + )} + {[ + 'FalconForCausalLM', + 'Gemma2ForCausalLM', + 'GPTBigCodeForCausalLM', + 'LlamaForCausalLM', + 'MistralForCausalLM', + 'Phi3ForCausalLM', + 'Qwen2ForCausalLM', + 'T5ForConditionalGeneration' + ].includes(row?.json_data?.architecture) && ( + <>🤗 </> + )} + {row?.json_data?.architecture} </Typography> </td> <td>{row?.json_data?.parameters}</td> @@ -257,8 +266,8 @@ const LocalModelsTable = ({ if ( confirm( "Are you sure you want to delete model '" + - row.model_id + - "'?" + row.model_id + + "'?" ) ) { await fetch(