From e32eb2a649974b520774154b75de2377d4ba3b59 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Fri, 29 Dec 2023 08:59:24 +1100
Subject: [PATCH] fix(ui): restore labels in model manager selects

---
 .../modelManager/subpanels/shared/BaseModelSelect.tsx    | 9 ++++++++-
 .../modelManager/subpanels/shared/ModelVariantSelect.tsx | 9 ++++++++-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/invokeai/frontend/web/src/features/modelManager/subpanels/shared/BaseModelSelect.tsx b/invokeai/frontend/web/src/features/modelManager/subpanels/shared/BaseModelSelect.tsx
index 4da25d137d..0fc9f26d7c 100644
--- a/invokeai/frontend/web/src/features/modelManager/subpanels/shared/BaseModelSelect.tsx
+++ b/invokeai/frontend/web/src/features/modelManager/subpanels/shared/BaseModelSelect.tsx
@@ -1,9 +1,11 @@
+import { InvControl } from 'common/components/InvControl/InvControl';
 import { InvSelect } from 'common/components/InvSelect/InvSelect';
 import type {
   InvSelectOption,
   InvSelectProps,
 } from 'common/components/InvSelect/types';
 import { MODEL_TYPE_MAP } from 'features/parameters/types/constants';
+import { useTranslation } from 'react-i18next';
 
 const options: InvSelectOption[] = [
   { value: 'sd-1', label: MODEL_TYPE_MAP['sd-1'] },
@@ -15,5 +17,10 @@ const options: InvSelectOption[] = [
 type BaseModelSelectProps = Omit<InvSelectProps, 'options'>;
 
 export default function BaseModelSelect(props: BaseModelSelectProps) {
-  return <InvSelect options={options} {...props} />;
+  const { t } = useTranslation();
+  return (
+    <InvControl label={t('modelManager.baseModel')}>
+      <InvSelect options={options} {...props} />
+    </InvControl>
+  );
 }
diff --git a/invokeai/frontend/web/src/features/modelManager/subpanels/shared/ModelVariantSelect.tsx b/invokeai/frontend/web/src/features/modelManager/subpanels/shared/ModelVariantSelect.tsx
index 700edee040..543603c0ea 100644
--- a/invokeai/frontend/web/src/features/modelManager/subpanels/shared/ModelVariantSelect.tsx
+++ b/invokeai/frontend/web/src/features/modelManager/subpanels/shared/ModelVariantSelect.tsx
@@ -1,8 +1,10 @@
+import { InvControl } from 'common/components/InvControl/InvControl';
 import { InvSelect } from 'common/components/InvSelect/InvSelect';
 import type {
   InvSelectOption,
   InvSelectProps,
 } from 'common/components/InvSelect/types';
+import { useTranslation } from 'react-i18next';
 
 const options: InvSelectOption[] = [
   { value: 'normal', label: 'Normal' },
@@ -13,5 +15,10 @@ const options: InvSelectOption[] = [
 type VariantSelectProps = Omit<InvSelectProps, 'options'>;
 
 export default function ModelVariantSelect(props: VariantSelectProps) {
-  return <InvSelect options={options} {...props} />;
+  const { t } = useTranslation();
+  return (
+    <InvControl label={t('modelManager.variant')}>
+      <InvSelect options={options} {...props} />
+    </InvControl>
+  );
 }