fix(ui): fix workflow editor model selector, excise ONNX

Ensure workflow editor model selector component gets a value

This introduced some funky type issues related to ONNX models. ONNX doesn't work anyways (unmaintained). Instead of fixing the types to work with a non-working feature, ONNX is now removed entirely from the UI.

- Remove all refs to ONNX (and Olives)
- Fix some type issues
- Add ONNX nodes to the nodes denylist (so they are not visible in UI)
- Update VAE graph helper, which still had some ONNX logic. It's a very simple change and doesn't change any logic. Just removes some conditions that were for ONNX. I tested it and nothing broke.
- Regenerate types
- Fix prettier and eslint ignores for generated types
- Lint

Author: psychedelicious

invokeai/frontend/web/.eslintignore

@@ -7,4 +7,4 @@ stats.html
 index.html
 .yarn/
 *.scss
-src/services/api/schema.d.ts
+src/services/api/schema.ts

invokeai/frontend/web/.prettierignore

@@ -9,7 +9,7 @@ index.html
 .yarn/
 .yalc/
 *.scss
-src/services/api/schema.d.ts
+src/services/api/schema.ts
 static/
 src/theme/css/overlayscrollbars.css
 src/theme_/css/overlayscrollbars.css

invokeai/frontend/web/src/common/components/IAIColorPicker.tsx

@@ -70,7 +70,7 @@ const IAIColorPicker = (props: IAIColorPickerProps) => {
               step={1}
               w={numberInputWidth}
               defaultValue={90}
-              />
+            />
           </InvControl>
           <InvControl label="Green">
             <InvNumberInput
@@ -81,7 +81,7 @@ const IAIColorPicker = (props: IAIColorPickerProps) => {
               step={1}
               w={numberInputWidth}
               defaultValue={90}
-              />
+            />
           </InvControl>
           <InvControl label="Blue">
             <InvNumberInput
@@ -92,7 +92,7 @@ const IAIColorPicker = (props: IAIColorPickerProps) => {
               step={1}
               w={numberInputWidth}
               defaultValue={255}
-              />
+            />
           </InvControl>
           <InvControl label="Alpha">
             <InvNumberInput

invokeai/frontend/web/src/features/dynamicPrompts/components/ParamDynamicPromptsMaxPrompts.tsx

@@ -3,9 +3,7 @@ import { stateSelector } from 'app/store/store';
 import { useAppDispatch, useAppSelector } from 'app/store/storeHooks';
 import { InvControl } from 'common/components/InvControl/InvControl';
 import { InvSlider } from 'common/components/InvSlider/InvSlider';
-import {
-  maxPromptsChanged,
-} from 'features/dynamicPrompts/store/dynamicPromptsSlice';
+import { maxPromptsChanged } from 'features/dynamicPrompts/store/dynamicPromptsSlice';
 import { memo, useCallback } from 'react';
 import { useTranslation } from 'react-i18next';
 

invokeai/frontend/web/src/features/lora/components/LoRACard.tsx

@@ -9,10 +9,7 @@ import { InvNumberInput } from 'common/components/InvNumberInput/InvNumberInput'
 import { InvSlider } from 'common/components/InvSlider/InvSlider';
 import { InvText } from 'common/components/InvText/wrapper';
 import type { LoRA } from 'features/lora/store/loraSlice';
-import {
-  loraRemoved,
-  loraWeightChanged,
-} from 'features/lora/store/loraSlice';
+import { loraRemoved, loraWeightChanged } from 'features/lora/store/loraSlice';
 import { memo, useCallback } from 'react';
 import { FaTrashCan } from 'react-icons/fa6';
 

invokeai/frontend/web/src/features/modelManager/subpanels/ModelManagerPanel/ModelList.tsx

@@ -13,12 +13,10 @@ import { ALL_BASE_MODELS } from 'services/api/constants';
 import type {
   LoRAModelConfigEntity,
   MainModelConfigEntity,
-  OnnxModelConfigEntity,
 } from 'services/api/endpoints/models';
 import {
   useGetLoRAModelsQuery,
   useGetMainModelsQuery,
-  useGetOnnxModelsQuery,
 } from 'services/api/endpoints/models';
 
 import ModelListItem from './ModelListItem';
@@ -28,9 +26,9 @@ type ModelListProps = {
   setSelectedModelId: (name: string | undefined) => void;
 };
 
-type ModelFormat = 'all' | 'checkpoint' | 'diffusers' | 'olive' | 'onnx';
+type ModelFormat = 'all' | 'checkpoint' | 'diffusers';
 
-type ModelType = 'main' | 'lora' | 'onnx';
+type ModelType = 'main' | 'lora';
 
 type CombinedModelFormat = ModelFormat | 'lora';
 
@@ -77,26 +75,6 @@ const ModelList = (props: ModelListProps) => {
     }
   );
 
-  const { filteredOnnxModels, isLoadingOnnxModels } = useGetOnnxModelsQuery(
-    ALL_BASE_MODELS,
-    {
-      selectFromResult: ({ data, isLoading }) => ({
-        filteredOnnxModels: modelsFilter(data, 'onnx', 'onnx', nameFilter),
-        isLoadingOnnxModels: isLoading,
-      }),
-    }
-  );
-
-  const { filteredOliveModels, isLoadingOliveModels } = useGetOnnxModelsQuery(
-    ALL_BASE_MODELS,
-    {
-      selectFromResult: ({ data, isLoading }) => ({
-        filteredOliveModels: modelsFilter(data, 'onnx', 'olive', nameFilter),
-        isLoadingOliveModels: isLoading,
-      }),
-    }
-  );
-
   const handleSearchFilter = useCallback((e: ChangeEvent<HTMLInputElement>) => {
     setNameFilter(e.target.value);
   }, []);
@@ -126,20 +104,6 @@ const ModelList = (props: ModelListProps) => {
           >
             {t('modelManager.checkpointModels')}
           </InvButton>
-          <InvButton
-            size="sm"
-            onClick={setModelFormatFilter.bind(null, 'onnx')}
-            isChecked={modelFormatFilter === 'onnx'}
-          >
-            {t('modelManager.onnxModels')}
-          </InvButton>
-          <InvButton
-            size="sm"
-            onClick={setModelFormatFilter.bind(null, 'olive')}
-            isChecked={modelFormatFilter === 'olive'}
-          >
-            {t('modelManager.oliveModels')}
-          </InvButton>
           <InvButton
             size="sm"
             onClick={setModelFormatFilter.bind(null, 'lora')}
@@ -202,34 +166,6 @@ const ModelList = (props: ModelListProps) => {
                 key="loras"
               />
             )}
-          {/* Olive List */}
-          {isLoadingOliveModels && (
-            <FetchingModelsLoader loadingMessage="Loading Olives..." />
-          )}
-          {['all', 'olive'].includes(modelFormatFilter) &&
-            !isLoadingOliveModels &&
-            filteredOliveModels.length > 0 && (
-              <ModelListWrapper
-                title="Olives"
-                modelList={filteredOliveModels}
-                selected={{ selectedModelId, setSelectedModelId }}
-                key="olive"
-              />
-            )}
-          {/* Onnx List */}
-          {isLoadingOnnxModels && (
-            <FetchingModelsLoader loadingMessage="Loading ONNX..." />
-          )}
-          {['all', 'onnx'].includes(modelFormatFilter) &&
-            !isLoadingOnnxModels &&
-            filteredOnnxModels.length > 0 && (
-              <ModelListWrapper
-                title="ONNX"
-                modelList={filteredOnnxModels}
-                selected={{ selectedModelId, setSelectedModelId }}
-                key="onnx"
-              />
-            )}
         </Flex>
       </Flex>
     </Flex>
@@ -238,12 +174,7 @@ const ModelList = (props: ModelListProps) => {
 
 export default memo(ModelList);
 
-const modelsFilter = <
-  T extends
-    | MainModelConfigEntity
-    | LoRAModelConfigEntity
-    | OnnxModelConfigEntity,
->(
+const modelsFilter = <T extends MainModelConfigEntity | LoRAModelConfigEntity>(
   data: EntityState<T, string> | undefined,
   model_type: ModelType,
   model_format: ModelFormat | undefined,
@@ -282,10 +213,7 @@ StyledModelContainer.displayName = 'StyledModelContainer';
 
 type ModelListWrapperProps = {
   title: string;
-  modelList:
-    | MainModelConfigEntity[]
-    | LoRAModelConfigEntity[]
-    | OnnxModelConfigEntity[];
+  modelList: MainModelConfigEntity[] | LoRAModelConfigEntity[];
   selected: ModelListProps;
 };
 

invokeai/frontend/web/src/features/modelManager/subpanels/ModelManagerPanel/ModelListItem.tsx

@@ -14,15 +14,14 @@ import { useTranslation } from 'react-i18next';
 import type {
   LoRAModelConfigEntity,
   MainModelConfigEntity,
-  OnnxModelConfigEntity,
 } from 'services/api/endpoints/models';
 import {
   useDeleteLoRAModelsMutation,
   useDeleteMainModelsMutation,
 } from 'services/api/endpoints/models';
 
 type ModelListItemProps = {
-  model: MainModelConfigEntity | OnnxModelConfigEntity | LoRAModelConfigEntity;
+  model: MainModelConfigEntity | LoRAModelConfigEntity;
   isSelected: boolean;
   setSelectedModelId: (v: string | undefined) => void;
 };
@@ -44,7 +43,6 @@ const ModelListItem = (props: ModelListItemProps) => {
     const method = {
       main: deleteMainModel,
       lora: deleteLoRAModel,
-      onnx: deleteMainModel,
     }[model.model_type];
 
     method(model)

invokeai/frontend/web/src/features/nodes/components/flow/nodes/Invocation/fields/inputs/MainModelFieldInputComponent.tsx

@@ -45,6 +45,7 @@ const MainModelFieldInputComponent = (props: Props) => {
       modelEntities: data,
       onChange: _onChange,
       isLoading,
+      selectedModel: field.value,
     });
 
   return (

invokeai/frontend/web/src/features/nodes/components/flow/nodes/Invocation/fields/inputs/RefinerModelFieldInputComponent.tsx

@@ -45,6 +45,7 @@ const RefinerModelFieldInputComponent = (props: Props) => {
       modelEntities: data,
       onChange: _onChange,
       isLoading,
+      selectedModel: field.value,
     });
 
   return (

invokeai/frontend/web/src/features/nodes/components/flow/nodes/Invocation/fields/inputs/SDXLMainModelFieldInputComponent.tsx

@@ -45,6 +45,7 @@ const SDXLMainModelFieldInputComponent = (props: Props) => {
       modelEntities: data,
       onChange: _onChange,
       isLoading,
+      selectedModel: field.value,
     });
 
   return (