do not display sdxl models in main model loader

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

View File

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

View File

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