Skip to content
Snippets Groups Projects
Unverified Commit c794fe30 authored by ali asaria's avatar ali asaria Committed by GitHub
Browse files

Merge pull request #281 from transformerlab/add/imported-model-icon

Add imported model icon
parents 0aec5542 a235bbed
No related branches found
No related tags found
No related merge requests found
......@@ -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 />
&nbsp;
</>
)}
{row?.json_data?.architecture == 'GGUF' && (
<>
<img
src="https://avatars.githubusercontent.com/ggerganov"
width="24"
valign="middle"
style={{ borderRadius: '50%' }}
/>{' '}
&nbsp;
</>
)}
{[
'FalconForCausalLM',
'Gemma2ForCausalLM',
'GPTBigCodeForCausalLM',
'LlamaForCausalLM',
'MistralForCausalLM',
'Phi3ForCausalLM',
'Qwen2ForCausalLM',
'T5ForConditionalGeneration'
].includes(row?.json_data?.architecture) && (
<>🤗 &nbsp;</>
)}
{row?.json_data?.architecture}
<Typography style={{ overflow: 'hidden' }}>
{' '}
{row?.json_data?.architecture == 'MLX' && (
<>
<TinyMLXLogo />
&nbsp;
</>
)}
{row?.json_data?.architecture == 'GGUF' && (
<>
<img
src="https://avatars.githubusercontent.com/ggerganov"
width="24"
valign="middle"
style={{ borderRadius: '50%' }}
/>{' '}
&nbsp;
</>
)}
{[
'FalconForCausalLM',
'Gemma2ForCausalLM',
'GPTBigCodeForCausalLM',
'LlamaForCausalLM',
'MistralForCausalLM',
'Phi3ForCausalLM',
'Qwen2ForCausalLM',
'T5ForConditionalGeneration'
].includes(row?.json_data?.architecture) && (
<>🤗 &nbsp;</>
)}
{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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment