diff --git a/invokeai/frontend/web/src/features/parameters/components/VAEModel/ParamFLUXVAEModelSelect.tsx b/invokeai/frontend/web/src/features/parameters/components/VAEModel/ParamFLUXVAEModelSelect.tsx index 49502ebc62..df1b7f8f5f 100644 --- a/invokeai/frontend/web/src/features/parameters/components/VAEModel/ParamFLUXVAEModelSelect.tsx +++ b/invokeai/frontend/web/src/features/parameters/components/VAEModel/ParamFLUXVAEModelSelect.tsx @@ -6,14 +6,14 @@ import { fluxVAESelected, selectFLUXVAE } from 'features/controlLayers/store/par import { zModelIdentifierField } from 'features/nodes/types/common'; import { memo, useCallback } from 'react'; import { useTranslation } from 'react-i18next'; -import { useFluxVAEModels } from 'services/api/hooks/modelsByType'; +import { useFlux1VAEModels } from 'services/api/hooks/modelsByType'; import type { VAEModelConfig } from 'services/api/types'; const ParamFLUXVAEModelSelect = () => { const dispatch = useAppDispatch(); const { t } = useTranslation(); const vae = useAppSelector(selectFLUXVAE); - const [modelConfigs, { isLoading }] = useFluxVAEModels(); + const [modelConfigs, { isLoading }] = useFlux1VAEModels(); const _onChange = useCallback( (vae: VAEModelConfig | null) => { diff --git a/invokeai/frontend/web/src/services/api/hooks/modelsByType.ts b/invokeai/frontend/web/src/services/api/hooks/modelsByType.ts index b771dd7840..1819b64992 100644 --- a/invokeai/frontend/web/src/services/api/hooks/modelsByType.ts +++ b/invokeai/frontend/web/src/services/api/hooks/modelsByType.ts @@ -56,7 +56,6 @@ export const useCLIPEmbedModels = () => buildModelsHook(isCLIPEmbedModelConfigOr export const useSpandrelImageToImageModels = buildModelsHook(isSpandrelImageToImageModelConfig); export const useEmbeddingModels = buildModelsHook(isTIModelConfig); export const useVAEModels = () => buildModelsHook(isVAEModelConfigOrSubmodel)(); -export const useFluxVAEModels = () => buildModelsHook(isFluxVAEModelConfig)(); export const useFlux1VAEModels = () => buildModelsHook(isFlux1VAEModelConfig)(); export const useFlux2VAEModels = () => buildModelsHook(isFlux2VAEModelConfig)(); export const useZImageDiffusersModels = () => buildModelsHook(isZImageDiffusersMainModelConfig)();