github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

tidy(ui): use StarterModel type directly

Browse Source
This commit is contained in:
psychedelicious
2024-10-15 12:18:02 +10:00
committed by Mary Hipp Rogers
parent 8613efb03a
commit f78a542401
2 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StartModelsResultItem.tsx
View File

@@ -5,10 +5,10 @@ import ModelBaseBadge from 'features/modelManagerV2/subpanels/ModelManagerPanel/
import { memo, useCallback, useMemo } from 'react';
import { useTranslation } from 'react-i18next';
import { PiPlusBold } from 'react-icons/pi';
import type { GetStarterModelsResponse } from 'services/api/endpoints/models';
import type { StarterModel } from 'services/api/types';
type Props = {
model: GetStarterModelsResponse['starter_models'][number];
model: StarterModel;
};
export const StarterModelsResultItem = memo(({ model }: Props) => {
const { t } = useTranslation();

19
invokeai/frontend/web/src/features/modelManagerV2/subpanels/AddModelPanel/StarterModels/StarterBundle.tsx
View File

@@ -5,15 +5,10 @@ import { MODEL_TYPE_SHORT_MAP } from 'features/parameters/types/constants';
import { toast } from 'features/toast/toast';
import { useCallback, useMemo } from 'react';
import { useTranslation } from 'react-i18next';
import { type GetStarterModelsResponse, useInstallModelMutation } from 'services/api/endpoints/models';
import { useInstallModelMutation } from 'services/api/endpoints/models';
import type { StarterModel } from 'services/api/types';
export const StarterBundle = ({
bundleName,
bundle,
}: {
bundleName: string;
bundle: GetStarterModelsResponse['starter_bundles'][number];
}) => {
export const StarterBundle = ({ bundleName, bundle }: { bundleName: string; bundle: StarterModel[] }) => {
const [installModel] = useInstallModelMutation();
const buildModelToInstall = useBuildModelsToInstall();
const { t } = useTranslation();
@@ -71,10 +66,12 @@ export const StarterBundle = ({
</Flex>
}
>
<Button size="sm" onClick={handleClickBundle} py={6} >
<Button size="sm" onClick={handleClickBundle} py={6}>
<Flex flexDir="column">
<Text>{isMainModelBase(bundleName) && MODEL_TYPE_SHORT_MAP[bundleName]}</Text>
<Text fontSize="xs">({bundle.length} {t("settings.models")})</Text>
<Text>{isMainModelBase(bundleName) && MODEL_TYPE_SHORT_MAP[bundleName]}</Text>
<Text fontSize="xs">
({bundle.length} {t('settings.models')})
</Text>
</Flex>
</Button>
</Tooltip>