fix: Better file and component naming for Add Models

This commit is contained in:
blessedcoolant 2023-07-17 13:58:11 +12:00
parent 641b90cc3f
commit 540f40c293
5 changed files with 41 additions and 41 deletions

View File

@ -1,8 +1,8 @@
import { ButtonGroup, Flex } from '@chakra-ui/react'; import { ButtonGroup, Flex } from '@chakra-ui/react';
import IAIButton from 'common/components/IAIButton'; import IAIButton from 'common/components/IAIButton';
import { useState } from 'react'; import { useState } from 'react';
import AutoAddModels from './AutoAddModels'; import AdvancedAddModels from './AdvancedAddModels';
import ManualAddModels from './ManualAddModels'; import SimpleAddModels from './SimpleAddModels';
export default function AddModels() { export default function AddModels() {
const [addModelMode, setAddModelMode] = useState<'simple' | 'advanced'>( const [addModelMode, setAddModelMode] = useState<'simple' | 'advanced'>(
@ -40,8 +40,8 @@ export default function AddModels() {
_dark: { background: 'base.800' }, _dark: { background: 'base.800' },
}} }}
> >
{addModelMode === 'simple' && <AutoAddModels />} {addModelMode === 'simple' && <SimpleAddModels />}
{addModelMode === 'advanced' && <ManualAddModels />} {addModelMode === 'advanced' && <AdvancedAddModels />}
</Flex> </Flex>
</Flex> </Flex>
); );

View File

@ -14,11 +14,11 @@ import BaseModelSelect from '../shared/BaseModelSelect';
import CheckpointConfigsSelect from '../shared/CheckpointConfigsSelect'; import CheckpointConfigsSelect from '../shared/CheckpointConfigsSelect';
import ModelVariantSelect from '../shared/ModelVariantSelect'; import ModelVariantSelect from '../shared/ModelVariantSelect';
export default function ManualAddCheckpoint() { export default function AdvancedAddCheckpoint() {
const { t } = useTranslation(); const { t } = useTranslation();
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const manualAddCheckpointForm = useForm<CheckpointModelConfig>({ const advancedAddCheckpointForm = useForm<CheckpointModelConfig>({
initialValues: { initialValues: {
model_name: '', model_name: '',
base_model: 'sd-1', base_model: 'sd-1',
@ -37,7 +37,7 @@ export default function ManualAddCheckpoint() {
const [useCustomConfig, setUseCustomConfig] = useState<boolean>(false); const [useCustomConfig, setUseCustomConfig] = useState<boolean>(false);
const manualAddCheckpointFormHandler = (values: CheckpointModelConfig) => { const advancedAddCheckpointFormHandler = (values: CheckpointModelConfig) => {
addMainModel({ addMainModel({
body: values, body: values,
}) })
@ -51,7 +51,7 @@ export default function ManualAddCheckpoint() {
}) })
) )
); );
manualAddCheckpointForm.reset(); advancedAddCheckpointForm.reset();
}) })
.catch((error) => { .catch((error) => {
if (error) { if (error) {
@ -69,8 +69,8 @@ export default function ManualAddCheckpoint() {
return ( return (
<form <form
onSubmit={manualAddCheckpointForm.onSubmit((v) => onSubmit={advancedAddCheckpointForm.onSubmit((v) =>
manualAddCheckpointFormHandler(v) advancedAddCheckpointFormHandler(v)
)} )}
style={{ width: '100%' }} style={{ width: '100%' }}
> >
@ -78,39 +78,39 @@ export default function ManualAddCheckpoint() {
<IAIMantineTextInput <IAIMantineTextInput
label="Model Name" label="Model Name"
required required
{...manualAddCheckpointForm.getInputProps('model_name')} {...advancedAddCheckpointForm.getInputProps('model_name')}
/> />
<BaseModelSelect <BaseModelSelect
{...manualAddCheckpointForm.getInputProps('base_model')} {...advancedAddCheckpointForm.getInputProps('base_model')}
/> />
<IAIMantineTextInput <IAIMantineTextInput
label="Model Location" label="Model Location"
required required
{...manualAddCheckpointForm.getInputProps('path')} {...advancedAddCheckpointForm.getInputProps('path')}
/> />
<IAIMantineTextInput <IAIMantineTextInput
label="Description" label="Description"
{...manualAddCheckpointForm.getInputProps('description')} {...advancedAddCheckpointForm.getInputProps('description')}
/> />
<IAIMantineTextInput <IAIMantineTextInput
label="VAE Location" label="VAE Location"
{...manualAddCheckpointForm.getInputProps('vae')} {...advancedAddCheckpointForm.getInputProps('vae')}
/> />
<ModelVariantSelect <ModelVariantSelect
{...manualAddCheckpointForm.getInputProps('variant')} {...advancedAddCheckpointForm.getInputProps('variant')}
/> />
<Flex flexDirection="column" width="100%" gap={2}> <Flex flexDirection="column" width="100%" gap={2}>
{!useCustomConfig ? ( {!useCustomConfig ? (
<CheckpointConfigsSelect <CheckpointConfigsSelect
required required
width="100%" width="100%"
{...manualAddCheckpointForm.getInputProps('config')} {...advancedAddCheckpointForm.getInputProps('config')}
/> />
) : ( ) : (
<IAIMantineTextInput <IAIMantineTextInput
required required
label="Custom Config File Location" label="Custom Config File Location"
{...manualAddCheckpointForm.getInputProps('config')} {...advancedAddCheckpointForm.getInputProps('config')}
/> />
)} )}
<IAISimpleCheckbox <IAISimpleCheckbox

View File

@ -11,13 +11,13 @@ import { DiffusersModelConfig } from 'services/api/types';
import BaseModelSelect from '../shared/BaseModelSelect'; import BaseModelSelect from '../shared/BaseModelSelect';
import ModelVariantSelect from '../shared/ModelVariantSelect'; import ModelVariantSelect from '../shared/ModelVariantSelect';
export default function ManualAddDiffusers() { export default function AdvancedAddDiffusers() {
const { t } = useTranslation(); const { t } = useTranslation();
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const [addMainModel] = useAddMainModelsMutation(); const [addMainModel] = useAddMainModelsMutation();
const manualAddDiffusersForm = useForm<DiffusersModelConfig>({ const advancedAddDiffusersForm = useForm<DiffusersModelConfig>({
initialValues: { initialValues: {
model_name: '', model_name: '',
base_model: 'sd-1', base_model: 'sd-1',
@ -30,7 +30,7 @@ export default function ManualAddDiffusers() {
variant: 'normal', variant: 'normal',
}, },
}); });
const manualAddDiffusersFormHandler = (values: DiffusersModelConfig) => { const advancedAddDiffusersFormHandler = (values: DiffusersModelConfig) => {
addMainModel({ addMainModel({
body: values, body: values,
}) })
@ -44,7 +44,7 @@ export default function ManualAddDiffusers() {
}) })
) )
); );
manualAddDiffusersForm.reset(); advancedAddDiffusersForm.reset();
}) })
.catch((error) => { .catch((error) => {
if (error) { if (error) {
@ -62,8 +62,8 @@ export default function ManualAddDiffusers() {
return ( return (
<form <form
onSubmit={manualAddDiffusersForm.onSubmit((v) => onSubmit={advancedAddDiffusersForm.onSubmit((v) =>
manualAddDiffusersFormHandler(v) advancedAddDiffusersFormHandler(v)
)} )}
style={{ width: '100%' }} style={{ width: '100%' }}
> >
@ -71,27 +71,27 @@ export default function ManualAddDiffusers() {
<IAIMantineTextInput <IAIMantineTextInput
required required
label="Model Name" label="Model Name"
{...manualAddDiffusersForm.getInputProps('model_name')} {...advancedAddDiffusersForm.getInputProps('model_name')}
/> />
<BaseModelSelect <BaseModelSelect
{...manualAddDiffusersForm.getInputProps('base_model')} {...advancedAddDiffusersForm.getInputProps('base_model')}
/> />
<IAIMantineTextInput <IAIMantineTextInput
required required
label="Model Location" label="Model Location"
placeholder="Provide the path to a local folder where your Diffusers Model is stored" placeholder="Provide the path to a local folder where your Diffusers Model is stored"
{...manualAddDiffusersForm.getInputProps('path')} {...advancedAddDiffusersForm.getInputProps('path')}
/> />
<IAIMantineTextInput <IAIMantineTextInput
label="Description" label="Description"
{...manualAddDiffusersForm.getInputProps('description')} {...advancedAddDiffusersForm.getInputProps('description')}
/> />
<IAIMantineTextInput <IAIMantineTextInput
label="VAE Location" label="VAE Location"
{...manualAddDiffusersForm.getInputProps('vae')} {...advancedAddDiffusersForm.getInputProps('vae')}
/> />
<ModelVariantSelect <ModelVariantSelect
{...manualAddDiffusersForm.getInputProps('variant')} {...advancedAddDiffusersForm.getInputProps('variant')}
/> />
<IAIButton mt={2} type="submit"> <IAIButton mt={2} type="submit">
{t('modelManager.addModel')} {t('modelManager.addModel')}

View File

@ -2,29 +2,29 @@ import { Flex } from '@chakra-ui/react';
import { SelectItem } from '@mantine/core'; import { SelectItem } from '@mantine/core';
import IAIMantineSelect from 'common/components/IAIMantineSelect'; import IAIMantineSelect from 'common/components/IAIMantineSelect';
import { useState } from 'react'; import { useState } from 'react';
import ManualAddCheckpoint from './ManualAddCheckpoint'; import AdvancedAddCheckpoint from './AdvancedAddCheckpoint';
import ManualAddDiffusers from './ManualAddDiffusers'; import AdvancedAddDiffusers from './AdvancedAddDiffusers';
const manualAddModeData: SelectItem[] = [ const advancedAddModeData: SelectItem[] = [
{ label: 'Diffusers', value: 'diffusers' }, { label: 'Diffusers', value: 'diffusers' },
{ label: 'Checkpoint / Safetensors', value: 'checkpoint' }, { label: 'Checkpoint / Safetensors', value: 'checkpoint' },
]; ];
type ManualAddMode = 'diffusers' | 'checkpoint'; type ManualAddMode = 'diffusers' | 'checkpoint';
export default function ManualAddModels() { export default function AdvancedAddModels() {
const [manualAddMode, setManualAddMode] = const [advancedAddMode, setAdvancedAddMode] =
useState<ManualAddMode>('diffusers'); useState<ManualAddMode>('diffusers');
return ( return (
<Flex flexDirection="column" gap={4} width="100%"> <Flex flexDirection="column" gap={4} width="100%">
<IAIMantineSelect <IAIMantineSelect
label="Model Type" label="Model Type"
value={manualAddMode} value={advancedAddMode}
data={manualAddModeData} data={advancedAddModeData}
onChange={(v) => { onChange={(v) => {
if (!v) return; if (!v) return;
setManualAddMode(v as ManualAddMode); setAdvancedAddMode(v as ManualAddMode);
}} }}
/> />
@ -38,8 +38,8 @@ export default function ManualAddModels() {
}, },
}} }}
> >
{manualAddMode === 'diffusers' && <ManualAddDiffusers />} {advancedAddMode === 'diffusers' && <AdvancedAddDiffusers />}
{manualAddMode === 'checkpoint' && <ManualAddCheckpoint />} {advancedAddMode === 'checkpoint' && <AdvancedAddCheckpoint />}
</Flex> </Flex>
</Flex> </Flex>
); );

View File

@ -25,7 +25,7 @@ type ExtendedImportModelConfig = {
prediction_type?: 'v_prediction' | 'epsilon' | 'sample' | 'none' | undefined; prediction_type?: 'v_prediction' | 'epsilon' | 'sample' | 'none' | undefined;
}; };
export default function AutoAddModels() { export default function SimpleAddModels() {
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const { t } = useTranslation(); const { t } = useTranslation();