root/gpustack-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: rerank clear failed

Browse Source
This commit is contained in:
jialin
2024-11-27 22:47:53 +08:00
parent 324bd57e3d
commit af789ee886
9 changed files with 223 additions and 160 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/pages/llmodels/components/table-list.tsx
+76 -73
View File
@@ -423,80 +423,83 @@ const Models: React.FC<ModelsProps> = ({
[]
);
const renderModelTags = useCallback((record: ListItem) => {
if (record.reranker) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
Reranker
</Tag>
);
}
const renderModelTags = useCallback(
(record: ListItem) => {
if (record.reranker) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
Reranker
</Tag>
);
}
if (record.embedding_only && !record.reranker) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
Embedding Only
</Tag>
);
}
if (record.text_to_speech) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
{intl.formatMessage({ id: 'playground.audio.texttospeech' })}
</Tag>
);
}
if (record.speech_to_text) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
{intl.formatMessage({ id: 'playground.audio.speechtotext' })}
</Tag>
);
}
if (record.image_only) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
Image Only
</Tag>
);
}
return null;
}, []);
if (record.embedding_only && !record.reranker) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
Embedding Only
</Tag>
);
}
if (record.text_to_speech) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
{intl.formatMessage({ id: 'playground.audio.texttospeech' })}
</Tag>
);
}
if (record.speech_to_text) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
{intl.formatMessage({ id: 'playground.audio.speechtotext' })}
</Tag>
);
}
if (record.image_only) {
return (
<Tag
style={{
margin: 0,
opacity: 0.8,
transform: 'scale(0.9)'
}}
color="geekblue"
>
Image Only
</Tag>
);
}
return null;
},
[intl]
);
const renderChildren = useCallback(
(list: any, parent?: any) => {
return (
src/pages/llmodels/config/llama-config.ts
+49
View File
@@ -27,6 +27,55 @@ const options = [
{
label: '--images',
value: '--images'
},
{
label: '--image-max-batch',
value: '--image-max-batch'
},
{
label: '--image-max-height',
value: '--image-max-height'
},
{
label: '--image-max-width',
value: '--image-max-width'
},
{
label: '--image-guidance',
value: '--image-guidance'
},
{
label: '--image-strength',
value: '--image-strength'
},
{
label: '--image-sampler',
value: '--image-sampler',
options: [
'euler_a',
'euler',
'heun',
'dpm2',
'dpm++2s_a',
'dpm++2m',
'dpm++2mv2',
'ipndm',
'ipndm_v',
'lcm'
]
},
{
label: '--image-sampler-steps',
value: '--image-sample-steps'
},
{
label: '--image-cfg-scale',
value: '--image-cfg-scale'
},
{
label: '--image-schedule',
value: '--image-schedule',
options: ['default', 'discrete', 'karras', 'exponential', 'ays', 'gits']
}
];