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