chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: type

Browse Source
This commit is contained in:
BennyKok 2024-01-14 00:15:26 +08:00
parent dc4c63e654
commit 6c49bff095
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web/src/components/custom-form/ModelPickerView.tsx
View File

@ -173,6 +173,14 @@ export function ModelPickerView({
); );
} }
function mapType(type: string) {
switch (type) {
case "checkpoint":
return "checkpoints";
}
return type;
}
function mapModelsList( function mapModelsList(
models: z.infer<typeof CivitalModelSchema> models: z.infer<typeof CivitalModelSchema>
): z.infer<typeof ModelListWrapper> { ): z.infer<typeof ModelListWrapper> {
@ -181,7 +189,7 @@ function mapModelsList(
const v = item.modelVersions[0]; const v = item.modelVersions[0];
return { return {
name: `${item.name} ${v.name} (${v.files[0].name})`, name: `${item.name} ${v.name} (${v.files[0].name})`,
type: item.type.toLowerCase(), type: mapType(item.type.toLowerCase()),
base: v.baseModel, base: v.baseModel,
save_path: "default", save_path: "default",
description: item.description, description: item.description,