feat: Make the depth anything small model the default

This commit is contained in:
blessedcoolant 2024-01-23 09:14:47 +05:30 committed by Kent Keirsey
parent 6a2eb1d2e4
commit f36a691219
3 changed files with 4 additions and 4 deletions

View File

@ -619,7 +619,7 @@ class DepthAnythingImageProcessorInvocation(ImageProcessorInvocation):
"""Generates a depth map based on the Depth Anything algorithm"""
model_size: DEPTH_ANYTHING_MODEL_SIZES = InputField(
default="large", description="The size of the depth model to use"
default="small", description="The size of the depth model to use"
)
offload: bool = InputField(default=False)

View File

@ -42,9 +42,9 @@ const DepthAnythingProcessor = (props: Props) => {
const options: { label: string; value: DepthAnythingModelSize }[] = useMemo(
() => [
{ label: t('controlnet.large'), value: 'large' },
{ label: t('controlnet.base'), value: 'base' },
{ label: t('controlnet.small'), value: 'small' },
{ label: t('controlnet.base'), value: 'base' },
{ label: t('controlnet.large'), value: 'large' },
],
[t]
);

View File

@ -94,7 +94,7 @@ export const CONTROLNET_PROCESSORS: ControlNetProcessorsDict = {
default: {
id: 'depth_anything_image_processor',
type: 'depth_anything_image_processor',
model_size: 'large',
model_size: 'small',
offload: false,
},
},