Spaces:
Sleeping
Sleeping
machineuser
commited on
Commit
•
7787a53
1
Parent(s):
2b1e8c5
Sync widgets demo
Browse files
packages/tasks/src/library-ui-elements.ts
CHANGED
@@ -72,7 +72,7 @@ model = BaseModel.from_pretrained("${model.id}")`,
|
|
72 |
];
|
73 |
|
74 |
function get_base_diffusers_model(model: ModelData): string {
|
75 |
-
return model.cardData?.base_model ?? "fill-in-base-model";
|
76 |
}
|
77 |
|
78 |
const bertopic = (model: ModelData) => [
|
|
|
72 |
];
|
73 |
|
74 |
function get_base_diffusers_model(model: ModelData): string {
|
75 |
+
return model.cardData?.base_model?.toString() ?? "fill-in-base-model";
|
76 |
}
|
77 |
|
78 |
const bertopic = (model: ModelData) => [
|
packages/tasks/src/model-data.ts
CHANGED
@@ -93,7 +93,7 @@ export interface ModelData {
|
|
93 |
| {
|
94 |
parameters?: Record<string, unknown>;
|
95 |
};
|
96 |
-
base_model?: string;
|
97 |
};
|
98 |
/**
|
99 |
* Library name
|
|
|
93 |
| {
|
94 |
parameters?: Record<string, unknown>;
|
95 |
};
|
96 |
+
base_model?: string | string[];
|
97 |
};
|
98 |
/**
|
99 |
* Library name
|