diff --git a/packages/fakeyou.com/src/v2/view/PageContainer.tsx b/packages/fakeyou.com/src/v2/view/PageContainer.tsx index f7631fd27a..e940538c04 100644 --- a/packages/fakeyou.com/src/v2/view/PageContainer.tsx +++ b/packages/fakeyou.com/src/v2/view/PageContainer.tsx @@ -373,6 +373,7 @@ class PageContainer extends React.Component< render={props => ( ; ttsInferenceJobs: Array; @@ -49,6 +52,7 @@ interface WeightProps { } export default function WeightPage({ + sessionWrapper, sessionSubscriptionsWrapper, inferenceJobs, ttsInferenceJobs, @@ -100,6 +104,15 @@ export default function WeightPage({ const deleteWeight = () => remove(!!user?.can_ban_users); + switch (weight?.weight_type) { + case WeightType.SD_15: + case WeightType.SDXL: + case WeightType.LORA: + if (!sessionWrapper.canAccessStudio()) { + return + } + } + function renderWeightComponent(weight: Weight) { switch (weight.weight_category) { case WeightCategory.TTS: