fix: lora tag
This commit is contained in:
@@ -98,7 +98,7 @@ const RouteItem: React.FC<TargetItemProps> = ({
|
|||||||
}}
|
}}
|
||||||
>
|
>
|
||||||
<AutoTooltip ghost>{data.name}</AutoTooltip>
|
<AutoTooltip ghost>{data.name}</AutoTooltip>
|
||||||
{!!data.lora_module_name && (
|
{!!data.overridden_model_name && !!data.model_id && (
|
||||||
<FilesTag color="purple" variant="outlined">
|
<FilesTag color="purple" variant="outlined">
|
||||||
<span style={{ opacity: 1 }}>LoRA</span>
|
<span style={{ opacity: 1 }}>LoRA</span>
|
||||||
</FilesTag>
|
</FilesTag>
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ export interface RouteTargetFormItem {
|
|||||||
weight?: number | null;
|
weight?: number | null;
|
||||||
model_id?: number;
|
model_id?: number;
|
||||||
provider_id?: number;
|
provider_id?: number;
|
||||||
lora_module_name?: string;
|
overridden_model_name?: string;
|
||||||
fallback_status_codes?: string[];
|
fallback_status_codes?: string[];
|
||||||
parentId?: string | number;
|
parentId?: string | number;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ const isSameTarget = (
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
if (left.model_id && left.model_id === right.model_id) {
|
if (left.model_id && left.model_id === right.model_id) {
|
||||||
return left.lora_module_name === right.lora_module_name;
|
return left.overridden_model_name === right.overridden_model_name;
|
||||||
}
|
}
|
||||||
if (left.provider_id && left.provider_id === right.provider_id) {
|
if (left.provider_id && left.provider_id === right.provider_id) {
|
||||||
return left.overridden_model_name === right.overridden_model_name;
|
return left.overridden_model_name === right.overridden_model_name;
|
||||||
@@ -48,10 +48,11 @@ const normalizeTarget = (
|
|||||||
model_id: target?.model_id,
|
model_id: target?.model_id,
|
||||||
provider_id: target?.provider_id,
|
provider_id: target?.provider_id,
|
||||||
overridden_model_name: target?.overridden_model_name,
|
overridden_model_name: target?.overridden_model_name,
|
||||||
lora_module_name: target?.lora_module_name,
|
|
||||||
fallback_status_codes: target?.fallback_status_codes
|
fallback_status_codes: target?.fallback_status_codes
|
||||||
};
|
};
|
||||||
|
|
||||||
|
console.log('normalizeTarget', target, normalizedTarget);
|
||||||
|
|
||||||
return _.omitBy(normalizedTarget, _.isUndefined) as RouteTargetFormItem;
|
return _.omitBy(normalizedTarget, _.isUndefined) as RouteTargetFormItem;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -174,8 +175,8 @@ const AccessForm: React.FC<ProviderFormProps> = forwardRef((props, ref) => {
|
|||||||
value: ep.model_id
|
value: ep.model_id
|
||||||
? [
|
? [
|
||||||
'deployments',
|
'deployments',
|
||||||
ep.lora_module_name
|
ep.overridden_model_name
|
||||||
? `${ep.model_id}_lora_${ep.lora_module_name}`
|
? `${ep.model_id}_lora_${ep.overridden_model_name}`
|
||||||
: ep.model_id
|
: ep.model_id
|
||||||
]
|
]
|
||||||
: [ep.provider_id, ep.overridden_model_name]
|
: [ep.provider_id, ep.overridden_model_name]
|
||||||
@@ -202,8 +203,8 @@ const AccessForm: React.FC<ProviderFormProps> = forwardRef((props, ref) => {
|
|||||||
value: fallbackTarget.model_id
|
value: fallbackTarget.model_id
|
||||||
? [
|
? [
|
||||||
'deployments',
|
'deployments',
|
||||||
fallbackTarget.lora_module_name
|
fallbackTarget.overridden_model_name
|
||||||
? `${fallbackTarget.model_id}_lora_${fallbackTarget.lora_module_name}`
|
? `${fallbackTarget.model_id}_lora_${fallbackTarget.overridden_model_name}`
|
||||||
: fallbackTarget.model_id
|
: fallbackTarget.model_id
|
||||||
]
|
]
|
||||||
: [fallbackTarget.provider_id, fallbackTarget.overridden_model_name]
|
: [fallbackTarget.provider_id, fallbackTarget.overridden_model_name]
|
||||||
|
|||||||
@@ -71,7 +71,7 @@ const useTargetSourceModels = () => {
|
|||||||
data: {
|
data: {
|
||||||
model_id: model.id,
|
model_id: model.id,
|
||||||
parentId: 'deployments',
|
parentId: 'deployments',
|
||||||
lora_module_name: loraName
|
overridden_model_name: loraName
|
||||||
},
|
},
|
||||||
source: 'deployment',
|
source: 'deployment',
|
||||||
isLora: true
|
isLora: true
|
||||||
|
|||||||
Reference in New Issue
Block a user