Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

do not display sdxl models in main model loader

Browse Source
This commit is contained in:
Lincoln Stein 2023-07-11 08:51:02 -04:00
parent 4d9a342437
commit 34cff848c7
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
invokeai/frontend/web/src/features/nodes/components/InputFieldComponent.tsx
View File

@ -150,6 +150,7 @@ const InputFieldComponent = (props: InputFieldComponentProps) => {
nodeId={nodeId} nodeId={nodeId}
field={field} field={field}
template={template} template={template}
base_models={['sd-1', 'sd-2']}
/> />
); );
} }

4
invokeai/frontend/web/src/features/nodes/components/fields/ModelInputFieldComponent.tsx
View File

@ -17,14 +17,14 @@ import { FieldComponentProps } from './types';
const ModelInputFieldComponent = ( const ModelInputFieldComponent = (
props: FieldComponentProps<ModelInputFieldValue, ModelInputFieldTemplate> props: FieldComponentProps<ModelInputFieldValue, ModelInputFieldTemplate>
) => { ) => {
const { nodeId, field } = props; const { nodeId, field, base_models } = props;
const dispatch = useAppDispatch(); const dispatch = useAppDispatch();
const { t } = useTranslation(); const { t } = useTranslation();
const { data: mainModels } = useGetMainModelsQuery({ const { data: mainModels } = useGetMainModelsQuery({
model_type: 'main', model_type: 'main',
base_models: ['sd-1', 'sd-2', 'sdxl', 'sdxl-refiner'], base_models: base_models,
}); });
const data = useMemo(() => { const data = useMemo(() => {