From d6c9360fa3ea76249da25aeed051ce6488412edf Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Tue, 20 Feb 2024 10:45:05 +1100
Subject: [PATCH] fix(ui): get embedding select working

---
 .../frontend/web/src/features/embedding/EmbeddingSelect.tsx   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/frontend/web/src/features/embedding/EmbeddingSelect.tsx b/invokeai/frontend/web/src/features/embedding/EmbeddingSelect.tsx
index 426ddd21e2..fd05edc466 100644
--- a/invokeai/frontend/web/src/features/embedding/EmbeddingSelect.tsx
+++ b/invokeai/frontend/web/src/features/embedding/EmbeddingSelect.tsx
@@ -18,7 +18,7 @@ export const EmbeddingSelect = memo(({ onSelect, onClose }: EmbeddingSelectProps
 
   const getIsDisabled = useCallback(
     (embedding: TextualInversionConfig): boolean => {
-      const isCompatible = currentBaseModel === embedding.base_model;
+      const isCompatible = currentBaseModel === embedding.base;
       const hasMainModel = Boolean(currentBaseModel);
       return !hasMainModel || !isCompatible;
     },
@@ -31,7 +31,7 @@ export const EmbeddingSelect = memo(({ onSelect, onClose }: EmbeddingSelectProps
       if (!embedding) {
         return;
       }
-      onSelect(embedding.model_name);
+      onSelect(embedding.key);
     },
     [onSelect]
   );