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