diff --git a/packages/components/src/api/_common/enums/WeightCategory.ts b/packages/components/src/api/_common/enums/WeightCategory.ts
index e0c256eb67..834ed7b43f 100644
--- a/packages/components/src/api/_common/enums/WeightCategory.ts
+++ b/packages/components/src/api/_common/enums/WeightCategory.ts
@@ -4,4 +4,5 @@ export enum WeightCategory {
VC = "voice_conversion",
ZS = "zero_shot",
SD = "image_generation",
+ WF = "workflow_config",
}
diff --git a/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx b/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx
index e8879b68db..ee8ee9d52b 100644
--- a/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx
+++ b/packages/fakeyou.com/src/components/common/Card/WeightsCards.tsx
@@ -9,12 +9,14 @@ interface Props {
}
export default function WeightsCards({ props, type }: Props) {
+ console.log(props);
switch (type) {
case WeightCategory.TTS:
case WeightCategory.VC:
case WeightCategory.ZS:
return ;
case WeightCategory.SD:
+ case WeightCategory.WF:
return ;
default:
return
Unsupported weight type
;
diff --git a/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx b/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
index 11e051d2c3..667c9a7a63 100644
--- a/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/weight/WeightPage.tsx
@@ -73,13 +73,7 @@ export default function WeightPage({
const history = useHistory();
const bookmarks = useBookmarks();
const ratings = useRatings();
- const {
- data: weight,
- fetchError,
- isLoading,
- title,
- remove,
- } = useWeightFetch({
+ const fetchedWeight = useWeightFetch({
onRemove: () => {
history.push(source || "");
},
@@ -89,7 +83,14 @@ export default function WeightPage({
},
token: weight_token,
});
-
+ const {
+ data: weight,
+ fetchError,
+ isLoading,
+ title,
+ remove,
+ } = fetchedWeight;
+ console.log(fetchedWeight);
const timeUpdated = moment(weight?.updated_at || "").fromNow();
const dateUpdated = moment(weight?.updated_at || "").format("LLL");
const dateCreated = moment(weight?.created_at || "").format("LLL");
@@ -287,6 +288,7 @@ export default function WeightPage({
[WeightCategory.SD]: { weightCategory: "Image Generation" },
[WeightCategory.ZS]: { weightCategory: "Voice Designer" },
[WeightCategory.VOCODER]: { weightCategory: "Vocoder" },
+ [WeightCategory.WF]: { weightCategory: "Workflow Config" },
};
let { weightCategory } = weightCategoryMap[weight.weight_category] || {
@@ -321,14 +323,11 @@ export default function WeightPage({
switch (weight.weight_category) {
case WeightCategory.TTS:
- weightDetails = ;
- break;
case WeightCategory.VC:
- weightDetails = ;
- break;
case WeightCategory.ZS:
weightDetails = ;
break;
+ case WeightCategory.WF:
case WeightCategory.SD:
weightDetails = ;
break;