diff --git a/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx b/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx
index 9925a48381..ddc6f4c6c3 100644
--- a/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx
+++ b/invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx
@@ -150,6 +150,7 @@ const InputFieldComponent = (props: InputFieldComponentProps) => {
         nodeId={nodeId}
         field={field}
         template={template}
+        base_models={['sd-1', 'sd-2']}
       />
     );
   }
diff --git a/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx b/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx
index 3f03c76d50..c221833cfb 100644
--- a/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx
+++ b/invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx
@@ -17,14 +17,14 @@ import { FieldComponentProps } from './types';
 const ModelInputFieldComponent = (
   props: FieldComponentProps<ModelInputFieldValue, ModelInputFieldTemplate>
 ) => {
-  const { nodeId, field } = props;
+  const { nodeId, field, base_models } = props;
 
   const dispatch = useAppDispatch();
   const { t } = useTranslation();
 
   const { data: mainModels } = useGetMainModelsQuery({
     model_type: 'main',
-    base_models: ['sd-1', 'sd-2', 'sdxl', 'sdxl-refiner'],
+    base_models: base_models,
   });
 
   const data = useMemo(() => {